Modify models

Author: eleftherioszisis

src/entitysdk/models/__init__.py

@@ -5,8 +5,8 @@
 from entitysdk.models.brain_location import BrainLocation
 from entitysdk.models.brain_region import BrainRegion
 from entitysdk.models.contribution import Contribution, Role
-from entitysdk.models.entity import Entity
 from entitysdk.models.ion_channel_model import IonChannelModel, NeuronBlock, UseIon
+from entitysdk.models.license import License
 from entitysdk.models.morphology import ReconstructionMorphology
 from entitysdk.models.mtype import MTypeClass
 from entitysdk.models.taxonomy import Species, Strain, Taxonomy
@@ -16,8 +16,8 @@
     "BrainLocation",
     "BrainRegion",
     "Contribution",
-    "Entity",
     "IonChannelModel",
+    "License",
     "MTypeClass",
     "NeuronBlock",
     "Organization",

tests/integration/test_searching.py

@@ -1,19 +1,24 @@
 import pytest
 
-from entitysdk.models.agent import Organization, Person
-from entitysdk.models.contribution import Role
-from entitysdk.models.morphology import (
+from entitysdk.models import (
+    Contribution,
+    IonChannelModel,
     License,
+    MTypeClass,
+    Organization,
+    Person,
     ReconstructionMorphology,
+    Role,
     Species,
     Strain,
 )
-from entitysdk.models.mtype import MTypeClass
 
 
 @pytest.mark.parametrize(
     "entity_type",
     [
+        Contribution,
+        IonChannelModel,
         License,
         MTypeClass,
         Person,