move ModelBaseModel back to models

zenml-io · Sep 20, 2023 · 0fa0d5a · 0fa0d5a
1 parent a3db9c9
commit 0fa0d5a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/src/zenml/model/model_config.py b/src/zenml/model/model_config.py
@@ -19,8 +19,8 @@
 
 from zenml.exceptions import EntityExistsError
 from zenml.logger import get_logger
-from zenml.model.base_model import ModelBaseModel
 from zenml.model.model_stages import ModelStages
+from zenml.models.model_base_model import ModelBaseModel
 
 if TYPE_CHECKING:
     from zenml.models.model_models import (

diff --git a/src/zenml/model/base_model.py → src/zenml/models/model_base_model.py b/src/zenml/model/base_model.py → src/zenml/models/model_base_model.py
@@ -17,8 +17,7 @@
 
 from pydantic import BaseModel, Field
 
-STR_FIELD_MAX_LENGTH = 255
-TEXT_FIELD_MAX_LENGTH = 65535
+from zenml.models.constants import STR_FIELD_MAX_LENGTH, TEXT_FIELD_MAX_LENGTH
 
 
 class ModelBaseModel(BaseModel):

diff --git a/src/zenml/models/model_models.py b/src/zenml/models/model_models.py
@@ -18,14 +18,14 @@
 
 from pydantic import BaseModel, Field, validator
 
-from zenml.model.base_model import ModelBaseModel
 from zenml.models.artifact_models import ArtifactResponseModel
 from zenml.models.base_models import (
     WorkspaceScopedRequestModel,
     WorkspaceScopedResponseModel,
 )
 from zenml.models.constants import STR_FIELD_MAX_LENGTH, TEXT_FIELD_MAX_LENGTH
 from zenml.models.filter_models import WorkspaceScopedFilterModel
+from zenml.models.model_base_model import ModelBaseModel
 from zenml.models.pipeline_run_models import PipelineRunResponseModel
 
 if TYPE_CHECKING: