zenml-nightly 0.84.1.dev20250806__py3-none-any.whl → 0.84.2.dev20250808__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.84.1.dev20250806
+0.84.2.dev20250808

zenml/integrations/huggingface/__init__.py

@@ -54,7 +54,7 @@ class HuggingfaceIntegration(Integration):
             "accelerate",
             "bitsandbytes>=0.41.3",
             "peft",
-            "transformers",
+            "transformers<4.55.0",
         ]
 
         # Add the pandas integration requirements

zenml/integrations/hyperai/__init__.py

@@ -28,7 +28,7 @@ class HyperAIIntegration(Integration):
 
     NAME = HYPERAI
     REQUIREMENTS = [
-        "paramiko>=3.4.0",
+        "paramiko>=3.4.0,<4.0.0",
     ]
 
     @classmethod

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/zen_stores/migrations/versions/0.84.2_release.py

@@ -0,0 +1,23 @@
+"""Release [0.84.2].
+
+Revision ID: 0.84.2
+Revises: 0.84.1
+Create Date: 2025-08-06 07:52:42.915891
+
+"""
+
+# revision identifiers, used by Alembic.
+revision = "0.84.2"
+down_revision = "0.84.1"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    """Upgrade database schema and/or data, creating a new revision."""
+    pass
+
+
+def downgrade() -> None:
+    """Downgrade database schema and/or data back to the previous revision."""
+    pass

