zenml-nightly 0.66.0.dev20240913__py3-none-any.whl → 0.66.0.dev20240915__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.66.0.dev20240913
1
+ 0.66.0.dev20240915
zenml/steps/base_step.py CHANGED
@@ -592,16 +592,34 @@ class BaseStep(metaclass=BaseStepMeta):
592
592
  from zenml.new.pipelines.pipeline import Pipeline
593
593
 
594
594
  if not Pipeline.ACTIVE_PIPELINE:
595
- # The step is being called outside the context of a pipeline, either
596
- # run the step function or run it as a single step pipeline on the
597
- # active stack
595
+ from zenml import constants, get_step_context
596
+
597
+ # If the environment variable was set to explicitly not run on the
598
+ # stack, we do that.
598
599
  run_without_stack = handle_bool_env_var(
599
600
  ENV_ZENML_RUN_SINGLE_STEPS_WITHOUT_STACK, default=False
600
601
  )
601
602
  if run_without_stack:
602
603
  return self.call_entrypoint(*args, **kwargs)
604
+
605
+ try:
606
+ get_step_context()
607
+ except RuntimeError:
608
+ pass
603
609
  else:
604
- return run_as_single_step_pipeline(self, *args, **kwargs)
610
+ # We're currently inside the execution of a different step
611
+ # -> We don't want to launch another single step pipeline here,
612
+ # but instead just call the step function
613
+ return self.call_entrypoint(*args, **kwargs)
614
+
615
+ if constants.SHOULD_PREVENT_PIPELINE_EXECUTION:
616
+ logger.info(
617
+ "Preventing execution of step '%s'.",
618
+ self.name,
619
+ )
620
+ return
621
+
622
+ return run_as_single_step_pipeline(self, *args, **kwargs)
605
623
 
606
624
  (
607
625
  input_artifacts,
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: zenml-nightly
3
- Version: 0.66.0.dev20240913
3
+ Version: 0.66.0.dev20240915
4
4
  Summary: ZenML: Write production-ready ML code.
5
5
  Home-page: https://zenml.io
6
6
  License: Apache-2.0
@@ -6,7 +6,7 @@ RELEASE_NOTES.md,sha256=GjuHlUd-5AcepXC8ETMkIRMxKVQtBekiuY_CW0oP4XQ,365106
6
6
  ROADMAP.md,sha256=hiLSmr16BH8Dfx7SaQM4JcXCGCVl6mFZPFAwJeDTrJU,407
7
7
  SECURITY.md,sha256=9DepA8y03yvCZLHEfcXLTDH4lUyKHquAdukBsccNN7c,682
8
8
  zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
9
- zenml/VERSION,sha256=ROKzFghy4O4DnUbwD7E-q2OY0RGQuY1sP2kCiGOOyrI,19
9
+ zenml/VERSION,sha256=gsTn_K4OdAV7Ke2H-FNrghdK3jeoLda2OXd1SytKyUM,19
10
10
  zenml/__init__.py,sha256=X_JnjAYTBG2ZYgcM2_FDPssygdj7z7VTAmU9HEi17p0,2306
11
11
  zenml/actions/__init__.py,sha256=mrt6wPo73iKRxK754_NqsGyJ3buW7RnVeIGXr1xEw8Y,681
12
12
  zenml/actions/base_action.py,sha256=UcaHev6BTuLDwuswnyaPjdA8AgUqB5xPZ-lRtuvf2FU,25553
@@ -731,7 +731,7 @@ zenml/step_operators/base_step_operator.py,sha256=ZRnY6lcEHY8xZskrKKdPhgKg2BlEoh
731
731
  zenml/step_operators/step_operator_entrypoint_configuration.py,sha256=vG0Z9D2C72PYOZ_P3eXYTmm2TvL-dS_kHhSMq-coAKE,3690
732
732
  zenml/steps/__init__.py,sha256=IvxepI3tYp6LiVHFJnet9xZ7yFlH6NHCAI69SvQsfaI,1682
733
733
  zenml/steps/base_parameters.py,sha256=rKnALd0lMv0jOX8E9TwHJkeP3H-ijEFXY0T5dMCu9EQ,756
734
- zenml/steps/base_step.py,sha256=RsSIETBgBYk3phwTzN407iJy-Lp8Mtdzni8zNP4g7Sg,51852
734
+ zenml/steps/base_step.py,sha256=sQbGnPt_1-upcYAGPqqKW0xrULNZZc9NR_7fu2y5qQk,52434
735
735
  zenml/steps/entrypoint_function_utils.py,sha256=25LxlZ8E-3YoWEdAUFK4UZ04hbp4GNMqpKwD2O-Rp_0,12484
736
736
  zenml/steps/external_artifact.py,sha256=63ngwlSJvR8GoM93ttz1Mp6BuAg7xH-QZAsBvW3M80Q,1109
737
737
  zenml/steps/step_decorator.py,sha256=_rV5MGMziqNiOoxupZUy055o-a0b8Y4RtLuB6nLyYyg,8188
@@ -1399,8 +1399,8 @@ zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=Bq1djrUP9saoD7vECjS7
1399
1399
  zenml/zen_stores/sql_zen_store.py,sha256=4z458K5ItHkAO0bEUAl6rWJDHKh6GS-sRrt2RPridnE,392435
1400
1400
  zenml/zen_stores/template_utils.py,sha256=UB_CBoMIfBequa8g3H7aTI6V_ke1SJr4IW1qwFCL1jc,8932
1401
1401
  zenml/zen_stores/zen_store_interface.py,sha256=kzR_i8vHjULld3MquSaMorcab8lJk1e9RZquw1VXjHY,93510
1402
- zenml_nightly-0.66.0.dev20240913.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
1403
- zenml_nightly-0.66.0.dev20240913.dist-info/METADATA,sha256=OInj2XGlAGtuckrQScaRo6jDy68hTD-2ZV4kGNVUBhw,20961
1404
- zenml_nightly-0.66.0.dev20240913.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
1405
- zenml_nightly-0.66.0.dev20240913.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
1406
- zenml_nightly-0.66.0.dev20240913.dist-info/RECORD,,
1402
+ zenml_nightly-0.66.0.dev20240915.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
1403
+ zenml_nightly-0.66.0.dev20240915.dist-info/METADATA,sha256=wincMYueB6wMgWBU82zaHDKreO1w4RGGenV8jsClA84,20961
1404
+ zenml_nightly-0.66.0.dev20240915.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
1405
+ zenml_nightly-0.66.0.dev20240915.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
1406
+ zenml_nightly-0.66.0.dev20240915.dist-info/RECORD,,