zenml-nightly 0.82.1.dev20250526__py3-none-any.whl → 0.82.1.dev20250527__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.82.1.dev20250526
+0.82.1.dev20250527

zenml/artifacts/utils.py

@@ -16,6 +16,7 @@
 import base64
 import contextlib
 import os
+import re
 import tempfile
 import zipfile
 from pathlib import Path
@@ -940,6 +941,13 @@ def _load_file_from_artifact_store(
         )
 
 
+def _strip_timestamp_from_multiline_string(
+    input: str,
+    pattern: str = r"^\[?\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2} UTC\]?\s*",
+) -> str:
+    return re.sub(pattern, "", input, flags=re.MULTILINE)
+
+
 # --------------------
 # Model Artifact Utils
 # --------------------

zenml/logging/step_logging.py

@@ -28,6 +28,7 @@ from zenml.artifact_stores import BaseArtifactStore
 from zenml.artifacts.utils import (
     _load_artifact_store,
     _load_file_from_artifact_store,
+    _strip_timestamp_from_multiline_string,
 )
 from zenml.constants import (
     ENV_ZENML_DISABLE_STEP_NAMES_IN_LOGS,
@@ -117,6 +118,7 @@ def fetch_logs(
     logs_uri: str,
     offset: int = 0,
     length: int = 1024 * 1024 * 16,  # Default to 16MiB of data
+    strip_timestamp: bool = False,
 ) -> str:
     """Fetches the logs from the artifact store.
 
@@ -126,6 +128,7 @@ def fetch_logs(
         logs_uri: The URI of the artifact.
         offset: The offset from which to start reading.
         length: The amount of bytes that should be read.
+        strip_timestamp: Whether to strip timestamps in logs or not
 
     Returns:
         The logs as a string.
@@ -136,9 +139,12 @@ def fetch_logs(
     """
 
     def _read_file(
-        uri: str, offset: int = 0, length: Optional[int] = None
+        uri: str,
+        offset: int = 0,
+        length: Optional[int] = None,
+        strip_timestamp: bool = False,
     ) -> str:
-        return str(
+        file_content = str(
             _load_file_from_artifact_store(
                 uri,
                 artifact_store=artifact_store,
@@ -147,16 +153,22 @@ def fetch_logs(
                 length=length,
             ).decode()
         )
+        if strip_timestamp:
+            file_content = _strip_timestamp_from_multiline_string(file_content)
+        return file_content
 
     artifact_store = _load_artifact_store(artifact_store_id, zen_store)
     try:
         if not artifact_store.isdir(logs_uri):
-            return _read_file(logs_uri, offset, length)
+            return _read_file(logs_uri, offset, length, strip_timestamp)
         else:
             files = artifact_store.listdir(logs_uri)
             if len(files) == 1:
                 return _read_file(
-                    os.path.join(logs_uri, str(files[0])), offset, length
+                    os.path.join(logs_uri, str(files[0])),
+                    offset,
+                    length,
+                    strip_timestamp,
                 )
             else:
                 is_negative_offset = offset < 0
@@ -191,6 +203,7 @@ def fetch_logs(
                             os.path.join(logs_uri, str(file)),
                             offset,
                             length,
+                            strip_timestamp,
                         )
                     )
                     offset = 0

zenml/zen_server/cloud_utils.py

@@ -40,7 +40,7 @@ class ZenMLCloudConnection:
         method: str,
         endpoint: str,
         params: Optional[Dict[str, Any]] = None,
-        data: Optional[Dict[str, Any]] = None,
+        data: Any = None,
     ) -> requests.Response:
         """Send a request using the active session.
 
@@ -132,7 +132,7 @@ class ZenMLCloudConnection:
         self,
         endpoint: str,
         params: Optional[Dict[str, Any]] = None,
-        data: Optional[Dict[str, Any]] = None,
+        data: Any = None,
     ) -> requests.Response:
         """Send a POST request using the active session.
 

zenml/zen_server/rbac/zenml_cloud_rbac.py

@@ -64,11 +64,12 @@ class ZenMLCloudRBAC(RBACInterface):
 
         params = {
             "user_id": str(user.external_user_id),
-            "resources": resources,
             "action": str(action),
         }
-        response = self._connection.get(
-            endpoint=PERMISSIONS_ENDPOINT, params=params
+        response = self._connection.post(
+            endpoint=PERMISSIONS_ENDPOINT,
+            params=params,
+            data=[str(resource) for resource in resources],
         )
         value = response.json()
 

zenml/zen_server/routers/steps_endpoints.py

@@ -251,6 +251,7 @@ def get_step_logs(
     step_id: UUID,
     offset: int = 0,
     length: int = 1024 * 1024 * 16,  # Default to 16MiB of data
+    strip_timestamp: bool = False,
     _: AuthContext = Security(authorize),
 ) -> str:
     """Get the logs of a specific step.
@@ -259,6 +260,7 @@ def get_step_logs(
         step_id: ID of the step for which to get the logs.
         offset: The offset from which to start reading.
         length: The amount of bytes that should be read.
+        strip_timestamp: Whether to strip the timestamp in logs or not.
 
     Returns:
         The logs of the step.
@@ -282,4 +284,5 @@ def get_step_logs(
         logs_uri=logs.uri,
         offset=offset,
         length=length,
+        strip_timestamp=strip_timestamp,
     )

{zenml_nightly-0.82.1.dev20250526.dist-info → zenml_nightly-0.82.1.dev20250527.dist-info}/METADATA

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

{zenml_nightly-0.82.1.dev20250526.dist-info → zenml_nightly-0.82.1.dev20250527.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=iltxrjUF8slUWnlzIsLSiY9uZW7U-jny3VhsedJzMXU,19
+zenml/VERSION,sha256=-wGxpLHmRzljSa9KO0HqAlwwblx2bkKt1yCBHUi-kA0,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
@@ -25,7 +25,7 @@ zenml/artifacts/external_artifact.py,sha256=7nLANV0vsGC36H1s_B_awX4hnZgXHCGIscQ2
 zenml/artifacts/external_artifact_config.py,sha256=P172p0JOu8Xx1F8RCQut1dnRpt5lpWXGNqMbY-V90sI,3323
 zenml/artifacts/preexisting_data_materializer.py,sha256=dcahDcHUD3Lvn0-6zE2BG84bkyo_ydAgzBWxtbyJJZQ,3325
 zenml/artifacts/unmaterialized_artifact.py,sha256=JNPKq_sNifQx5wP8jEw7TGBEi26zwKirPGlWX9uxbJI,1300
-zenml/artifacts/utils.py,sha256=G7V0L3Pii1b3eKznhCfGt7CJzBuhHoYwQr8Jx5VSLAI,35255
+zenml/artifacts/utils.py,sha256=JEsqAKXsfJoK7TP2ycaVaTBu0O4UwDmQzFTpjoWEyqU,35468
 zenml/cli/__init__.py,sha256=nxq4ifwLV5qT7Qghb42h02XUOcOihBkZp2xBqgiykM8,75670
 zenml/cli/annotator.py,sha256=JRR7_TJOWKyiKGv1kwSjG1Ay6RBWPVgm0X-D0uSBlyE,6976
 zenml/cli/artifact.py,sha256=7lsAS52DroBTFkFWxkyb-lIDOGP5jPL_Se_RDG_2jgg,9564
@@ -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=MB-BwnQ7sPhztL_B2Nosxg3Tdp1cv1lz8qgR0jah62Y,20420
+zenml/logging/step_logging.py,sha256=GApmyuFGybkj1rFbv8Ej3fNym2P4HzGKyUBKFu2YO90,20907
 zenml/login/__init__.py,sha256=Evi7hq8tpUn57IM3iX3hYP0r8oIeEWUhS471TAOyVGs,644
 zenml/login/credentials.py,sha256=RtLmYkFQ5trbprhsO8NCRKwBA136KzGoUWY52dZP9GA,12722
 zenml/login/credentials_store.py,sha256=k55mNSqc_RaBuvLWtg5ubQAwlSbaTxTVAqoZ1OQ2YcM,24750
@@ -805,7 +805,7 @@ zenml/utils/yaml_utils.py,sha256=RvEr-N84wwG1Aq8rfdPAP7lev_SEaSQlqcL9fB_vHk8,584
 zenml/zen_server/__init__.py,sha256=WyltI9TzFW2mEHZVOs6alLWMCQrrZaFALtrQXs83STA,1355
 zenml/zen_server/auth.py,sha256=sJ_UAG58Qa0sUx8P1QkWmwp5zfAafFnm8WBHAOnPi4w,40070
 zenml/zen_server/cache.py,sha256=Tc4TSugmsU1bhThxlYfE8rv0KmltIX1CcVHgzrJ0Eus,6633
-zenml/zen_server/cloud_utils.py,sha256=vKsjWadgPctZDtfeG9qJC-5gKOir_tZsglYMBkC8QqU,11874
+zenml/zen_server/cloud_utils.py,sha256=aHuMn3-oAsl-glrnv_JUgc5AE0m6Ep5lSLNhXx7F-eU,11832
 zenml/zen_server/csrf.py,sha256=Jsbi_IKriWCOuquSYTOEWqSXiGORJATIhR5Wrkm4XzQ,2684
 zenml/zen_server/dashboard/assets/404-_AtuLtaX.js,sha256=Y2jAo7K2feru-k7QcMxjnfUNkDrO7xv341YyKKpjZiU,1033
 zenml/zen_server/dashboard/assets/@radix-C7hRs6Kx.js,sha256=cNA9UX8LGrKUQubGrls3E3Wq1fCrlK51W14yh22FE_U,296700
@@ -1025,7 +1025,7 @@ zenml/zen_server/rbac/models.py,sha256=Xhz0XqmVqg3ji9uTwjlhQ4IuQ2ivrT4gVdHi5ZkjF
 zenml/zen_server/rbac/rbac_interface.py,sha256=VPMNnUIPcqJWDngITX6cfaPlM6zJwzCApXAnT_oaxSQ,3431
 zenml/zen_server/rbac/rbac_sql_zen_store.py,sha256=ZBjAbCUlhMsp8GZiIkULl2ztJ8K2Wf-0QdSxzQXwovM,5930
 zenml/zen_server/rbac/utils.py,sha256=09r1dSZ04hcag21rg6Ug2fMCdYfNjSBnx_W8C336GzE,24254
-zenml/zen_server/rbac/zenml_cloud_rbac.py,sha256=L1Tio4IcNNxNWnVpAqBWYgdVfrqR20ddvdhWAo1i3f4,6055
+zenml/zen_server/rbac/zenml_cloud_rbac.py,sha256=fQRQsHYZsifjAAYAZALqiSB8ZjqIxjF1FfnXcyU7HF8,6093
 zenml/zen_server/routers/__init__.py,sha256=ViyAhWL-ogHxE9wBvB_iMcur5H1NRVrzXkpogVY7FBA,641
 zenml/zen_server/routers/actions_endpoints.py,sha256=-zzeTawwTrkWOnp8ltrGS_tUnWnyiXMFNaBRXI3zddk,9500
 zenml/zen_server/routers/artifact_endpoint.py,sha256=d-5yPQO4N9O2W97j0f_5fghNThnNe0izT9WpZLaJyuA,5054
@@ -1055,7 +1055,7 @@ zenml/zen_server/routers/service_endpoints.py,sha256=37CY-22h4hTscEkBByKJqdHcOWs
 zenml/zen_server/routers/stack_components_endpoints.py,sha256=MBQ3iz29WN1n9T_BLpd2BtWxWoLIBP6QT7BfaAkGEMY,8580
 zenml/zen_server/routers/stack_deployment_endpoints.py,sha256=r7N4UaDAwj9ACYfdyUofRHpNPRrzWl-vmGoSErN9oYI,5381
 zenml/zen_server/routers/stacks_endpoints.py,sha256=ci-WZ774Q2T3-XN6cZARJ95wmRUW2-W1mh7gKlbNE54,7527
-zenml/zen_server/routers/steps_endpoints.py,sha256=KZpVsEsycLMmiEGnwT2EkmQvaTIIyL4QOUJBXMu7NSo,8239
+zenml/zen_server/routers/steps_endpoints.py,sha256=L-hnCASvhI4H5wmJwWuuHjX-yCuIOGIVuf0gw1fceLo,8387
 zenml/zen_server/routers/tag_resource_endpoints.py,sha256=54HBt31whSj7wjzhw9Xzb6aJLBDVPd5iyg5fJ2Ej5qU,3268
 zenml/zen_server/routers/tags_endpoints.py,sha256=H0aXms8iAkcxo9lsBwpDrVhJnY5n2cqBd3Iz1OI8r_k,4545
 zenml/zen_server/routers/triggers_endpoints.py,sha256=Z9ry01MMpCZ2wcjcvJjnEu1eDhkcGACKuBL_F9m8smI,10100
@@ -1319,8 +1319,8 @@ zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=LPFW757WCJLP1S8vrvjs
 zenml/zen_stores/sql_zen_store.py,sha256=jsx_1_MJZE4dnNXxj8bAMX8pT5JfoJliADSBF9n6kJc,444809
 zenml/zen_stores/template_utils.py,sha256=GbJ7LgGVYHSCKPEA8RNTxPoVTWqpC77F_lGzjJ4O1Fw,9220
 zenml/zen_stores/zen_store_interface.py,sha256=fF_uL_FplnvGvM5o3jOQ8i1zHXhuhKLL2n4nvIKSR7E,92090
-zenml_nightly-0.82.1.dev20250526.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.82.1.dev20250526.dist-info/METADATA,sha256=F8-F75rBPo2RV4tMmjh_trUmdf3JNT1d8GgXNu1qvcM,24317
-zenml_nightly-0.82.1.dev20250526.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-zenml_nightly-0.82.1.dev20250526.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.82.1.dev20250526.dist-info/RECORD,,
+zenml_nightly-0.82.1.dev20250527.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.82.1.dev20250527.dist-info/METADATA,sha256=iB0qEGvEML3Ql26mih_FjKfI34M9Lh8W_ao9jWBXn9E,24317
+zenml_nightly-0.82.1.dev20250527.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+zenml_nightly-0.82.1.dev20250527.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.82.1.dev20250527.dist-info/RECORD,,