From 32ef6dd741b70abac633d5f4435b4a7847627245 Mon Sep 17 00:00:00 2001
From: esad <esad@peridotml.io>
Date: Tue, 9 May 2023 08:55:37 -0400
Subject: [PATCH] lint

---
 plugins/flytekit-papermill/flytekitplugins/papermill/task.py | 2 +-
 plugins/flytekit-papermill/tests/test_task.py                | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/plugins/flytekit-papermill/flytekitplugins/papermill/task.py b/plugins/flytekit-papermill/flytekitplugins/papermill/task.py
index 43a21d9e36b..7dd33ceaa07 100644
--- 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
index e46d8422d25..0e54e7082ee 100644
--- 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"