lint

BlueBrain · Dec 2, 2024 · 81f3702 · 81f3702
1 parent a4481b7
commit 81f3702
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 8 deletions.
diff --git a/src/neuroagent/cell_types.py b/src/neuroagent/cell_types.py
@@ -17,7 +17,7 @@ class CellTypesMeta:
     """
 
     def __init__(self) -> None:
-        self.name_: dict[str, str] = {}
+        self.name_: dict[str, str | None] = {}
         self.descendants_ids: dict[str, set[str]] = {}
 
     def descendants(self, ids: str | set[str]) -> set[str]:

diff --git a/swarm_copy/cell_types.py b/swarm_copy/cell_types.py
@@ -17,7 +17,7 @@ class CellTypesMeta:
     """
 
     def __init__(self) -> None:
-        self.name_: dict[str, str] = {}
+        self.name_: dict[str, str | None] = {}
         self.descendants_ids: dict[str, set[str]] = {}
 
     def descendants(self, ids: str | set[str]) -> set[str]:

diff --git a/swarm_copy/tools/bluenaas_memodel_getall.py b/swarm_copy/tools/bluenaas_memodel_getall.py
@@ -29,11 +29,9 @@ class InputMEModelGetAll(BaseModel):
     page_size: int = Field(
         default=20, description="Number of results returned by the API."
     )
-    memodel_type: Literal["single-neuron-simulation", "synaptome-simulation"] = (
-        Field(
-            default="single-neuron-simulation",
-            description="Type of simulation to retrieve.",
-        )
+    memodel_type: Literal["single-neuron-simulation", "synaptome-simulation"] = Field(
+        default="single-neuron-simulation",
+        description="Type of simulation to retrieve.",
     )
 
 
@@ -57,7 +55,7 @@ async def arun(self) -> PaginatedResponseUnionMEModelResponseSynaptomeModelRespo
         response = await self.metadata.httpx_client.get(
             url=f"{self.metadata.bluenaas_url}/neuron-model/{self.metadata.vlab_id}/{self.metadata.project_id}/me-models",
             params={
-                "simulation_type": self.input_schema.simulation_type,
+                "simulation_type": self.input_schema.memodel_type,
                 "offset": self.input_schema.offset,
                 "page_size": self.input_schema.page_size,
             },