zenml-nightly 0.82.0.dev20250511__py3-none-any.whl → 0.82.0.dev20250513__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.82.0.dev20250511
+0.82.0.dev20250513

zenml/cli/login.py

@@ -1061,7 +1061,7 @@ def logout(
             if clear:
                 credentials_store.clear_credentials(server_url=server)
             cli_utils.declare(
-                f"Logged out from {server}."
+                f"Logged out from {server}.\n"
                 f"Hint: You can run `zenml login '{server}'` to log in again "
                 "to the same server or 'zenml server list' to view other available "
                 "servers that you can connect to with 'zenml login <server-url>'."

zenml/zen_stores/migrations/alembic.py

@@ -39,7 +39,7 @@ exclude_tables = ["sqlite_sequence"]
 
 
 def include_object(
-    object: Any, name: str, type_: str, *args: Any, **kwargs: Any
+    object: Any, name: Optional[str], type_: str, *args: Any, **kwargs: Any
 ) -> bool:
     """Function used to exclude tables from the migration scripts.
 
@@ -135,6 +135,7 @@ class Alembic:
             fn_context_args["fn"] = fn
 
         with self.engine.connect() as connection:
+            # Configure the context with our metadata
             self.environment_context.configure(
                 connection=connection,
                 target_metadata=self.metadata,
@@ -180,9 +181,15 @@ class Alembic:
         def do_get_current_rev(rev: _RevIdType, context: Any) -> List[Any]:
             nonlocal current_revisions
 
-            for r in self.script_directory.get_all_current(
-                rev  # type:ignore [arg-type]
-            ):
+            # Handle rev parameter in a way that's compatible with different alembic versions
+            rev_input: Any
+            if isinstance(rev, str):
+                rev_input = rev
+            else:
+                rev_input = tuple(str(r) for r in rev)
+
+            # Get current revision(s)
+            for r in self.script_directory.get_all_current(rev_input):
                 if r is None:
                     continue
                 current_revisions.append(r.revision)
@@ -200,7 +207,13 @@ class Alembic:
         """
 
         def do_stamp(rev: _RevIdType, context: Any) -> List[Any]:
-            return self.script_directory._stamp_revs(revision, rev)
+            # Handle rev parameter in a way that's compatible with different alembic versions
+            if isinstance(rev, str):
+                return self.script_directory._stamp_revs(revision, rev)
+            else:
+                # Convert to tuple for compatibility
+                rev_tuple = tuple(str(r) for r in rev)
+                return self.script_directory._stamp_revs(revision, rev_tuple)
 
         self.run_migrations(do_stamp)
 
@@ -212,10 +225,16 @@ class Alembic:
         """
 
         def do_upgrade(rev: _RevIdType, context: Any) -> List[Any]:
-            return self.script_directory._upgrade_revs(
-                revision,
-                rev,  # type:ignore [arg-type]
-            )
+            # Handle rev parameter in a way that's compatible with different alembic versions
+            if isinstance(rev, str):
+                return self.script_directory._upgrade_revs(revision, rev)
+            else:
+                if rev:
+                    # Use first element or revs for compatibility
+                    return self.script_directory._upgrade_revs(
+                        revision, str(rev[0])
+                    )
+                return []
 
         self.run_migrations(do_upgrade)
 
@@ -227,9 +246,14 @@ class Alembic:
         """
 
         def do_downgrade(rev: _RevIdType, context: Any) -> List[Any]:
-            return self.script_directory._downgrade_revs(
-                revision,
-                rev,  # type:ignore [arg-type]
-            )
+            # Handle rev parameter in a way that's compatible with different alembic versions
+            if isinstance(rev, str):
+                return self.script_directory._downgrade_revs(revision, rev)
+            else:
+                if rev:
+                    return self.script_directory._downgrade_revs(
+                        revision, str(rev[0])
+                    )
+                return self.script_directory._downgrade_revs(revision, None)
 
         self.run_migrations(do_downgrade)

zenml/zen_stores/rest_zen_store.py

@@ -4413,11 +4413,15 @@ class RestZenStore(BaseZenStore):
                         "The current token is no longer valid, and "
                         "it is not possible to generate a new token using the "
                         "configured credentials. Please run "
-                        f"`zenml login --url {self.url}` to re-authenticate to "
-                        "the server or authenticate using an API key. See "
+                        f"`zenml login {self.url}` to "
+                        "re-authenticate to the server or authenticate using "
+                        "an API key. See "
                         "https://docs.zenml.io/how-to/project-setup-and-management/connecting-to-zenml/connect-with-a-service-account "
                         "for more information."
                     )
+                    # Clear the current token from the credentials store to
+                    # force a new authentication flow next time.
+                    get_credentials_store().clear_token(self.url)
                     raise e
                 elif not re_authenticated:
                     # The last request was authenticated with an API token

{zenml_nightly-0.82.0.dev20250511.dist-info → zenml_nightly-0.82.0.dev20250513.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zenml-nightly
-Version: 0.82.0.dev20250511
+Version: 0.82.0.dev20250513
 Summary: ZenML: Write production-ready ML code.
 License: Apache-2.0
 Keywords: machine learning,production,pipeline,mlops,devops
@@ -41,7 +41,7 @@ Provides-Extra: terraform
 Provides-Extra: vertex
 Requires-Dist: Jinja2 ; extra == "server"
 Requires-Dist: adlfs (>=2021.10.0) ; extra == "adlfs"
-Requires-Dist: alembic (>=1.8.1,<1.9.0)
+Requires-Dist: alembic (>=1.8.1,<=1.15.2)
 Requires-Dist: aws-profile-manager (>=0.5.0) ; extra == "connectors-aws"
 Requires-Dist: azure-ai-ml (==1.23.1) ; extra == "azureml"
 Requires-Dist: azure-identity (>=1.4.0) ; extra == "secrets-azure" or extra == "connectors-azure"

{zenml_nightly-0.82.0.dev20250511.dist-info → zenml_nightly-0.82.0.dev20250513.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=sNYKOurs-d4b5GAaVuicbVQeHyNYJIL-DKO5dTE8P_E,19
+zenml/VERSION,sha256=GgvcZ4qW8lpLu66ULVW2IWU5xLriT6P_N-LWtNfqCL8,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
@@ -38,7 +38,7 @@ zenml/cli/downgrade.py,sha256=eTpXts8y4s3wEUwOlvZGWsTngoMV8Stuzj0K-SAQUGU,1887
 zenml/cli/feature.py,sha256=Q8tNvWBlRze3FUyn0_VpOdl316ZW87476j7ezJb16GA,4387
 zenml/cli/formatter.py,sha256=OoAzgxpOEXrzT9FSFG0Bs5VRtVFEYck0gDYo02GCjzI,6440
 zenml/cli/integration.py,sha256=Rq37R-QiSvaFOMmlA0YJTwDti4slLp-hEf05vbWxkFk,17346
-zenml/cli/login.py,sha256=-tPuTaWctRDkWZOcY4Af2uens6_oEWFkehl8ITeKFHk,38836
+zenml/cli/login.py,sha256=7PHA7YEHF3NuBeZLslWivjHbIRtS2POa_V_Gz6Kv0fc,38838
 zenml/cli/model.py,sha256=HvzyRqcbpcgAPMcFAh4UmHZWbCJq0RMQzYVOUGNUXfM,22638
 zenml/cli/model_registry.py,sha256=zzxWXXFhKu2B1Wp0u7prKVnN1ftM-JdGdQwlD-5G-QM,20786
 zenml/cli/pipeline.py,sha256=Vlz1OgGb1Ep-4Ekgd-Wz5SmieWigfx56i8wA5BGl228,19222
@@ -1072,7 +1072,7 @@ zenml/zen_stores/__init__.py,sha256=6LTgH6XwDeDxKqVJ1JTfGhmS8II1NLopPloINGmdyI0,
 zenml/zen_stores/base_zen_store.py,sha256=AplsW2NR-G9_CU54XvNTQJo4W0KJ5TJV22cjKW4n2BY,16124
 zenml/zen_stores/migrations/README.md,sha256=x04jsb6EOP6PBEGMQlDELiqKEham2O-iztAs9AylMFc,4898
 zenml/zen_stores/migrations/__init__.py,sha256=N9CHfdz0AZ6KniQ450VCIV3H0CuWtx83AloYy82woho,657
-zenml/zen_stores/migrations/alembic.py,sha256=JDqx7Md6DxnHtP3xrZG1I0cNv6NyTR0oO3tPRUPaS2I,7455
+zenml/zen_stores/migrations/alembic.py,sha256=wGiWy9R2MNAos4zj5lYQOJ9ZSfMTmUFu3SGhp1AoMd8,8753
 zenml/zen_stores/migrations/env.py,sha256=hN6GqD2toKL-r9y0FFAf2seJfr79Mzaeqslh5kObcos,1730
 zenml/zen_stores/migrations/script.py.mako,sha256=wTJhgE4DA8I2iVA29sfx74WLfbi3GBnXEwGnH5nNj4s,695
 zenml/zen_stores/migrations/utils.py,sha256=_o6qr0_fKrcYYtrVK3kT0ZrjpyqFiMzRGsSmsqRPaZ0,29527
@@ -1272,7 +1272,7 @@ zenml/zen_stores/migrations/versions/f49904a80aa7_increase_length_of_artifact_ta
 zenml/zen_stores/migrations/versions/f76a368a25a5_add_stack_description.py,sha256=u8fRomaasFeGhxvM2zU-Ab-AEpVsWm5zRcixxKFXdRw,904
 zenml/zen_stores/migrations/versions/fbd7f18ced1e_increase_step_run_field_lengths.py,sha256=kn-ng5EHe_mmLfffIFbz7T59z-to3oMx8III_4wOsz4,1956
 zenml/zen_stores/migrations/versions/ff538a321a92_migrate_onboarding_state.py,sha256=gsUFLJQ32_o9U35JCVqkqJVVk-zfq3yel25hXhzVFm4,3829
-zenml/zen_stores/rest_zen_store.py,sha256=-Xjw9QTY6lvgp5mDE4bBIhU_ANADS83Si88xMcLTFIE,158371
+zenml/zen_stores/rest_zen_store.py,sha256=d5JHoGeO_FSg55MmVNIFYmwN1JdAnJ2QbE6R590fKZU,158599
 zenml/zen_stores/schemas/__init__.py,sha256=4EXqExiVyxdnGxhQ_Hz79mOdRuMD0LsGlw0PaP2Ef6o,4333
 zenml/zen_stores/schemas/action_schemas.py,sha256=sv2J2TP12MeyGPQR2JsOPIivbPQ5OImg64exYS7CZBM,6496
 zenml/zen_stores/schemas/api_key_schemas.py,sha256=0pK7b9HlJuQL3DuKT4eGjFb87tyd4x-E2VyxJLpRv3o,7459
@@ -1318,8 +1318,8 @@ zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=LPFW757WCJLP1S8vrvjs
 zenml/zen_stores/sql_zen_store.py,sha256=biOoDb2_zYmpsN-J-FSlKICYdwM9KDIe-_KN_yDf_mA,441414
 zenml/zen_stores/template_utils.py,sha256=GbJ7LgGVYHSCKPEA8RNTxPoVTWqpC77F_lGzjJ4O1Fw,9220
 zenml/zen_stores/zen_store_interface.py,sha256=fF_uL_FplnvGvM5o3jOQ8i1zHXhuhKLL2n4nvIKSR7E,92090
-zenml_nightly-0.82.0.dev20250511.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.82.0.dev20250511.dist-info/METADATA,sha256=d-G4MWVNQJVUJbr34gl5sEHO-l5rUjoR09KrF0PaRHU,24315
-zenml_nightly-0.82.0.dev20250511.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-zenml_nightly-0.82.0.dev20250511.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.82.0.dev20250511.dist-info/RECORD,,
+zenml_nightly-0.82.0.dev20250513.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.82.0.dev20250513.dist-info/METADATA,sha256=hMwBzq4y98Z1jeZUvJMdUimsg1jZUzvmzqDMXpMrSIs,24317
+zenml_nightly-0.82.0.dev20250513.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+zenml_nightly-0.82.0.dev20250513.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.82.0.dev20250513.dist-info/RECORD,,