From f9552c56c2efff51b277c056127d11317a166d47 Mon Sep 17 00:00:00 2001 From: Kevin Yang Date: Tue, 16 Jan 2024 16:09:19 -0500 Subject: [PATCH] linter --- jupyter-extension/jupyterlab_pachyderm/env.py | 3 ++- jupyter-extension/jupyterlab_pachyderm/handlers.py | 8 ++++---- 2 files changed, 6 insertions(+), 5 deletions(-) diff --git a/jupyter-extension/jupyterlab_pachyderm/env.py b/jupyter-extension/jupyterlab_pachyderm/env.py index 88425c7c799..e4f422e92bc 100644 --- a/jupyter-extension/jupyterlab_pachyderm/env.py +++ b/jupyter-extension/jupyterlab_pachyderm/env.py @@ -5,7 +5,8 @@ PACH_CONFIG = os.environ.get("PACH_CONFIG", Path.home() / ".pachyderm/config.json") PFS_MOUNT_DIR = os.environ.get("PFS_MOUNT_DIR", "/pfs") PACHD_ADDRESS = os.environ.get("PACHD_ADDRESS", None) -DEX_TOKEN = os.environ.get("DEX_TOKEN", None) # If specified, this is the ID_TOKEN used in auth +# If specified, this is the ID_TOKEN used in auth +DEX_TOKEN = os.environ.get("DEX_TOKEN", None) PACHYDERM_EXT_DEBUG = strtobool(os.environ.get("PACHYDERM_EXT_DEBUG", "False").lower()) if PACHYDERM_EXT_DEBUG: diff --git a/jupyter-extension/jupyterlab_pachyderm/handlers.py b/jupyter-extension/jupyterlab_pachyderm/handlers.py index 3045994212d..0bbacce0e0f 100644 --- a/jupyter-extension/jupyterlab_pachyderm/handlers.py +++ b/jupyter-extension/jupyterlab_pachyderm/handlers.py @@ -332,9 +332,7 @@ async def put(self): address = body["pachd_address"] cas = bytes(body["server_cas"], "utf-8") if "server_cas" in body else None - client = Client().from_pachd_address( - pachd_address=address, root_certs=cas - ) + client = Client().from_pachd_address(pachd_address=address, root_certs=cas) self.settings["pachyderm_client"] = client self.settings["pfs_contents_manager"] = PFSManager(client=client) self.settings["datum_contents_manager"] = DatumManager(client=client) @@ -462,7 +460,9 @@ def setup_handlers(web_app): if PACHD_ADDRESS: client = Client().from_pachd_address(pachd_address=PACHD_ADDRESS) if DEX_TOKEN: - client.auth_token = client.auth.authenticate(id_token=DEX_TOKEN).pach_token + client.auth_token = client.auth.authenticate( + id_token=DEX_TOKEN + ).pach_token else: client = Client() get_logger().debug(