Rename model

Author: schustmi

src/zenml/config/strict_base_model.py renamed to src/zenml/config/frozen_base_model.py

@@ -16,7 +16,7 @@
 from pydantic import BaseModel, ConfigDict
 
 
-class StrictBaseModel(BaseModel):
-    """Immutable pydantic model which prevents extra attributes."""
+class FrozenBaseModel(BaseModel):
+    """Immutable pydantic model which ignores extra attributes."""
 
     model_config = ConfigDict(frozen=True, extra="ignore")

src/zenml/config/pipeline_configurations.py

@@ -22,7 +22,7 @@
 from zenml.config.constants import DOCKER_SETTINGS_KEY
 from zenml.config.retry_config import StepRetryConfig
 from zenml.config.source import SourceWithValidator
-from zenml.config.strict_base_model import StrictBaseModel
+from zenml.config.strict_base_model import FrozenBaseModel
 from zenml.model.model import Model
 from zenml.utils.tag_utils import Tag
 from zenml.utils.time_utils import utc_now
@@ -35,7 +35,7 @@
 DISALLOWED_PIPELINE_NAMES = ["unlisted"]
 
 
-class PipelineConfigurationUpdate(StrictBaseModel):
+class PipelineConfigurationUpdate(FrozenBaseModel):
     """Class for pipeline configuration updates."""
 
     enable_cache: Optional[bool] = None

src/zenml/config/pipeline_run_configuration.py

@@ -23,15 +23,15 @@
 from zenml.config.schedule import Schedule
 from zenml.config.source import SourceWithValidator
 from zenml.config.step_configurations import StepConfigurationUpdate
-from zenml.config.strict_base_model import StrictBaseModel
+from zenml.config.strict_base_model import FrozenBaseModel
 from zenml.model.model import Model
 from zenml.models import PipelineBuildBase
 from zenml.utils import pydantic_utils
 from zenml.utils.tag_utils import Tag
 
 
 class PipelineRunConfiguration(
-    StrictBaseModel, pydantic_utils.YAMLSerializationMixin
+    FrozenBaseModel, pydantic_utils.YAMLSerializationMixin
 ):
     """Class for pipeline run configurations."""
 

src/zenml/config/pipeline_spec.py

@@ -18,11 +18,11 @@
 
 from zenml.config.source import Source, SourceWithValidator
 from zenml.config.step_configurations import StepSpec
-from zenml.config.strict_base_model import StrictBaseModel
+from zenml.config.strict_base_model import FrozenBaseModel
 from zenml.utils.json_utils import pydantic_encoder
 
 
-class PipelineSpec(StrictBaseModel):
+class PipelineSpec(FrozenBaseModel):
     """Specification of a pipeline."""
 
     # Versions:

src/zenml/config/retry_config.py

@@ -13,10 +13,10 @@
 #  permissions and limitations under the License.
 """Retry configuration for a step."""
 
-from zenml.config.strict_base_model import StrictBaseModel
+from zenml.config.strict_base_model import FrozenBaseModel
 
 
-class StepRetryConfig(StrictBaseModel):
+class StepRetryConfig(FrozenBaseModel):
     """Retry configuration for a step.
 
     Delay is an integer (specified in seconds).

src/zenml/config/step_configurations.py

@@ -40,9 +40,9 @@
 from zenml.client_lazy_loader import ClientLazyLoader
 from zenml.config.base_settings import BaseSettings, SettingsOrDict
 from zenml.config.constants import DOCKER_SETTINGS_KEY, RESOURCE_SETTINGS_KEY
+from zenml.config.frozen_base_model import FrozenBaseModel
 from zenml.config.retry_config import StepRetryConfig
 from zenml.config.source import Source, SourceWithValidator
-from zenml.config.strict_base_model import StrictBaseModel
 from zenml.logger import get_logger
 from zenml.model.lazy_load import ModelVersionDataLazyLoader
 from zenml.model.model import Model
@@ -56,7 +56,7 @@
 logger = get_logger(__name__)
 
 
-class PartialArtifactConfiguration(StrictBaseModel):
+class PartialArtifactConfiguration(FrozenBaseModel):
     """Class representing a partial input/output artifact configuration."""
 
     materializer_source: Optional[Tuple[SourceWithValidator, ...]] = None
@@ -140,7 +140,7 @@ def _convert_source(
         return value
 
 
-class StepConfigurationUpdate(StrictBaseModel):
+class StepConfigurationUpdate(FrozenBaseModel):
     """Class for step configuration updates."""
 
     enable_cache: Optional[bool] = None
@@ -263,14 +263,14 @@ def _get_full_substitutions(
         return ret
 
 
-class InputSpec(StrictBaseModel):
+class InputSpec(FrozenBaseModel):
     """Step input specification."""
 
     step_name: str
     output_name: str
 
 
-class StepSpec(StrictBaseModel):
+class StepSpec(FrozenBaseModel):
     """Specification of a pipeline."""
 
     source: SourceWithValidator
@@ -309,7 +309,7 @@ def __eq__(self, other: Any) -> bool:
         return NotImplemented
 
 
-class Step(StrictBaseModel):
+class Step(FrozenBaseModel):
     """Class representing a ZenML step."""
 
     spec: StepSpec

src/zenml/config/step_run_info.py

@@ -18,10 +18,10 @@
 
 from zenml.config.pipeline_configurations import PipelineConfiguration
 from zenml.config.step_configurations import StepConfiguration
-from zenml.config.strict_base_model import StrictBaseModel
+from zenml.config.strict_base_model import FrozenBaseModel
 
 
-class StepRunInfo(StrictBaseModel):
+class StepRunInfo(FrozenBaseModel):
     """All information necessary to run a step."""
 
     step_run_id: UUID