fix lint issue

flyteorg · Aug 18, 2023 · 1ed6da8 · 1ed6da8
1 parent 5b81336
commit 1ed6da8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/flytekit/core/context_manager.py b/flytekit/core/context_manager.py
@@ -349,7 +349,9 @@ def __getattr__(self, item: str) -> _GroupSecrets:
         """
         return self._GroupSecrets(item, self)
 
-    def get(self, group: str, key: Optional[str] = None, group_version: Optional[str] = None, encode_mode: str = "r") -> str:
+    def get(
+        self, group: str, key: Optional[str] = None, group_version: Optional[str] = None, encode_mode: str = "r"
+    ) -> str:
         """
         Retrieves a secret using the resolution order -> Env followed by file. If not found raises a ValueError
         """

diff --git a/tests/flytekit/unit/core/test_context_manager.py b/tests/flytekit/unit/core/test_context_manager.py
@@ -1,7 +1,7 @@
+import base64
 import os
 from datetime import datetime
 
-import base64
 import mock
 import py
 import pytest
@@ -168,7 +168,7 @@ def test_secrets_manager_file(tmpdir: py.path.local):
     assert sec.get("group", "test") == "my-password"
     assert sec.group.test == "my-password"
 
-    base64_string ="R2Vla3NGb3JHZWV =="
+    base64_string = "R2Vla3NGb3JHZWV =="
     base64_bytes = base64_string.encode("ascii")
     base64_str = base64.b64encode(base64_bytes)
     with open(f, "wb") as w: