lint

flyteorg · May 9, 2023 · 32ef6dd · 32ef6dd
1 parent e632803
commit 32ef6dd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/plugins/flytekit-papermill/flytekitplugins/papermill/task.py b/plugins/flytekit-papermill/flytekitplugins/papermill/task.py
@@ -4,7 +4,7 @@
 import sys
 import tempfile
 import typing
-from typing import Any, Dict
+from typing import Any
 
 import nbformat
 import papermill as pm

diff --git a/plugins/flytekit-papermill/tests/test_task.py b/plugins/flytekit-papermill/tests/test_task.py
@@ -8,7 +8,7 @@
 from kubernetes.client import V1Container, V1PodSpec
 
 import flytekit
-from flytekit import StructuredDataset, kwtypes, task, workflow
+from flytekit import StructuredDataset, kwtypes, task
 from flytekit.configuration import Image, ImageConfig
 from flytekit.types.directory import FlyteDirectory
 from flytekit.types.file import FlyteFile, PythonNotebook
@@ -171,4 +171,4 @@ def create_sd() -> StructuredDataset:
         outputs=kwtypes(success=bool),
     )
     success, out, render = nb_types.execute(ff=ff, fd=fd, sd=sd)
-    assert success == True, "Notebook execution failed"
+    assert success is True, "Notebook execution failed"