{zenml_nightly-0.84.1.dev20250806.dist-info → zenml_nightly-0.84.2.dev20250808.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zenml-nightly
-Version: 0.84.1.dev20250806
+Version: 0.84.2.dev20250808
 Summary: ZenML: Write production-ready ML code.
 License: Apache-2.0
 Keywords: machine learning,production,pipeline,mlops,devops
@@ -125,7 +125,7 @@ Requires-Dist: types-PyYAML (>=6.0.0,<7.0.0) ; extra == "dev"
 Requires-Dist: types-certifi (>=2021.10.8.0,<2022.0.0.0) ; extra == "dev"
 Requires-Dist: types-croniter (>=1.0.2,<2.0.0) ; extra == "dev"
 Requires-Dist: types-futures (>=3.3.1,<4.0.0) ; extra == "dev"
-Requires-Dist: types-paramiko (>=3.4.0) ; extra == "dev"
+Requires-Dist: types-paramiko (>=3.4.0,<4.0.0) ; extra == "dev"
 Requires-Dist: types-passlib (>=1.7.7,<2.0.0) ; extra == "dev"
 Requires-Dist: types-protobuf (>=3.18.0,<4.0.0) ; extra == "dev"
 Requires-Dist: types-psutil (>=5.8.13,<6.0.0) ; extra == "dev"
@@ -168,7 +168,7 @@ Description-Content-Type: text/markdown
 [pypi-shield]: https://img.shields.io/pypi/pyversions/zenml?color=281158
 [pypi-url]: https://pypi.org/project/zenml/
 [pypiversion-shield]: https://img.shields.io/pypi/v/zenml?color=361776
-[downloads-shield]: https://img.shields.io/pypi/dm/zenml?color=431D93
+[downloads-shield]: https://img.shields.io/pepy/dt/zenml?color=431D93
 [downloads-url]: https://pypi.org/project/zenml/
 [contributors-shield]: https://img.shields.io/github/contributors/zenml-io/zenml?color=7A3EF4
 [contributors-url]: https://github.com/zenml-io/zenml/graphs/contributors

{zenml_nightly-0.84.1.dev20250806.dist-info → zenml_nightly-0.84.2.dev20250808.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=KMpAeP_9YhseXa7vHadCSXfPfoKYBJW_PucPxE3onyI,19
+zenml/VERSION,sha256=BH0oSd6m3Nunr0JQEu5sZIhNu4KwT7PTv5ms8cHaL9M,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
@@ -297,7 +297,7 @@ zenml/integrations/great_expectations/steps/__init__.py,sha256=OGsp32yJs9GItypFR
 zenml/integrations/great_expectations/steps/ge_profiler.py,sha256=ea6WLF1B8pvkGe-dBaAX3tNV2W8mVhUhk6WQpKgqKEA,2141
 zenml/integrations/great_expectations/steps/ge_validator.py,sha256=kdFzhkzJtQZGOul8E8BE5_315mYGvMuDINYagPflKVk,2946
 zenml/integrations/great_expectations/utils.py,sha256=4DXjAfsKUVcp_lSGAPiAsAI-WLNjr_DLMsGJOYGkSjE,3138
-zenml/integrations/huggingface/__init__.py,sha256=sFV6Wg5E6tauYfIhWwXgfk0F_jlqWmFXWfBHQmwKwo0,2667
+zenml/integrations/huggingface/__init__.py,sha256=BkIpcmQIDFG-2m21GJ5OmRynOVFvE5vL2UEiC9nYrNA,2674
 zenml/integrations/huggingface/flavors/__init__.py,sha256=NXMxZXrS7fHdZnz1G_Sf83k4zkE84C5UoYJzxXSY-R0,970
 zenml/integrations/huggingface/flavors/huggingface_model_deployer_flavor.py,sha256=QITTxFrpKu5JNH29A_riAWiC0-gY3qcxGWQf__0aQII,4032
 zenml/integrations/huggingface/materializers/__init__.py,sha256=HoiSCzfMTxtcvkDBconFm_-pdGZXzXDelkuPtcrJIgA,1267
@@ -313,7 +313,7 @@ zenml/integrations/huggingface/services/huggingface_deployment.py,sha256=m-8qKca
 zenml/integrations/huggingface/steps/__init__.py,sha256=tjsmnE9lJcXsE46YGPNWJICHXDUWn34m7jE-os-CFLo,879
 zenml/integrations/huggingface/steps/accelerate_runner.py,sha256=0fazQcqopAVRCInNePsIEy2Vn4EePioAetWXhewXpFA,6387
 zenml/integrations/huggingface/steps/huggingface_deployer.py,sha256=Wk1wzYwy-kpA-IJAHzukUpIGRxrhZTuyXNYVriXBqCI,4103
-zenml/integrations/hyperai/__init__.py,sha256=6ed5rgRgiRr2Ksi3DDFUaNQwPkCfE4BRSnX58bKFshg,1663
+zenml/integrations/hyperai/__init__.py,sha256=wEi-7px6yKIfcqqswxoWrwdH1P5txMNp6iXsAoizuaU,1670
 zenml/integrations/hyperai/flavors/__init__.py,sha256=PUGBPmQ7y3H7QU2zAj7Ri0rrUBbOWnM_L59AIVWUYwU,800
 zenml/integrations/hyperai/flavors/hyperai_orchestrator_flavor.py,sha256=r4w16s4IB72nBKEgvSlJ6TRZ88kyP8mduOUZpQ-njn4,5328
 zenml/integrations/hyperai/orchestrators/__init__.py,sha256=kSYpMZPEWwNu2vxoOC6PeyQ9RLzsPAgTHxL35K36MiE,784
@@ -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
@@ -1191,6 +1191,7 @@ zenml/zen_stores/migrations/versions/0.83.0_release.py,sha256=Y3Pe9I_LJvUgTtehyD
 zenml/zen_stores/migrations/versions/0.83.1_release.py,sha256=p35ZcMR4k5zsPU6sya0fHLDDjKF-mgwNvpMcxbI7cQo,462
 zenml/zen_stores/migrations/versions/0.84.0_release.py,sha256=Lpg4Xqm-CfchmY3sFyMDrC-J1pAoaSz3gGlPSP4eXJQ,462
 zenml/zen_stores/migrations/versions/0.84.1_release.py,sha256=95TBBvlSduocvY5Hy6yT1QjK8Ml2l5JVcTYVNy33om4,462
+zenml/zen_stores/migrations/versions/0.84.2_release.py,sha256=igIzI5EJzwiUurUzgrGAI8jLZd48KL2TWZ3nC-yT388,450
 zenml/zen_stores/migrations/versions/026d4577b6a0_add_code_path.py,sha256=hXLzvQcylNrbCVD6vha52PFkSPNC2klW9kA0vuQX_cE,1091
 zenml/zen_stores/migrations/versions/03742aa7fdd7_add_secrets.py,sha256=gewKqu1AnzvNTjVvK1eaAwP0hVneWDUyDRSLTvRCdpg,1587
 zenml/zen_stores/migrations/versions/0392807467dc_add_build_duration.py,sha256=YlkDBlfBBv45FsrMO11YcdRn4Maqmlg77t8gWJO4DfA,982
@@ -1354,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.1.dev20250806.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.84.1.dev20250806.dist-info/METADATA,sha256=VZaHyV--b2EcSDdhFPJE0noToZKb34x6Cq6CKNDsSk4,24296
-zenml_nightly-0.84.1.dev20250806.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-zenml_nightly-0.84.1.dev20250806.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.84.1.dev20250806.dist-info/RECORD,,
+zenml_nightly-0.84.2.dev20250808.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.84.2.dev20250808.dist-info/METADATA,sha256=BtymD3zSN6xneRM23MaVs-g30o7X0sNysTxKEDtkea0,24303
+zenml_nightly-0.84.2.dev20250808.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+zenml_nightly-0.84.2.dev20250808.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.84.2.dev20250808.dist-info/RECORD,,