zenml-nightly 0.84.2.dev20250807__py3-none-any.whl → 0.84.2.dev20250809__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.84.2.dev20250807
+0.84.2.dev20250809

zenml/steps/base_step.py

@@ -271,6 +271,15 @@ class BaseStep:
         """
         return inspect.getsource(self.source_object)
 
+    @property
+    def source_code_cache_value(self) -> str:
+        """The source code cache value of this step.
+
+        Returns:
+            The source code cache value of this step.
+        """
+        return self.source_code
+
     @property
     def docstring(self) -> Optional[str]:
         """The docstring of this step.
@@ -288,9 +297,9 @@ class BaseStep:
             A dictionary containing the caching parameters
         """
         parameters = {
-            CODE_HASH_PARAMETER_NAME: source_code_utils.get_hashed_source_code(
-                self.source_object
-            )
+            CODE_HASH_PARAMETER_NAME: hashlib.sha256(
+                self.source_code_cache_value.encode("utf-8")
+            ).hexdigest()
         }
         for name, output in self.configuration.outputs.items():
             if output.materializer_source:

zenml/steps/decorated_step.py

@@ -13,15 +13,111 @@
 #  permissions and limitations under the License.
 """Internal BaseStep subclass used by the step decorator."""
 
-from typing import Any
+from typing import Any, Optional
 
 from zenml.config.source import Source
-from zenml.steps import BaseStep
+from zenml.steps import BaseStep, step
+
+
+def remove_decorator_from_source_code(
+    source_code: str, decorator_name: str
+) -> str:
+    """Remove a decorator from the source code of a function.
+
+    Args:
+        source_code: The source code of the function.
+        decorator_name: The name of the decorator.
+
+    Returns:
+        The source code of the function without the decorator.
+    """
+    import ast
+
+    # Remove potential indentation from the source code
+    source_code_lines = source_code.split("\n")
+    if source_code_lines:
+        first_line = source_code_lines[0]
+        leading_spaces = len(first_line) - len(first_line.lstrip())
+        source_code_lines = [
+            line[leading_spaces:] if len(line) >= leading_spaces else line
+            for line in source_code_lines
+        ]
+
+    module = ast.parse("\n".join(source_code_lines))
+    if not module.body or not isinstance(module.body[0], ast.FunctionDef):
+        return source_code
+
+    function_def = module.body[0]
+    if not function_def.decorator_list:
+        return source_code
+
+    for decorator in function_def.decorator_list:
+        lineno = None
+        end_lineno = None
+
+        if isinstance(decorator, ast.Call) and isinstance(
+            decorator.func, ast.Name
+        ):
+            if decorator.func.id == decorator_name:
+                lineno = decorator.lineno
+                end_lineno = decorator.end_lineno
+        elif isinstance(decorator, ast.Name):
+            if decorator.id == decorator_name:
+                lineno = decorator.lineno
+                end_lineno = decorator.end_lineno
+        else:
+            continue
+
+        if lineno is not None and end_lineno is not None:
+            # The line numbers are 1-indexed, so we need to
+            # subtract 1
+            source_code_lines = (
+                source_code_lines[: lineno - 1]
+                + source_code_lines[end_lineno:]
+            )
+            source_code = "\n".join(source_code_lines)
+            break
+
+    return source_code
 
 
 class _DecoratedStep(BaseStep):
     """Internal BaseStep subclass used by the step decorator."""
 
+    def _get_step_decorator_name(self) -> Optional[str]:
+        """The name of the step decorator.
+
+        Returns:
+            The name of the step decorator.
+        """
+        decorator_names = [
+            key
+            for key, value in self.entrypoint.__globals__.items()
+            if value is step
+        ]
+        if not decorator_names:
+            return None
+
+        return decorator_names[0]
+
+    @property
+    def source_code_cache_value(self) -> str:
+        """The source code cache value of this step.
+
+        Returns:
+            The source code cache value of this step.
+        """
+        try:
+            if decorator_name := self._get_step_decorator_name():
+                return remove_decorator_from_source_code(
+                    source_code=self.source_code,
+                    decorator_name=decorator_name,
+                )
+        except Exception:
+            pass
+
+        return super().source_code_cache_value
+
     @property
     def source_object(self) -> Any:
         """The source object of this step.

{zenml_nightly-0.84.2.dev20250807.dist-info → zenml_nightly-0.84.2.dev20250809.dist-info}/METADATA

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

{zenml_nightly-0.84.2.dev20250807.dist-info → zenml_nightly-0.84.2.dev20250809.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=5Qbn3wnWDaFIZJ7bj5KdK188yXmxX25h-ppF-ENK_Q4,19
+zenml/VERSION,sha256=-UGFWian4KhYUNm1pRGgxZco7JYlUnH6eaju2_WoJRc,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
@@ -756,8 +756,8 @@ zenml/step_operators/__init__.py,sha256=tqj7fgnQyZubLjwUu4ITwkA-70KMQz4g37agbfF7
 zenml/step_operators/base_step_operator.py,sha256=ZRnY6lcEHY8xZskrKKdPhgKg2BlEoh2_kb8xCaM2D1g,3522
 zenml/step_operators/step_operator_entrypoint_configuration.py,sha256=wuQKmEI_ckn1CHHrxWGGdIhQyBFXG01aKC2-2b6Atjo,3690
 zenml/steps/__init__.py,sha256=KKWFOmCZGLDEikOD2E5YmDA7QHo47uPV37by21WwI0U,1453
-zenml/steps/base_step.py,sha256=6mz8ZBn4wd509KcwZeD1qdxQV4BF3wLknMMzn_g3wgg,44390
-zenml/steps/decorated_step.py,sha256=C8Ng5PCLc9eql4JF1N345HQ6LyC1qCUdTnysUTeoAJs,1315
+zenml/steps/base_step.py,sha256=HUNNQe3A5DyzEHm4gAbX_NCC2LiKRDI6nrngURW6WdI,44632
+zenml/steps/decorated_step.py,sha256=cr4m7qi525fUc_VTkb7J16dKdUSzc4UA5eyHg8vFWQg,4275
 zenml/steps/entrypoint_function_utils.py,sha256=H0WIkHpR_R0S9gl_tWr0nX-fcBlYnm8OQ1cimvrw-qo,9555
 zenml/steps/step_context.py,sha256=sFvVVvEyKmWTrucofor8Cuxv-72jbziVaQY-OlOvFAM,15526
 zenml/steps/step_decorator.py,sha256=0TKxP_oAVRizjN3JhBZ8ShFd3fSBdVPU2ijytTg1h2U,6506
@@ -1355,8 +1355,8 @@ zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=LPFW757WCJLP1S8vrvjs
 zenml/zen_stores/sql_zen_store.py,sha256=-3zeByIUjIvsx3564O2gJ463512QkZl04okL3eB-nJs,491568
 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.2.dev20250807.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.84.2.dev20250807.dist-info/METADATA,sha256=g0HviK0kuLK_THciwuXaQMPDkm8S3qL0hX0GOzmgAvE,24303
-zenml_nightly-0.84.2.dev20250807.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-zenml_nightly-0.84.2.dev20250807.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.84.2.dev20250807.dist-info/RECORD,,
+zenml_nightly-0.84.2.dev20250809.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.84.2.dev20250809.dist-info/METADATA,sha256=mcz4Yi9LcDIaBK_WnBRF3xN4SKiMu0SH932nozCUqeE,24303
+zenml_nightly-0.84.2.dev20250809.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+zenml_nightly-0.84.2.dev20250809.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.84.2.dev20250809.dist-info/RECORD,,