From abc902b3ac44085dfae47eebf555e2424154281d Mon Sep 17 00:00:00 2001 From: Yue Shang Date: Thu, 17 Aug 2023 21:24:10 -0700 Subject: [PATCH] fix lint issue Signed-off-by: Yue Shang --- flytekit/core/context_manager.py | 4 +++- tests/flytekit/unit/core/test_context_manager.py | 4 ++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/flytekit/core/context_manager.py b/flytekit/core/context_manager.py index fb816b39ecb..a4c913fd4cc 100644 --- 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 index 7b1db5a376f..2ec7eb8e195 100644 --- 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: