zenml-nightly 0.80.1.dev20250403__py3-none-any.whl → 0.80.1.dev20250404__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.80.1.dev20250403
+0.80.1.dev20250404

zenml/client.py

@@ -1213,7 +1213,6 @@ class Client(metaclass=ClientMetaClass):
             name=name,
             components=stack_components,
             stack_spec_path=stack_spec_file,
-            project=self.active_project.id,
             labels=labels,
         )
 
@@ -1339,7 +1338,6 @@ class Client(metaclass=ClientMetaClass):
 
         # Create the update model
         update_model = StackUpdate(
-            project=self.active_project.id,
             stack_spec_path=stack_spec_file,
         )
 
@@ -2027,7 +2025,6 @@ class Client(metaclass=ClientMetaClass):
             type=component_type,
             flavor=flavor,
             configuration=configuration,
-            project=self.active_project.id,
             labels=labels,
         )
 
@@ -2075,9 +2072,7 @@ class Client(metaclass=ClientMetaClass):
             allow_name_prefix_match=False,
         )
 
-        update_model = ComponentUpdate(
-            project=self.active_project.id,
-        )
+        update_model = ComponentUpdate()
 
         if name is not None:
             existing_components = self.list_stack_components(

zenml/constants.py

@@ -169,6 +169,7 @@ ENV_ZENML_SERVER = "ZENML_SERVER"
 ENV_ZENML_ENFORCE_TYPE_ANNOTATIONS = "ZENML_ENFORCE_TYPE_ANNOTATIONS"
 ENV_ZENML_ENABLE_IMPLICIT_AUTH_METHODS = "ZENML_ENABLE_IMPLICIT_AUTH_METHODS"
 ENV_ZENML_DISABLE_STEP_LOGS_STORAGE = "ZENML_DISABLE_STEP_LOGS_STORAGE"
+ENV_ZENML_DISABLE_STEP_NAMES_IN_LOGS = "ZENML_DISABLE_STEP_NAMES_IN_LOGS"
 ENV_ZENML_IGNORE_FAILURE_HOOK = "ZENML_IGNORE_FAILURE_HOOK"
 ENV_ZENML_CUSTOM_SOURCE_ROOT = "ZENML_CUSTOM_SOURCE_ROOT"
 ENV_ZENML_WHEEL_PACKAGE_NAME = "ZENML_WHEEL_PACKAGE_NAME"

zenml/integrations/s3/__init__.py

@@ -31,17 +31,10 @@ class S3Integration(Integration):
     NAME = S3
     # boto3 isn't required for the filesystem to work, but it is required
     # for the AWS/S3 connector that can be used with the artifact store.
-    # NOTE: to keep the dependency resolution for botocore consistent and fast
-    # between s3fs and boto3, the boto3 upper version used here should be the
-    # same as the one resolved by pip when installing boto3 without a
-    # restriction alongside s3fs, e.g.:
-    #
-    #   pip install 's3fs>2022.3.0,<=2023.4.0' boto3
-    #
-    # The above command installs boto3==1.26.76, so we use the same version
-    # here to avoid the dependency resolution overhead.
     REQUIREMENTS = [
-        "s3fs>2022.3.0",
+        # Explicitly exclude 2025.3.1 to avoid pulling in the yanked fsspec
+        # package with the same version
+        "s3fs>2022.3.0,!=2025.3.1",
         "boto3",
         # The following dependencies are only required for the AWS connector.
         "aws-profile-manager",

zenml/integrations/wandb/flavors/wandb_experiment_tracker_flavor.py

@@ -75,7 +75,7 @@ class WandbExperimentTrackerSettings(BaseSettings):
             # `make_static` or `to_dict` is available to convert the settings
             # to a dictionary
             if isinstance(value, BaseModel):
-                return value.model_dump()
+                return value.model_dump()  # type: ignore[no-untyped-call]
             elif hasattr(value, "make_static"):
                 return cast(Dict[str, Any], value.make_static())
             elif hasattr(value, "to_dict"):

zenml/logging/step_logging.py

@@ -22,11 +22,16 @@ from types import TracebackType
 from typing import Any, Callable, List, Optional, Type, Union
 from uuid import UUID, uuid4
 
+from zenml import get_step_context
 from zenml.artifact_stores import BaseArtifactStore
 from zenml.artifacts.utils import (
     _load_artifact_store,
     _load_file_from_artifact_store,
 )
+from zenml.constants import (
+    ENV_ZENML_DISABLE_STEP_NAMES_IN_LOGS,
+    handle_bool_env_var,
+)
 from zenml.exceptions import DoesNotExistException
 from zenml.logger import get_logger
 from zenml.logging import (
@@ -447,8 +452,8 @@ class StepLogsStorageContext:
         setattr(sys.stdout, "write", self._wrap_write(self.stdout_write))
         setattr(sys.stdout, "flush", self._wrap_flush(self.stdout_flush))
 
-        setattr(sys.stderr, "write", self._wrap_write(self.stdout_write))
-        setattr(sys.stderr, "flush", self._wrap_flush(self.stdout_flush))
+        setattr(sys.stderr, "write", self._wrap_write(self.stderr_write))
+        setattr(sys.stderr, "flush", self._wrap_flush(self.stderr_flush))
 
         redirected.set(True)
         return self
@@ -494,9 +499,32 @@ class StepLogsStorageContext:
         """
 
         def wrapped_write(*args: Any, **kwargs: Any) -> Any:
-            output = method(*args, **kwargs)
+            # Check if step names in logs are disabled via env var
+            step_names_disabled = handle_bool_env_var(
+                ENV_ZENML_DISABLE_STEP_NAMES_IN_LOGS, default=False
+            )
+
+            if step_names_disabled:
+                output = method(*args, **kwargs)
+            else:
+                # Try to get step context if not available yet
+                step_context = None
+                try:
+                    step_context = get_step_context()
+                except Exception:
+                    pass
+
+                if step_context and args[0] != "\n":
+                    message = f"[{step_context.step_name}] " + args[0]
+                else:
+                    message = args[0]
+
+                output = method(message, *args[1:], **kwargs)
+
+            # Save the original message without step name prefix to storage
             if args:
                 self.storage.write(args[0])
+
             return output
 
         return wrapped_write

{zenml_nightly-0.80.1.dev20250403.dist-info → zenml_nightly-0.80.1.dev20250404.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zenml-nightly
-Version: 0.80.1.dev20250403
+Version: 0.80.1.dev20250404
 Summary: ZenML: Write production-ready ML code.
 License: Apache-2.0
 Keywords: machine learning,production,pipeline,mlops,devops
@@ -108,7 +108,7 @@ Requires-Dist: pyyaml-include (<2.0) ; extra == "templates"
 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) ; extra == "s3fs"
+Requires-Dist: s3fs (>=2022.11.0,!=2025.3.1) ; extra == "s3fs"
 Requires-Dist: sagemaker (>=2.199.0) ; extra == "sagemaker"
 Requires-Dist: secure (>=0.3.0,<0.4.0) ; extra == "server"
 Requires-Dist: setuptools

{zenml_nightly-0.80.1.dev20250403.dist-info → zenml_nightly-0.80.1.dev20250404.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=aWwg-fBqgZ9oDR6dhHJc5o2EySW_O7NhHPKAZD1QVv4,19
+zenml/VERSION,sha256=_fy45dCRG5xzujlDy8BA-AChK8IcC5EBU-v8u74a1mc,19
 zenml/__init__.py,sha256=CKEyepFK-7akXYiMrNVh92Nb01Cjs23w4_YyI6sgdc8,2242
 zenml/actions/__init__.py,sha256=mrt6wPo73iKRxK754_NqsGyJ3buW7RnVeIGXr1xEw8Y,681
 zenml/actions/base_action.py,sha256=UcaHev6BTuLDwuswnyaPjdA8AgUqB5xPZ-lRtuvf2FU,25553
@@ -55,7 +55,7 @@ zenml/cli/text_utils.py,sha256=bY1GIjoULt1cW2FyrPlMoAXNS2R7cSOjDFEZQqrpVQ8,3553
 zenml/cli/user_management.py,sha256=sNnhaUxH-cHecbZBR1L0mEU0TnLNZHzI6ZBCUSQa7OY,13078
 zenml/cli/utils.py,sha256=6Bv2-gqkHC8DdUT4b_HL6pgGMErY8lwjuJb-eHgPTJA,86383
 zenml/cli/version.py,sha256=nm1iSU_1V6-MUwpMKeXcwFhLYGUMLswvQL67cEuCpxA,3635
-zenml/client.py,sha256=l8nHsajF_Ctn3ne4unrn7OHW-VG4OKOKObt-POYFZzE,292402
+zenml/client.py,sha256=QHcrJ9d5_UrXvhFl4lrLcFllFSiWX8vO2K4Sf4lfJpc,292217
 zenml/client_lazy_loader.py,sha256=MOBgS1ITYqGvPUnWQ6edn9s8Hr_72YfWbwEIfHKUr9g,7104
 zenml/code_repositories/__init__.py,sha256=W5bDfzAG8OXIKZSV1L-VHuzMcSCYa9qzTdPb3jqfyYw,920
 zenml/code_repositories/base_code_repository.py,sha256=Id6VjbUu8N3ZpNvBGhOgbahtoMiCAtYXed3G7YQ_iAc,5225
@@ -85,7 +85,7 @@ zenml/config/step_run_info.py,sha256=KiVRSTtKmZ1GbvseDTap2imr7XwMHD3jSFVpyLNEK1I
 zenml/config/store_config.py,sha256=Cla5p5dTB6nNlo8_OZDs9hod5hspi64vxwtZj882XgU,3559
 zenml/config/strict_base_model.py,sha256=iHnO9qOmLUP_eiy9IjRr3JjIs1l1I_CsRQ76EyAneYU,860
 zenml/console.py,sha256=hj_KerPQKwnyKACj0ehSqUQX0mGVCJBKE1QvCt6ik3A,1160
-zenml/constants.py,sha256=qK9iTSw_KNijkPIrotmKdMnY9QrlSCNKDnJ8RHO48TM,15855
+zenml/constants.py,sha256=2HKlfa8nG6FNqwg6gdvR5pMkOvAC-uZW6nJpmg9aBQ8,15929
 zenml/container_registries/__init__.py,sha256=ZSPbBIOnzhg88kQSpYgKe_POLuru14m629665-kAVAA,2200
 zenml/container_registries/azure_container_registry.py,sha256=t1sfDa94Vzbyqtb1iPFNutJ2EXV5_p9CUNITasoiQ70,2667
 zenml/container_registries/base_container_registry.py,sha256=6c2e32wuqxYHJXm5OV2LY1MtX9yopB7WZtes9fmTAz0,7625
@@ -449,7 +449,7 @@ zenml/integrations/pytorch_lightning/__init__.py,sha256=dgbBAEaMiXi1MIZcF58aG3vC
 zenml/integrations/pytorch_lightning/materializers/__init__.py,sha256=zZsRri9X8YMJy4_h-8_Uss5RHjkFkd-RGF9zKnd713Q,814
 zenml/integrations/pytorch_lightning/materializers/pytorch_lightning_materializer.py,sha256=dmNXbGLAzbiaFEa2K9ctOiOoyuQ2GNVsDfgcvdV14Is,1245
 zenml/integrations/registry.py,sha256=02WjPZVrTWdAKHUxEGDMVFud7IrJBDEVXdmWEh1GQBc,8383
-zenml/integrations/s3/__init__.py,sha256=UkZ4_-z-2yxW1SQjAJ0LvGLGIUrhjq16mhZ0PB6J2g8,2179
+zenml/integrations/s3/__init__.py,sha256=1Xjc_1ntyMx0aOPXtnS4feNnH30U-pokxnN9RWF5hRY,1842
 zenml/integrations/s3/artifact_stores/__init__.py,sha256=xK8pbJIg66f7BXlJfyCKVRgxxD7KmdAxiuXGtFvenVQ,793
 zenml/integrations/s3/artifact_stores/s3_artifact_store.py,sha256=0KD2b2-Fi3w5I4tnVh2XnmczwIcsNIjkOHCWGRO3KZE,18059
 zenml/integrations/s3/flavors/__init__.py,sha256=ReHjEsoH5edSLYTE6iFGiF_8bPxN_8qL0B8tIXM5tbY,849
@@ -553,7 +553,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=2Sszs0E8-AfMrVwdVSsVRBA85OdttSYx7jb69WxpMs0,4854
+zenml/integrations/wandb/flavors/wandb_experiment_tracker_flavor.py,sha256=uRsPtRAM_A5ktZqiHLE6OU1nF8_4Vrtc8LcdVv12b-4,4887
 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
@@ -577,7 +577,7 @@ zenml/io/filesystem_registry.py,sha256=stujDg4a5k983WMwp3rj4Z4puiUco4REyVoIoMIpI
 zenml/io/local_filesystem.py,sha256=xQTZPT5cpooptUV8KiifxZojS6pWCv1-6UUxitUYb_E,7386
 zenml/logger.py,sha256=LMV2sMFQ-6JK9xEn6kEt1C9vAoQ0lwuHVM5XHIeKubE,6966
 zenml/logging/__init__.py,sha256=lnqbOa31wAHwPP5f8vZazOrUwnP2QviLiIVwxoAefD8,975
-zenml/logging/step_logging.py,sha256=KQN9SZO5CQSASctP8OXymBmgNyNNfDT9QkbNwGV-QKM,18272
+zenml/logging/step_logging.py,sha256=ELd_Qs9i_aqnchM9nM8KW9md-9UJyesd3vXoetIj1u4,19228
 zenml/login/__init__.py,sha256=Evi7hq8tpUn57IM3iX3hYP0r8oIeEWUhS471TAOyVGs,644
 zenml/login/credentials.py,sha256=RtLmYkFQ5trbprhsO8NCRKwBA136KzGoUWY52dZP9GA,12722
 zenml/login/credentials_store.py,sha256=m5MaImmQleEi79S0ogPZIVeYvMhWcPO1UWuYns4Zr8E,23673
@@ -1310,8 +1310,8 @@ zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=nEO0bAPlULBLxLVk-UTR
 zenml/zen_stores/sql_zen_store.py,sha256=ldyC1uhMnmX5ojnqY9d_L2S-iC-eaNUwsexTkdPtqr4,440204
 zenml/zen_stores/template_utils.py,sha256=GWBP5QEOyvhzndS_MLPmvh28sQaOPpPoZFXCIX9CRL4,9065
 zenml/zen_stores/zen_store_interface.py,sha256=fF_uL_FplnvGvM5o3jOQ8i1zHXhuhKLL2n4nvIKSR7E,92090
-zenml_nightly-0.80.1.dev20250403.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.80.1.dev20250403.dist-info/METADATA,sha256=p2L6CBN_1i1UQ4BoPZS7fAbYx6DR-8cXP4TQcsZ3TQQ,24219
-zenml_nightly-0.80.1.dev20250403.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-zenml_nightly-0.80.1.dev20250403.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.80.1.dev20250403.dist-info/RECORD,,
+zenml_nightly-0.80.1.dev20250404.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.80.1.dev20250404.dist-info/METADATA,sha256=tZHAegNMoGaj7n2PZJanp4r7EyAASLcDbyMZYo4ui04,24230
+zenml_nightly-0.80.1.dev20250404.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+zenml_nightly-0.80.1.dev20250404.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.80.1.dev20250404.dist-info/RECORD,,