zenml-nightly 0.74.0.dev20250212__py3-none-any.whl → 0.74.0.dev20250213__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.74.0.dev20250212
+0.74.0.dev20250213

zenml/pipelines/pipeline_definition.py

@@ -757,10 +757,16 @@ To avoid this consider setting pipeline parameters only in one place (config or
             build=build_model,
             can_download_from_code_repository=can_download_from_code_repository,
         ):
-            code_archive = code_utils.CodeArchive(
-                root=source_utils.get_source_root()
+            source_root = source_utils.get_source_root()
+            code_archive = code_utils.CodeArchive(root=source_root)
+            logger.info(
+                "Archiving pipeline code directory: `%s`. If this is taking "
+                "longer than you expected, make sure your source root "
+                "is set correctly by running `zenml init`, and that it "
+                "does not contain unnecessarily huge files.",
+                source_root,
             )
-            logger.info("Archiving pipeline code...")
+
             code_path = code_utils.upload_code_if_necessary(code_archive)
 
         request = PipelineDeploymentRequest(

zenml/utils/source_utils.py

@@ -275,6 +275,35 @@ def resolve(
     )
 
 
+def get_implicit_source_root() -> str:
+    """Get the implicit source root (the parent directory of the main module).
+
+    Raises:
+        RuntimeError: If the main module file can't be found.
+
+    Returns:
+        The implicit source root.
+    """
+    main_module = sys.modules.get("__main__")
+    if main_module is None:
+        raise RuntimeError(
+            "Unable to determine source root because the main module could not "
+            "be found."
+        )
+
+    if not hasattr(main_module, "__file__") or not main_module.__file__:
+        raise RuntimeError(
+            "Unable to determine source root because the main module does not "
+            "have an associated file. This could be because you're running in "
+            "an interactive Python environment. If you are trying to run from "
+            "within a Jupyter notebook, please run `zenml init` from the root "
+            "where your notebook is located and restart your notebook server."
+        )
+
+    path = Path(main_module.__file__).resolve().parent
+    return str(path)
+
+
 def get_source_root() -> str:
     """Get the source root.
 
@@ -286,9 +315,6 @@ def get_source_root() -> str:
 
     Returns:
         The source root.
-
-    Raises:
-        RuntimeError: If the main module file can't be found.
     """
     if _CUSTOM_SOURCE_ROOT:
         logger.debug("Using custom source root: %s", _CUSTOM_SOURCE_ROOT)
@@ -301,26 +327,12 @@ def get_source_root() -> str:
         logger.debug("Using repository root as source root: %s", repo_root)
         return str(repo_root.resolve())
 
-    main_module = sys.modules.get("__main__")
-    if main_module is None:
-        raise RuntimeError(
-            "Unable to determine source root because the main module could not "
-            "be found."
-        )
-
-    if not hasattr(main_module, "__file__") or not main_module.__file__:
-        raise RuntimeError(
-            "Unable to determine source root because the main module does not "
-            "have an associated file. This could be because you're running in "
-            "an interactive Python environment. If you are trying to run from "
-            "within a Jupyter notebook, please run `zenml init` from the root "
-            "where your notebook is located and restart your notebook server.   "
-        )
-
-    path = Path(main_module.__file__).resolve().parent
-
-    logger.debug("Using main module parent directory as source root: %s", path)
-    return str(path)
+    implicit_source_root = get_implicit_source_root()
+    logger.debug(
+        "Using main module parent directory as source root: %s",
+        implicit_source_root,
+    )
+    return implicit_source_root
 
 
 def set_custom_source_root(source_root: Optional[str]) -> None:

{zenml_nightly-0.74.0.dev20250212.dist-info → zenml_nightly-0.74.0.dev20250213.dist-info}/METADATA

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

{zenml_nightly-0.74.0.dev20250212.dist-info → zenml_nightly-0.74.0.dev20250213.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=UhG6I4o4NTBc6VFy231HJr7TSwPCzeLflYctAPsY-y8,19
+zenml/VERSION,sha256=m5UFAlfM2bHdwaLw8Ttt42SvpS5yV6jrrfa_bMCNvyA,19
 zenml/__init__.py,sha256=SkMObQA41ajqdZqGErN00S1Vf3KAxpLvbZ-OBy5uYoo,2130
 zenml/actions/__init__.py,sha256=mrt6wPo73iKRxK754_NqsGyJ3buW7RnVeIGXr1xEw8Y,681
 zenml/actions/base_action.py,sha256=UcaHev6BTuLDwuswnyaPjdA8AgUqB5xPZ-lRtuvf2FU,25553
@@ -695,7 +695,7 @@ zenml/pipelines/__init__.py,sha256=hpIX7hN8jsQRHT5R-xSXZL88qrHwkmrvGLQeu1rWt4o,8
 zenml/pipelines/build_utils.py,sha256=DkID1YnRYkw569uU-gidpF8WR8E7K55_wn3CzRPQ3Cs,27562
 zenml/pipelines/pipeline_context.py,sha256=V_p-F9W7cBIlTjS0iv5-uJYMzaOj8bAUkc_uNhQgBms,3579
 zenml/pipelines/pipeline_decorator.py,sha256=FIbflYOMavbuyGmqsx3F5zZgg0oXMTi1eAcGXciljOs,4293
-zenml/pipelines/pipeline_definition.py,sha256=Pi4OGs7-WXbsm8_zczY5lpQEyLJPqkKZHjXAKfnk3HY,56936
+zenml/pipelines/pipeline_definition.py,sha256=pl5tV4_h03MQDaIJZQCJJ8mY5S3Js-89r5NFIBpxB6A,57244
 zenml/pipelines/run_utils.py,sha256=4KuHIQFtLXTZNQBScTEkIG5pqNtu6xGm6UZT7ptyyKs,11623
 zenml/plugins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zenml/plugins/base_plugin_flavor.py,sha256=88IxFW91UB_rQ8xPlfRnIhIJh7A308NEq2epMMdlOng,2530
@@ -787,7 +787,7 @@ zenml/utils/secret_utils.py,sha256=gEvqnhzAZwPO6mdOQWvioeH-xLoSObfaNRzt17N8zyU,5
 zenml/utils/settings_utils.py,sha256=lAK13CiDCDkcLygizDbWB9q-9ukteVBJPypzFCrne9k,4631
 zenml/utils/singleton.py,sha256=uFRrUlUdS5VyY9lLJyl_n5kqppsqJLKkBhSj4g5VPkY,2757
 zenml/utils/source_code_utils.py,sha256=8iyNA2MGIORYVEkSdxNTXfS1ZdFKXTAG1dZRkeQtPL0,3751
-zenml/utils/source_utils.py,sha256=ow3fbodjmAfpwmUXBSg3oyibHk065aSvi4UV9uJD6Yk,26736
+zenml/utils/source_utils.py,sha256=joKLghhDq9dh0fd8B0WRGX-nN-uwnGQdgmsyY_n-8gY,27033
 zenml/utils/string_utils.py,sha256=VPuAwFzxwXXo1cKmBWxHnoeTyDX35XPf_lXGiDMXPEw,7254
 zenml/utils/time_utils.py,sha256=-9Y9zwJ-6Gv7hoZQCoftPyC2LCLo2bYj6OgdyBaE44o,4076
 zenml/utils/typed_model.py,sha256=00EAo1I1VnOBHG4-ce8dPkyHRPpgi67SRIU-KdewRWs,4757
@@ -1280,8 +1280,8 @@ zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=nEO0bAPlULBLxLVk-UTR
 zenml/zen_stores/sql_zen_store.py,sha256=tKOghX2Wa0f0xSYu_ReqHgCwslvHrBhE3RePJG2tcA0,415921
 zenml/zen_stores/template_utils.py,sha256=EKYBgmDLTS_PSMWaIO5yvHPLiQvMqHcsAe6NUCrv-i4,9068
 zenml/zen_stores/zen_store_interface.py,sha256=vf2gKBWfUUPtcGZC35oQB6pPNVzWVyQC8nWxVLjfrxM,92692
-zenml_nightly-0.74.0.dev20250212.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.74.0.dev20250212.dist-info/METADATA,sha256=FzxLNILSrBhOG7JLb1Gs6gaiP3LOibyLTJWhRHHKIUQ,21430
-zenml_nightly-0.74.0.dev20250212.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-zenml_nightly-0.74.0.dev20250212.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.74.0.dev20250212.dist-info/RECORD,,
+zenml_nightly-0.74.0.dev20250213.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.74.0.dev20250213.dist-info/METADATA,sha256=FfHRQAkvQcU-SpjOdUwnsKp1KURqE1eBLVm5L6Pd67M,21430
+zenml_nightly-0.74.0.dev20250213.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+zenml_nightly-0.74.0.dev20250213.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.74.0.dev20250213.dist-info/RECORD,,