zenml-nightly 0.84.1.dev20250731__py3-none-any.whl → 0.84.1.dev20250802__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.84.1.dev20250731
+0.84.1.dev20250802

zenml/integrations/wandb/flavors/wandb_experiment_tracker_flavor.py

@@ -43,20 +43,19 @@ class WandbExperimentTrackerSettings(BaseSettings):
     """Settings for the Wandb experiment tracker."""
 
     run_name: Optional[str] = Field(
-        None,
-        description="The Wandb run name to use for tracking experiments."
+        None, description="The Wandb run name to use for tracking experiments."
     )
     tags: List[str] = Field(
         default_factory=list,
-        description="Tags to attach to the Wandb run for categorization and filtering."
+        description="Tags to attach to the Wandb run for categorization and filtering.",
     )
     settings: Dict[str, Any] = Field(
         default_factory=dict,
-        description="Additional settings for the Wandb run configuration."
+        description="Additional settings for the Wandb run configuration.",
     )
     enable_weave: bool = Field(
         False,
-        description="Whether to enable Weave integration for enhanced experiment tracking."
+        description="Whether to enable Weave integration for enhanced experiment tracking.",
     )
 
     @field_validator("settings", mode="before")
@@ -73,7 +72,10 @@ class WandbExperimentTrackerSettings(BaseSettings):
         Returns:
             Dict representation of the settings.
         """
-        import wandb
+        try:
+            import wandb
+        except ImportError:
+            return value
 
         if isinstance(value, wandb.Settings):
             # Depending on the wandb version, either `model_dump`,
@@ -103,12 +105,12 @@ class WandbExperimentTrackerConfig(
     entity: Optional[str] = Field(
         None,
         description="Name of an existing Wandb entity (team or user account) "
-        "to log experiments to."
+        "to log experiments to.",
     )
     project_name: Optional[str] = Field(
         None,
         description="Name of an existing Wandb project to log experiments to. "
-        "If not specified, a default project will be used."
+        "If not specified, a default project will be used.",
     )
 
 

zenml/logging/step_logging.py

@@ -335,11 +335,11 @@ class PipelineLogsStorage:
         except Exception as e:
             logger.error("Error in log storage thread: %s", e)
         finally:
-            # Always mark all queue tasks as done
             for _ in messages:
                 self.log_queue.task_done()
 
-            time.sleep(self.write_interval)
+            # Wait for the next write interval or until shutdown is requested
+            self.shutdown_event.wait(timeout=self.write_interval)
 
     def _log_storage_worker(self) -> None:
         """Log storage thread worker that processes the log queue."""

zenml/materializers/path_materializer.py

@@ -29,6 +29,30 @@ from zenml.materializers.base_materializer import BaseMaterializer
 from zenml.utils.io_utils import is_path_within_directory
 
 
+def _is_safe_tar_member(member: tarfile.TarInfo, directory: str) -> bool:
+    """Check if a tar member is safe to extract.
+
+    This function validates that the member name and any link targets
+    are within the specified directory to prevent path traversal attacks.
+
+    Args:
+        member: The tar member to validate.
+        directory: The target extraction directory.
+
+    Returns:
+        True if the member is safe to extract, False otherwise.
+    """
+    # Check if the member name is within the directory
+    if not is_path_within_directory(member.name, directory):
+        return False
+
+    # For symbolic links and hard links, validate the target path
+    if member.issym() or member.islnk():
+        return is_path_within_directory(member.linkname, directory)
+
+    return True
+
+
 class PathMaterializer(BaseMaterializer):
     """Materializer for Path objects.
 
@@ -73,14 +97,14 @@ class PathMaterializer(BaseMaterializer):
                 # Extract the archive to the temporary directory
                 with tarfile.open(archive_path_local, "r:gz") as tar:
                     # Validate archive members to prevent path traversal attacks
-                    # Filter members to only those with safe paths
+                    # Filter members to only those with safe paths and link targets
                     safe_members = []
                     for member in tar.getmembers():
