zenml-nightly 0.81.0.dev20250428__py3-none-any.whl → 0.81.0.dev20250430__py3-none-any.whl

zenml/zen_stores/sql_zen_store.py

@@ -790,7 +790,7 @@ class SqlZenStoreConfiguration(StoreConfiguration):
         # of the certificate files.
         validate_assignment=False,
         # Forbid extra attributes set in the class.
-        extra="forbid",
+        extra="ignore",
     )
 
 
@@ -5074,7 +5074,10 @@ class SqlZenStore(BaseZenStore):
         return new_run.to_model(include_metadata=True, include_resources=True)
 
     def get_run(
-        self, run_id: UUID, hydrate: bool = True
+        self,
+        run_id: UUID,
+        hydrate: bool = True,
+        include_python_packages: bool = False,
     ) -> PipelineRunResponse:
         """Gets a pipeline run.
 
@@ -5082,6 +5085,8 @@ class SqlZenStore(BaseZenStore):
             run_id: The ID of the pipeline run to get.
             hydrate: Flag deciding whether to hydrate the output model(s)
                 by including metadata fields in the response.
+            include_python_packages: Flag deciding whether to include the
+                python packages in the response.
 
         Returns:
             The pipeline run.
@@ -5093,7 +5098,9 @@ class SqlZenStore(BaseZenStore):
                 session=session,
             )
             return run.to_model(
-                include_metadata=hydrate, include_resources=True
+                include_metadata=hydrate,
+                include_resources=True,
+                include_python_packages=include_python_packages,
             )
 
     def _replace_placeholder_run(
@@ -10773,32 +10780,6 @@ class SqlZenStore(BaseZenStore):
                 include_metadata=hydrate, include_resources=True
             )
 
-    def _set_filter_model_id(
-        self,
-        filter_model: ModelVersionFilter,
-        session: Session,
-    ) -> None:
-        """Set the model ID on a filter model.
-
-        Args:
-            filter_model: The filter model to set the model ID on.
-            session: The DB session to use to use for queries.
-
-        Raises:
-            ValueError: if the filter is not scoped to a model.
-        """
-        if filter_model.model:
-            model = self._get_schema_by_name_or_id(
-                object_name_or_id=filter_model.model,
-                schema_class=ModelSchema,
-                project_name_or_id=filter_model.project,
-                session=session,
-            )
-        else:
-            raise ValueError("Model ID missing from the filter")
-
-        filter_model.model = model.id
-
     def list_model_versions(
         self,
         model_version_filter_model: ModelVersionFilter,
@@ -10820,11 +10801,6 @@ class SqlZenStore(BaseZenStore):
                 filter_model=model_version_filter_model,
                 session=session,
             )
-            self._set_filter_model_id(
-                filter_model=model_version_filter_model,
-                session=session,
-            )
-
             query = select(ModelVersionSchema)
 
             return self.filter_and_paginate(

{zenml_nightly-0.81.0.dev20250428.dist-info → zenml_nightly-0.81.0.dev20250430.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zenml-nightly
-Version: 0.81.0.dev20250428
+Version: 0.81.0.dev20250430
 Summary: ZenML: Write production-ready ML code.
 License: Apache-2.0
 Keywords: machine learning,production,pipeline,mlops,devops
@@ -110,7 +110,7 @@ Requires-Dist: requests (>=2.27.11,<3.0.0) ; extra == "connectors-azure"
 Requires-Dist: rich[jupyter] (>=12.0.0)
 Requires-Dist: ruff (>=0.1.7) ; extra == "templates" or extra == "dev"
 Requires-Dist: s3fs (>=2022.11.0,!=2025.3.1) ; extra == "s3fs"
-Requires-Dist: sagemaker (>=2.199.0) ; extra == "sagemaker"
+Requires-Dist: sagemaker (>=2.237.3) ; extra == "sagemaker"
 Requires-Dist: secure (>=0.3.0,<0.4.0) ; extra == "server"
 Requires-Dist: setuptools
 Requires-Dist: sqlalchemy (>=2.0.0,<3.0.0)