Format code

Author: FlorentinD

examples/dev/gds-sessions.ipynb

@@ -547,22 +547,8 @@
   }
  ],
  "metadata": {
-  "kernelspec": {
-   "display_name": "Python 3",
-   "language": "python",
-   "name": "python3"
-  },
   "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.11.7"
+   "name": "python"
   }
  },
  "nbformat": 4,

graphdatascience/gds_session/gds_sessions.py

@@ -12,8 +12,8 @@
 )
 from graphdatascience.gds_session.aura_graph_data_science import AuraGraphDataScience
 from graphdatascience.gds_session.dbms_connection_info import DbmsConnectionInfo
-from graphdatascience.gds_session.session_sizes import SessionSizeByMemory
 from graphdatascience.gds_session.region_suggester import closest_match
+from graphdatascience.gds_session.session_sizes import SessionSizeByMemory
 
 
 @dataclass
@@ -59,7 +59,7 @@ def get_or_create(
         db_instance = self._aura_api.list_instance(db_instance_id)
         if not db_instance:
             raise ValueError(f"Could not find Aura instance with the uri `{db_connection.uri}`")
-        
+
         region = self._ds_region(db_instance.region, db_instance.cloud_provider)
 
         create_details = self._aura_api.create_instance(
@@ -137,8 +137,9 @@ def _ds_region(self, region: str, cloud_provider: str) -> str:
 
         if not available_regions:
             raise ValueError(
-                f"Tenant `{tenant_details.id}` cannot create GDS sessions at cloud provider `{cloud_provider}`.")
-        
+                f"Tenant `{tenant_details.id}` cannot create GDS sessions at cloud provider `{cloud_provider}`."
+            )
+
         return closest_match(region, available_regions)
 
     def _construct_client(

graphdatascience/gds_session/region_suggester.py

@@ -1,8 +1,9 @@
-from typing import List
+from typing import Iterable
+
 import textdistance
 
 
-def closest_match(query: str, options: List[str]) -> str:
+def closest_match(query: str, options: Iterable[str]) -> str:
     if not options:
         raise ValueError("No options to choose from.")
 
@@ -13,4 +14,4 @@ def closest_match(query: str, options: List[str]) -> str:
             closest_option = ep
             curr_max_similarity = similarity
 
-    return closest_option
+    return closest_option

graphdatascience/tests/unit/test_gds_sessions.py

@@ -83,9 +83,7 @@ def wait_for_instance_running(
         return super().wait_for_instance_running(instance_id, sleep_time=0.0001, max_sleep_time=0.001)
 
     def tenant_details(self) -> TenantDetails:
-        return TenantDetails(
-            id=self._tenant_id, ds_type="fake-ds", regions_per_provider={"aws": {"leipzig-1"}}
-        )
+        return TenantDetails(id=self._tenant_id, ds_type="fake-ds", regions_per_provider={"aws": {"leipzig-1"}})
 
 
 @pytest.fixture