-                        if is_path_within_directory(member.name, directory):
+                        if _is_safe_tar_member(member, directory):
                             safe_members.append(member)
 
                     # Extract only safe members
-                    tar.extractall(path=directory, members=safe_members)  # nosec B202 - members are filtered through is_path_within_directory
+                    tar.extractall(path=directory, members=safe_members)  # nosec B202 - members are filtered through _is_safe_tar_member
 
                 # Clean up the archive file
                 os.remove(archive_path_local)

{zenml_nightly-0.84.1.dev20250731.dist-info → zenml_nightly-0.84.1.dev20250802.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zenml-nightly
-Version: 0.84.1.dev20250731
+Version: 0.84.1.dev20250802
 Summary: ZenML: Write production-ready ML code.
 License: Apache-2.0
 Keywords: machine learning,production,pipeline,mlops,devops

{zenml_nightly-0.84.1.dev20250731.dist-info → zenml_nightly-0.84.1.dev20250802.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=MuVwyZh81WpS0uMQqm5VrJ4CtGw4yagkVBb2bI5GXmY,19
+zenml/VERSION,sha256=myLpU8v50Vgzy97KA02rMlaQZilIphIXQaCoYw3pEQ4,19
 zenml/__init__.py,sha256=r7JUg2SVDf_dPhS7iU6vudKusEqK4ics7_jFMZhq0o4,2731
 zenml/actions/__init__.py,sha256=mrt6wPo73iKRxK754_NqsGyJ3buW7RnVeIGXr1xEw8Y,681
 zenml/actions/base_action.py,sha256=UcaHev6BTuLDwuswnyaPjdA8AgUqB5xPZ-lRtuvf2FU,25553
@@ -554,7 +554,7 @@ zenml/integrations/wandb/__init__.py,sha256=5aTIc27MeYzODDGeJHZmtRhTSNkg7Vt2tHgr
 zenml/integrations/wandb/experiment_trackers/__init__.py,sha256=8nFyyvh-PTF5d9ZfjS7xFSWTWSpreRB1azePv-Ex2sc,771
 zenml/integrations/wandb/experiment_trackers/wandb_experiment_tracker.py,sha256=GV5zDPgj6Dh3ho2MMUC1Da1ezPrNtr4RE9tisWGde00,5749
 zenml/integrations/wandb/flavors/__init__.py,sha256=b4oJHyCdMN98XB-8S-Pnv39HA-oXQWpup6eZmCmIAEY,894
-zenml/integrations/wandb/flavors/wandb_experiment_tracker_flavor.py,sha256=uBQl9bzGXgSXn4B8cVA41QY9T4CLr2MWieTSokevyfM,5341
+zenml/integrations/wandb/flavors/wandb_experiment_tracker_flavor.py,sha256=Y-_YVWEOvKyQzCRXVxtdRNqHJukeMZf6qxsWaweekMY,5408
 zenml/integrations/whylogs/__init__.py,sha256=PROjw-4-SDSpzxfoqxKud8vGnJm-HsrWxMumjGQuF3M,2423
 zenml/integrations/whylogs/constants.py,sha256=Txs7qQjmj4vuoqC6rJvoBJ-4yv41CrapExG0_5TvEpw,752
 zenml/integrations/whylogs/data_validators/__init__.py,sha256=cLblrK_3Hckc_p8YjqJir28V9Nx_-pFPEIknjodQNQQ,820
@@ -578,7 +578,7 @@ zenml/io/filesystem_registry.py,sha256=stujDg4a5k983WMwp3rj4Z4puiUco4REyVoIoMIpI
 zenml/io/local_filesystem.py,sha256=xQTZPT5cpooptUV8KiifxZojS6pWCv1-6UUxitUYb_E,7386
 zenml/logger.py,sha256=STfebaOdCGmw404gwKhwpDrrhtKDUREeerO7UtapzfE,6997
 zenml/logging/__init__.py,sha256=knhroJ2h0uHBCGzAiBBGJEiuhEA3cwI6XYBRIyXdbkQ,613
-zenml/logging/step_logging.py,sha256=BTFDAigkDj66KAwizdfMxXCXYArzwaypxAkCHwCPlwI,27717
+zenml/logging/step_logging.py,sha256=PTAHlqNfWF-Ly0zvBxroQ3S9vOMEhjSZTzi0cmZVKqQ,27767
 zenml/login/__init__.py,sha256=Evi7hq8tpUn57IM3iX3hYP0r8oIeEWUhS471TAOyVGs,644
 zenml/login/credentials.py,sha256=RtLmYkFQ5trbprhsO8NCRKwBA136KzGoUWY52dZP9GA,12722
 zenml/login/credentials_store.py,sha256=k55mNSqc_RaBuvLWtg5ubQAwlSbaTxTVAqoZ1OQ2YcM,24750
@@ -602,7 +602,7 @@ zenml/materializers/cloudpickle_materializer.py,sha256=PIZauXg1SBYmBUi9FBd2t3LFy
 zenml/materializers/materializer_registry.py,sha256=ic-aWhJ2Ex9F_rml2dDVAxhRfW3nd71QMxzfTPP6BIM,4002
 zenml/materializers/numpy_materializer.py,sha256=OLcHF9Z0tAqQ_U8TraA0vGmZjHoT7eT_XevncIutt0M,1715
 zenml/materializers/pandas_materializer.py,sha256=c4B-ly04504gysA66iCYcmEdeh0ClePRTxRCkmHqIgE,1725
-zenml/materializers/path_materializer.py,sha256=kY37mVwlfQndp3M7--g81dRt2qT81MJ1aURnl4uX29I,5422
+zenml/materializers/path_materializer.py,sha256=ydfKHhi14acjSb9jqizsXRRO4wKaCsSNUJg2woPw9XY,6217
 zenml/materializers/pydantic_materializer.py,sha256=eDp3eOR-X7FOHlpwALOJtVUtJti75Dsa7r0lzSjeQ-Q,2271
 zenml/materializers/service_materializer.py,sha256=OV5HFUuIc8UF8vE5y_FTWn4_mpCz0AtJPnNA2YFAjZE,2992
 zenml/materializers/structured_string_materializer.py,sha256=8LIvSH3JQn_QRAWGXK5_NhPQ4NgCt5zzz6f8GnGG9q4,4442
@@ -1352,8 +1352,8 @@ zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=LPFW757WCJLP1S8vrvjs
 zenml/zen_stores/sql_zen_store.py,sha256=YTd_YNqPvnlc6pnIO9zQpUZH9sAGH-r0Qr635u-7Ays,491254
 zenml/zen_stores/template_utils.py,sha256=iCXrXpqzVTY7roqop4Eh9J7DmLW6PQeILZexmw_l3b8,10074
 zenml/zen_stores/zen_store_interface.py,sha256=weiSULdI9AsbCE10a5TcwtybX-BJs9hKhjPJnTapWv4,93023
-zenml_nightly-0.84.1.dev20250731.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.84.1.dev20250731.dist-info/METADATA,sha256=e8WBVJaGZK2z2VGw-rHRSSBk5oSsy4n61NpNVjCnKk4,24296
-zenml_nightly-0.84.1.dev20250731.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-zenml_nightly-0.84.1.dev20250731.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.84.1.dev20250731.dist-info/RECORD,,
+zenml_nightly-0.84.1.dev20250802.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.84.1.dev20250802.dist-info/METADATA,sha256=6uK60EoPuvOWhVBa2L-1H0aj4GF5ls3OP1Sc-Q6F8Yc,24296
+zenml_nightly-0.84.1.dev20250802.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+zenml_nightly-0.84.1.dev20250802.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.84.1.dev20250802.dist-info/RECORD,,