post-formatting

polaris-hub · Feb 23, 2024 · b5496c0 · b5496c0
1 parent 52682bd
commit b5496c0
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 19 deletions.
diff --git a/polaris/hub/client.py b/polaris/hub/client.py
@@ -346,15 +346,15 @@ def read_zarr_file(self, owner: str, name: str, path: str, **kwargs):
         """
         polaris_fs = PolarisFSFileSystem(
             polaris_client=self,
-            polarisfs_url=str(self.base_url), 
+            polarisfs_url=str(self.base_url),
             default_expirations_seconds=10 * 60,
             dataset_owner=owner,
-            dataset_name=name
+            dataset_name=name,
         )
 
         try:
             store = zarr.storage.FSStore(path, fs=polaris_fs)
-            return zarr.open(store, mode="r")            
+            return zarr.open(store, mode="r")
         except Exception as e:
             raise PolarisHubError(f"Error opening Zarr store: {e}")
 

diff --git a/polaris/hub/polarisfs.py b/polaris/hub/polarisfs.py
@@ -1,5 +1,4 @@
 from typing import Optional
-from typing import Union
 
 import fsspec
 import httpx
@@ -8,6 +7,7 @@
 
 from polaris.utils.errors import PolarisHubError
 
+
 class PolarisfsSettings(BaseSettings):
     url: str = "http://localhost:3000/"
 
@@ -19,13 +19,15 @@ class PolarisFSFileSystem(fsspec.AbstractFileSystem):
     protocol = "polarisfs"
     async_impl = False
 
-    def __init__(self, 
-                polaris_client, 
-                polarisfs_url: Optional[str],
-                dataset_owner: str,
-                dataset_name: str,
-                default_expirations_seconds: int = 10 * 60, 
-                **kwargs):
+    def __init__(
+        self,
+        polaris_client,
+        polarisfs_url: Optional[str],
+        dataset_owner: str,
+        dataset_name: str,
+        default_expirations_seconds: int = 10 * 60,
+        **kwargs,
+    ):
         super().__init__(**kwargs)
 
         if polarisfs_url is None:
@@ -35,7 +37,7 @@ def __init__(self,
 
         self.default_expirations_seconds = default_expirations_seconds
 
-        self.polaris_client = polaris_client 
+        self.polaris_client = polaris_client
 
         self.http_client = httpx.Client(base_url=self.settings.url)
         self.http_fs = fsspec.filesystem("http")
@@ -56,11 +58,14 @@ def ls(self, path: str, detail=False, **kwargs):
             entries = [p["name"].replace(self.prefix, "") for p in response.json()]
             return entries
         else:
-            detailed_entries = [{"name": p["name"].replace(self.prefix, ""), "size": p["size"], "type": p["type"]} for p in response.json()]
+            detailed_entries = [
+                {"name": p["name"].replace(self.prefix, ""), "size": p["size"], "type": p["type"]}
+                for p in response.json()
+            ]
             return detailed_entries
 
-    def cat_file(self, path:str, **kwargs):
-        cat_path=f"{self.base_path}/{path}"
+    def cat_file(self, path: str, **kwargs):
+        cat_path = f"{self.base_path}/{path}"
 
         # GET request to Polaris Hub for signed URL of file
         response = self.polaris_client.get(cat_path, params={})
@@ -74,4 +79,4 @@ def cat_file(self, path:str, **kwargs):
 
         storage_response = self.http_client.get(response.json()["url"], timeout=(10, 200))
 
-        return storage_response.content
+        return storage_response.content
diff --git a/polaris/utils/types.py b/polaris/utils/types.py
@@ -115,9 +115,9 @@ class License(BaseModel):
             Else it is required to manually specify this.
     """
 
-    SPDX_LICENSE_DATA_PATH: ClassVar[str] = (
-        "https://raw.githubusercontent.com/spdx/license-list-data/main/json/licenses.json"
-    )
+    SPDX_LICENSE_DATA_PATH: ClassVar[
+        str
+    ] = "https://raw.githubusercontent.com/spdx/license-list-data/main/json/licenses.json"
 
     id: str
     reference: Optional[HttpUrlString] = None