Make morphology's brain_region/species optional

Author: eleftherioszisis

src/entitysdk/models/__init__.py

@@ -10,12 +10,14 @@
 from entitysdk.models.morphology import ReconstructionMorphology
 from entitysdk.models.mtype import MTypeClass
 from entitysdk.models.taxonomy import Species, Strain, Taxonomy
+from entitysdk.models.emodel import EModel
 
 __all__ = [
     "Asset",
     "BrainLocation",
     "BrainRegion",
     "Contribution",
+    "EModel",
     "IonChannelModel",
     "License",
     "MTypeClass",

src/entitysdk/models/morphology.py

@@ -31,11 +31,11 @@ class ReconstructionMorphology(HasAssets, Entity):
         ),
     ] = None
     brain_region: Annotated[
-        BrainRegion,
+        BrainRegion | None,
         Field(
             description="The region of the brain where the morphology is located.",
         ),
-    ]
+    ] = None
     description: Annotated[
         str,
         Field(
@@ -44,11 +44,11 @@ class ReconstructionMorphology(HasAssets, Entity):
         ),
     ]
     species: Annotated[
-        Species,
+        Species | None,
         Field(
             description="The species of the morphology.",
         ),
-    ]
+    ] = None
     strain: Annotated[
         Strain | None,
         Field(

tests/integration/test_searching.py

@@ -11,6 +11,10 @@
     Role,
     Species,
     Strain,
+    Role,
+    Organization,
+    Person,
+    EModel,
 )
 
 
@@ -27,6 +31,7 @@
         Species,
         Strain,
         Organization,
+        EModel,
     ],
 )
 def test_is_searchable(entity_type, client):