fix linting

Signed-off-by: Artem Petrov <[email protected]>
flyteorg · Apr 14, 2023 · 2bdd426 · 2bdd426
1 parent 1dbb4d4
commit 2bdd426
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/flytekit/clis/sdk_in_container/launchplan.py b/flytekit/clis/sdk_in_container/launchplan.py
@@ -66,6 +66,9 @@ def launchplan(
         )
         state = LaunchPlanState.ACTIVE if activate else LaunchPlanState.INACTIVE
         remote.client.update_launch_plan(id=launchplan.id, state=state)
-        click.secho(f"\n Launchplan was set to {LaunchPlanState.enum_to_string(state)}: {launchplan.name}:{launchplan.id.version}", fg="green")
+        click.secho(
+            f"\n Launchplan was set to {LaunchPlanState.enum_to_string(state)}: {launchplan.name}:{launchplan.id.version}",
+            fg="green",
+        )
     except StopIteration as e:
         click.secho(f"{e.value}", fg="red")
diff --git a/tests/flytekit/unit/cli/pyflyte/test_launchplan.py b/tests/flytekit/unit/cli/pyflyte/test_launchplan.py
@@ -1,18 +1,17 @@
+import pytest
 from click.testing import CliRunner
 from mock import mock
 
 from flytekit.clis.sdk_in_container import pyflyte
 from flytekit.remote import FlyteRemote
-import pytest
-
 
 
 @mock.patch("flytekit.clis.sdk_in_container.helpers.FlyteRemote", spec=FlyteRemote)
 @pytest.mark.parametrize(
     ("action", "expected_state"),
     [
         ("activate", "ACTIVE"),
-        ("deactivate", "INACTIVE")
+        ("deactivate", "INACTIVE"),
     ],
 )
 def test_pyflyte_launchplan(mock_remote, action, expected_state):