zenml-nightly 0.83.0.dev20250602__py3-none-any.whl → 0.83.0.dev20250604__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.83.0.dev20250602
+0.83.0.dev20250604

zenml/config/global_config.py

@@ -751,13 +751,15 @@ class GlobalConfiguration(BaseModel, metaclass=GlobalConfigMetaClass):
         Returns:
             The model of the active project.
         """
-        project_id = self.get_active_project_id()
+        # Initialize the store first. This might create a local database if it
+        # doesn't exist yet, which will then refresh our active project ID.
+        _ = self.zen_store
 
         if self._active_project is not None:
             return self._active_project
 
         project = self.zen_store.get_project(
-            project_name_or_id=project_id,
+            project_name_or_id=self.get_active_project_id(),
         )
         return self.set_active_project(project)
 

zenml/logging/step_logging.py

@@ -549,8 +549,14 @@ class PipelineLogsStorageContext:
                 except Exception:
                     pass
 
-                if step_context and args[0] != "\n":
-                    message = f"[{step_context.step_name}] " + message
+                if step_context and args[0] not in ["\n", ""]:
+                    # For progress bar updates (with \r), inject the step name after the \r
+                    if "\r" in message:
+                        message = message.replace(
+                            "\r", f"\r[{step_context.step_name}] "
+                        )
+                    else:
+                        message = f"[{step_context.step_name}] {message}"
 
                 output = method(message, *args[1:], **kwargs)
 

zenml/zen_stores/sql_zen_store.py

@@ -3001,6 +3001,10 @@ class SqlZenStore(BaseZenStore):
                     )
                     session.add(vis_schema)
 
+            # Commit the visualizations so potential future rollbacks don't
+            # remove them
+            session.commit()
+
             # Save tags of the artifact
             self._attach_tags_to_resources(
                 tags=artifact_version.tags,
@@ -11994,16 +11998,17 @@ class SqlZenStore(BaseZenStore):
         validate_name(tag)
         self._set_request_user_id(request_model=tag, session=session)
 
-        tag_schema = TagSchema.from_request(tag)
-        session.add(tag_schema)
+        with session.begin_nested() as nested_session:
+            tag_schema = TagSchema.from_request(tag)
+            session.add(tag_schema)
 
-        try:
-            session.commit()
-        except IntegrityError:
-            session.rollback()
-            raise EntityExistsError(
-                f"Tag with name `{tag.name}` already exists."
-            )
+            try:
+                session.commit()
+            except IntegrityError:
+                nested_session.rollback()
+                raise EntityExistsError(
+                    f"Tag with name `{tag.name}` already exists."
+                )
         return tag_schema
 
     @track_decorator(AnalyticsEvent.CREATED_TAG)

{zenml_nightly-0.83.0.dev20250602.dist-info → zenml_nightly-0.83.0.dev20250604.dist-info}/METADATA

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

{zenml_nightly-0.83.0.dev20250602.dist-info → zenml_nightly-0.83.0.dev20250604.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=IgUqPOSkmIyOMqpOMjsvV2TBC_cDQZxshiEmNBMHAvI,19
+zenml/VERSION,sha256=vOk95Zv12MpjL0GyiQF6b_h7g_FKv7O4bvZHJ5fJgHA,19
 zenml/__init__.py,sha256=CKEyepFK-7akXYiMrNVh92Nb01Cjs23w4_YyI6sgdc8,2242
 zenml/actions/__init__.py,sha256=mrt6wPo73iKRxK754_NqsGyJ3buW7RnVeIGXr1xEw8Y,681
 zenml/actions/base_action.py,sha256=UcaHev6BTuLDwuswnyaPjdA8AgUqB5xPZ-lRtuvf2FU,25553
@@ -68,7 +68,7 @@ zenml/config/build_configuration.py,sha256=jGGNwP0Cb7a80JXArNxDgxzxl9ytSZRtv-WW7
 zenml/config/compiler.py,sha256=bK3LCDkrFc9SapJYH-vuQZ_o8scHNs-FdC5DblIUU4U,23024
 zenml/config/constants.py,sha256=QvSgMwXWxtspcJ45CrFDP1ZY3w6gS3bIhXLOtIDAbZA,713
 zenml/config/docker_settings.py,sha256=xZvoeC6v6RG_5MFK_RgdTgrkOLu-QT6frY4SGQNZUTo,18460
-zenml/config/global_config.py,sha256=ZD3WodfcWMBJZOl1FNn3ztzwilGDDv9EKdHClLqSO8s,29562
+zenml/config/global_config.py,sha256=59orl6xp4moxSo2Q8tXtDfpIBX4AxJ9FHcm8s2VHyWw,29712
 zenml/config/pipeline_configurations.py,sha256=7trCbElpqGGgawii2FrdLW8fKaAWCR8jACkNqdG_vcQ,3983
 zenml/config/pipeline_run_configuration.py,sha256=Y9C5mVUH-w4gc1w1PCQFpjEmfBBpSMvb8riA_sL78hY,2311
 zenml/config/pipeline_spec.py,sha256=uWpiIfznJvXAiKs1yMIUDT1h1tYEFNO-RDVTYcIv9CE,2821
@@ -578,7 +578,7 @@ zenml/io/filesystem_registry.py,sha256=stujDg4a5k983WMwp3rj4Z4puiUco4REyVoIoMIpI
 zenml/io/local_filesystem.py,sha256=xQTZPT5cpooptUV8KiifxZojS6pWCv1-6UUxitUYb_E,7386
 zenml/logger.py,sha256=LMV2sMFQ-6JK9xEn6kEt1C9vAoQ0lwuHVM5XHIeKubE,6966
 zenml/logging/__init__.py,sha256=lnqbOa31wAHwPP5f8vZazOrUwnP2QviLiIVwxoAefD8,975
-zenml/logging/step_logging.py,sha256=GApmyuFGybkj1rFbv8Ej3fNym2P4HzGKyUBKFu2YO90,20907
+zenml/logging/step_logging.py,sha256=Yz2LJkCZcjWcU0OCMtpAZwggpKTKPDr2Cgy14mzM6gk,21222
 zenml/login/__init__.py,sha256=Evi7hq8tpUn57IM3iX3hYP0r8oIeEWUhS471TAOyVGs,644
 zenml/login/credentials.py,sha256=RtLmYkFQ5trbprhsO8NCRKwBA136KzGoUWY52dZP9GA,12722
 zenml/login/credentials_store.py,sha256=k55mNSqc_RaBuvLWtg5ubQAwlSbaTxTVAqoZ1OQ2YcM,24750
@@ -1324,11 +1324,11 @@ zenml/zen_stores/secrets_stores/hashicorp_secrets_store.py,sha256=5err1a-TrV3SR5
 zenml/zen_stores/secrets_stores/secrets_store_interface.py,sha256=Q2Jbnt2Pp7NGlR-u1YBfRZV2g8su2Fd0ArBMdksAE-Q,2819
 zenml/zen_stores/secrets_stores/service_connector_secrets_store.py,sha256=S87ne23D08PAwtfRVlVnBn8R0ilTpEh6r8blauNV5WQ,6941
 zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=LPFW757WCJLP1S8vrvjsrl2Tf1yo281xUTjSBsos4qk,8788
-zenml/zen_stores/sql_zen_store.py,sha256=TxqVUsE-2NlJ1EasyMfIOtLCPMCtbd19Jgye9b4PlVE,465963
+zenml/zen_stores/sql_zen_store.py,sha256=L5EhDaO-799UXJZQChGmOos343HJx4S3L16_x_N2gqw,466193
 zenml/zen_stores/template_utils.py,sha256=GbJ7LgGVYHSCKPEA8RNTxPoVTWqpC77F_lGzjJ4O1Fw,9220
 zenml/zen_stores/zen_store_interface.py,sha256=_ap55L3_mrHgegsLkMRSmmNXVasYC53LwjcEeuS1YT4,92411
-zenml_nightly-0.83.0.dev20250602.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.83.0.dev20250602.dist-info/METADATA,sha256=l74OX17ZAaXBQ1cgj4S1ANctJdSEU9lF9jSnXChjA0M,24317
-zenml_nightly-0.83.0.dev20250602.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-zenml_nightly-0.83.0.dev20250602.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.83.0.dev20250602.dist-info/RECORD,,
+zenml_nightly-0.83.0.dev20250604.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.83.0.dev20250604.dist-info/METADATA,sha256=UC_wYNNxKBdlqijXKERAmru_bMBTX4MUnJq-YIhkUoM,24317
+zenml_nightly-0.83.0.dev20250604.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+zenml_nightly-0.83.0.dev20250604.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.83.0.dev20250604.dist-info/RECORD,,