zenml-nightly 0.82.1.dev20250519__py3-none-any.whl → 0.82.1.dev20250521__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
zenml/VERSION CHANGED
@@ -1 +1 @@
1
- 0.82.1.dev20250519
1
+ 0.82.1.dev20250521
zenml/config/schedule.py CHANGED
@@ -108,7 +108,7 @@ class Schedule(BaseModel):
108
108
  if self.cron_expression and periodic_schedule:
109
109
  logger.warning(
110
110
  "This schedule was created with a cron expression as well as "
111
- "values for `start_time` and `interval_seconds`. The resulting "
111
+ "values for `start_time` and `interval_second`. The resulting "
112
112
  "behavior depends on the concrete orchestrator implementation "
113
113
  "but will usually ignore the interval and use the cron "
114
114
  "expression."
@@ -414,7 +414,8 @@ class LocalDaemonService(BaseService):
414
414
  p = psutil.Process(pid)
415
415
  except psutil.Error:
416
416
  logger.error(
417
- "Could not find process for for service '%s' ...", self
417
+ "Could not find process for service '%s' ...",
418
+ self,
418
419
  )
419
420
  return
420
421
  if force:
zenml/utils/yaml_utils.py CHANGED
@@ -14,6 +14,7 @@
14
14
  """Utility functions to help with YAML files and data."""
15
15
 
16
16
  import json
17
+ import os
17
18
  from pathlib import Path
18
19
  from typing import Any, Dict, List, Optional, Type, Union
19
20
  from uuid import UUID
@@ -99,9 +100,8 @@ def is_yaml(file_path: str) -> bool:
99
100
  Returns:
100
101
  True if is yaml, else False.
101
102
  """
102
- if file_path.endswith("yaml") or file_path.endswith("yml"):
103
- return True
104
- return False
103
+ extension = os.path.splitext(file_path)[1].lower()
104
+ return extension in (".yaml", ".yml")
105
105
 
106
106
 
107
107
  def comment_out_yaml(yaml_string: str) -> str:
zenml/zen_server/auth.py CHANGED
@@ -434,17 +434,10 @@ def authenticate_credentials(
434
434
  run does not exist.
435
435
  """
436
436
  try:
437
- pipeline_run = zen_store().get_run(
438
- pipeline_run_id, hydrate=True
439
- )
437
+ return zen_store().get_run_status(pipeline_run_id)
440
438
  except KeyError:
441
439
  return None, None
442
440
 
443
- return (
444
- pipeline_run.status,
445
- pipeline_run.end_time,
446
- )
447
-
448
441
  (
449
442
  pipeline_run_status,
450
443
  pipeline_run_end_time,
@@ -5103,6 +5103,26 @@ class SqlZenStore(BaseZenStore):
5103
5103
  include_python_packages=include_python_packages,
5104
5104
  )
5105
5105
 
5106
+ def get_run_status(
5107
+ self,
5108
+ run_id: UUID,
5109
+ ) -> Tuple[ExecutionStatus, Optional[datetime]]:
5110
+ """Gets the status of a pipeline run.
5111
+
5112
+ Args:
5113
+ run_id: The ID of the pipeline run to get.
5114
+
5115
+ Returns:
5116
+ The pipeline run status and end time.
5117
+ """
5118
+ with Session(self.engine) as session:
5119
+ run = self._get_schema_by_id(
5120
+ resource_id=run_id,
5121
+ schema_class=PipelineRunSchema,
5122
+ session=session,
5123
+ )
5124
+ return ExecutionStatus(run.status), run.end_time
5125
+
5106
5126
  def _replace_placeholder_run(
5107
5127
  self,
5108
5128
  pipeline_run: PipelineRunRequest,
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: zenml-nightly
3
- Version: 0.82.1.dev20250519
3
+ Version: 0.82.1.dev20250521
4
4
  Summary: ZenML: Write production-ready ML code.
5
5
  License: Apache-2.0
6
6
  Keywords: machine learning,production,pipeline,mlops,devops
@@ -1,5 +1,5 @@
1
1
  zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
2
- zenml/VERSION,sha256=_XBA0qWJ6UNVd3u0Smo0uXRJdBdtHQoNXtxMF9BwAVU,19
2
+ zenml/VERSION,sha256=FX-9_pQVSQnGY2Fhoo6Eck1dVTnyq1TCpchQLE2aZAI,19
3
3
  zenml/__init__.py,sha256=CKEyepFK-7akXYiMrNVh92Nb01Cjs23w4_YyI6sgdc8,2242
4
4
  zenml/actions/__init__.py,sha256=mrt6wPo73iKRxK754_NqsGyJ3buW7RnVeIGXr1xEw8Y,681
5
5
  zenml/actions/base_action.py,sha256=UcaHev6BTuLDwuswnyaPjdA8AgUqB5xPZ-lRtuvf2FU,25553
@@ -74,7 +74,7 @@ zenml/config/pipeline_run_configuration.py,sha256=Y9C5mVUH-w4gc1w1PCQFpjEmfBBpSM
74
74
  zenml/config/pipeline_spec.py,sha256=uWpiIfznJvXAiKs1yMIUDT1h1tYEFNO-RDVTYcIv9CE,2821
75
75
  zenml/config/resource_settings.py,sha256=0taXGHvDfXuvpMplxsuzpznB1sAvWJIGnXoVJ9-ySfw,3899
76
76
  zenml/config/retry_config.py,sha256=4UH1xqw0G6fSEbXSNKfmiFEkwadxQef9BGMe3JBm6NI,929
77
- zenml/config/schedule.py,sha256=qtMWa-mEo7jIKvDzQUstMwe57gdbvyWAQ7ggsoddbCA,5349
77
+ zenml/config/schedule.py,sha256=-83j99U9OyiG7E322XWA7QvuLSwQzF21whwpeiF0b30,5348
78
78
  zenml/config/secret_reference_mixin.py,sha256=YvY68MTd1gE23IVprf0BLkNn62hoxcvb5nqGgc8jMkU,5871
79
79
  zenml/config/secrets_store_config.py,sha256=y05zqyQhr_DGrs3IfBGa_FRoZ043hSYRT5wzrx-zHTU,2818
80
80
  zenml/config/server_config.py,sha256=or-LKYFl-9yE3MEq_yPa5aCCu3d5_f0tL9MqPtb_3-c,31852
@@ -725,7 +725,7 @@ zenml/services/container/container_service_endpoint.py,sha256=BbiWfXlibqIlizszOz
725
725
  zenml/services/container/entrypoint.py,sha256=flEuPHXZbeaiKUagapq2FKJxI8Ne3PKDKxezTjtihcs,3272
726
726
  zenml/services/local/__init__.py,sha256=0Ve3QQtblDD4aB9kvs8qbvN-osu-0l1uCYAY6-aeJHs,660
727
727
  zenml/services/local/local_daemon_entrypoint.py,sha256=LNXsLOaqYWVMM1LP2a0mwAGaP-kYybUyWIDJmRAlnlo,3369
728
- zenml/services/local/local_service.py,sha256=W-GlmOo6rgBnjrEfzfc5vL9KbUkVv1Xkwh4WJSj2R9Q,17485
728
+ zenml/services/local/local_service.py,sha256=mIr3YE78_0UtlDbXxwu51zVkbJIEnwfOr0HvY_xLmVc,17498
729
729
  zenml/services/local/local_service_endpoint.py,sha256=sayvh8_uFWb4bXd9bm4YygdQgUGZIx9y8-iFwgwj1ec,4976
730
730
  zenml/services/service.py,sha256=ygvS2mECXKWbj6oeY125LqagLo8JaIpxnjETbE5q_IA,18465
731
731
  zenml/services/service_endpoint.py,sha256=NFs2Y3mDC8yL3PGYq-CHBFarFMs_wNuSTGK_TWIYqBA,6464
@@ -801,9 +801,9 @@ zenml/utils/typed_model.py,sha256=00EAo1I1VnOBHG4-ce8dPkyHRPpgi67SRIU-KdewRWs,47
801
801
  zenml/utils/typing_utils.py,sha256=jP7JKrlLsuMIStXhwKFNWykE6SMOR72tJIJ_qEbQSNc,6555
802
802
  zenml/utils/uuid_utils.py,sha256=aOGQ2SdREexcVQICPU2jUAgjvAJxTmh4ESdM52PEhck,2045
803
803
  zenml/utils/visualization_utils.py,sha256=ZGAnmE4s_rBSyz2FBu9FOIu054blfXuxLEz8wWqv-3s,4739
804
- zenml/utils/yaml_utils.py,sha256=747M_BTf3lcHFgJDF8RJxnnGIbCU8e9YxBBMmoQ5O_U,5834
804
+ zenml/utils/yaml_utils.py,sha256=RvEr-N84wwG1Aq8rfdPAP7lev_SEaSQlqcL9fB_vHk8,5840
805
805
  zenml/zen_server/__init__.py,sha256=WyltI9TzFW2mEHZVOs6alLWMCQrrZaFALtrQXs83STA,1355
806
- zenml/zen_server/auth.py,sha256=5EDqBb0HZJ8zr6OhwiTUcWkqoBFIEl3vfyit232UoOo,40259
806
+ zenml/zen_server/auth.py,sha256=sJ_UAG58Qa0sUx8P1QkWmwp5zfAafFnm8WBHAOnPi4w,40070
807
807
  zenml/zen_server/cache.py,sha256=Tc4TSugmsU1bhThxlYfE8rv0KmltIX1CcVHgzrJ0Eus,6633
808
808
  zenml/zen_server/cloud_utils.py,sha256=qKMtHjnkVfAk_yxF2K9iC0TkMCjp8s0gdz_cEsbB4DE,10908
809
809
  zenml/zen_server/csrf.py,sha256=Jsbi_IKriWCOuquSYTOEWqSXiGORJATIhR5Wrkm4XzQ,2684
@@ -1316,11 +1316,11 @@ zenml/zen_stores/secrets_stores/hashicorp_secrets_store.py,sha256=5err1a-TrV3SR5
1316
1316
  zenml/zen_stores/secrets_stores/secrets_store_interface.py,sha256=Q2Jbnt2Pp7NGlR-u1YBfRZV2g8su2Fd0ArBMdksAE-Q,2819
1317
1317
  zenml/zen_stores/secrets_stores/service_connector_secrets_store.py,sha256=S87ne23D08PAwtfRVlVnBn8R0ilTpEh6r8blauNV5WQ,6941
1318
1318
  zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=LPFW757WCJLP1S8vrvjsrl2Tf1yo281xUTjSBsos4qk,8788
1319
- zenml/zen_stores/sql_zen_store.py,sha256=CK0owrZvLV9u8N0Fa-JrPld-aXxBDncX5MlUexjTIFk,441321
1319
+ zenml/zen_stores/sql_zen_store.py,sha256=oUB9u3tMVmRrSxslC1oEN5kHQa8i7g4bNEpJKgsYmqY,441911
1320
1320
  zenml/zen_stores/template_utils.py,sha256=GbJ7LgGVYHSCKPEA8RNTxPoVTWqpC77F_lGzjJ4O1Fw,9220
1321
1321
  zenml/zen_stores/zen_store_interface.py,sha256=fF_uL_FplnvGvM5o3jOQ8i1zHXhuhKLL2n4nvIKSR7E,92090
1322
- zenml_nightly-0.82.1.dev20250519.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
1323
- zenml_nightly-0.82.1.dev20250519.dist-info/METADATA,sha256=oJSSiIfKV77OF6LA0AJiMuPf4YiBU7L3gciaf0m6nzE,24317
1324
- zenml_nightly-0.82.1.dev20250519.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
1325
- zenml_nightly-0.82.1.dev20250519.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
1326
- zenml_nightly-0.82.1.dev20250519.dist-info/RECORD,,
1322
+ zenml_nightly-0.82.1.dev20250521.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
1323
+ zenml_nightly-0.82.1.dev20250521.dist-info/METADATA,sha256=0YoGPYbCHpQHx_aj7HeAvDvkV3GGQIWqJOn5UFaitPs,24317
1324
+ zenml_nightly-0.82.1.dev20250521.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
1325
+ zenml_nightly-0.82.1.dev20250521.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
1326
+ zenml_nightly-0.82.1.dev20250521.dist-info/RECORD,,