zenml-nightly 0.58.2.dev20240612__tar.gz → 0.58.2.dev20240615__tar.gz

{zenml_nightly-0.58.2.dev20240612 → zenml_nightly-0.58.2.dev20240615}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: zenml-nightly
-Version: 0.58.2.dev20240612
+Version: 0.58.2.dev20240615
 Summary: ZenML: Write production-ready ML code.
 Home-page: https://zenml.io
 License: Apache-2.0
@@ -212,6 +212,14 @@ Description-Content-Type: text/markdown
 
 ---
 
+## ⭐️ Show Your Support
+
+If you find ZenML helpful or interesting, please consider giving us a star on GitHub. Your support helps promote the project and lets others know that it's worth checking out. 
+
+Thank you for your support! 🌟
+
+[![Star this project](https://img.shields.io/github/stars/zenml-io/zenml?style=social)](https://github.com/zenml-io/zenml/stargazers)
+
 ## 🤸 Quickstart
 [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/zenml-io/zenml/blob/main/examples/quickstart/quickstart.ipynb)
 
@@ -343,10 +351,11 @@ enable collaborative features as the central MLOps interface for teams.
 
 Currently, there are two main ways to deploy ZenML:
 
-- **ZenML Cloud**: With [ZenML Cloud](cloud.zenml.io/?utm_source=readme&utm_medium=referral_link&utm_campaign=cloud_promotion&utm_content=signup_link),
-you can make use of a control plane to create ZenML servers, also known as tenants.
-These tenants are managed and maintained by ZenML’s dedicated team, alleviating
-the burden of server management from your end.
+- **ZenML Pro**: [ZenML Pro](cloud.zenml.io/?utm_source=readme&utm_medium=referral_link&utm_campaign=cloud_promotion&utm_content=signup_link),
+offers a SaaS version which comes with a control plane to create and manage multiple ZenML servers.
+These servers are managed and maintained by ZenML’s dedicated team, alleviating
+the burden of server management from your end. They also come with additional features like RBAC, a Model Control Plane
+and [much more](https://zenml-io.gitbook.io/zenml-documentation/getting-started/zenml-pro.
 - **Self-hosted deployment**: Alternatively, you have the flexibility to [deploy
 ZenML on your own self-hosted environment](https://docs.zenml.io/getting-started/deploying-zenml#deploying-a-zenml-server).
 This can be achieved through various methods, including using our CLI, Docker,
@@ -396,6 +405,14 @@ Or, if you
 prefer, [open an issue](https://github.com/zenml-io/zenml/issues/new/choose) on
 our GitHub repo.
 
+## ⭐️ Show Your Support
+
+If you find ZenML helpful or interesting, please consider giving us a star on GitHub. Your support helps promote the project and lets others know that it's worth checking out. 
+
+Thank you for your support! 🌟
+
+[![Star this project](https://img.shields.io/github/stars/zenml-io/zenml?style=social)](https://github.com/zenml-io/zenml/stargazers)
+
 ## 📜 License
 
 ZenML is distributed under the terms of the Apache License Version 2.0.
@@ -417,7 +434,7 @@ the Apache License Version 2.0.
     ·
     <a href="https://github.com/zenml-io/zenml/issues">Report Bug</a>
     ·
-    <a href="https://zenml.io/cloud">Sign up for Cloud</a>
+    <a href="https://zenml.io/cloud">Sign up for ZenML Pro</a>
     ·
     <a href="https://www.zenml.io/blog">Read Blog</a>
     ·

{zenml_nightly-0.58.2.dev20240612 → zenml_nightly-0.58.2.dev20240615}/README.md

@@ -75,6 +75,14 @@
 
 ---
 
+## ⭐️ Show Your Support
+
+If you find ZenML helpful or interesting, please consider giving us a star on GitHub. Your support helps promote the project and lets others know that it's worth checking out. 
+
+Thank you for your support! 🌟
+
+[![Star this project](https://img.shields.io/github/stars/zenml-io/zenml?style=social)](https://github.com/zenml-io/zenml/stargazers)
+
 ## 🤸 Quickstart
 [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/zenml-io/zenml/blob/main/examples/quickstart/quickstart.ipynb)
 
@@ -206,10 +214,11 @@ enable collaborative features as the central MLOps interface for teams.
 
 Currently, there are two main ways to deploy ZenML:
 
-- **ZenML Cloud**: With [ZenML Cloud](cloud.zenml.io/?utm_source=readme&utm_medium=referral_link&utm_campaign=cloud_promotion&utm_content=signup_link),
-you can make use of a control plane to create ZenML servers, also known as tenants.
-These tenants are managed and maintained by ZenML’s dedicated team, alleviating
-the burden of server management from your end.
+- **ZenML Pro**: [ZenML Pro](cloud.zenml.io/?utm_source=readme&utm_medium=referral_link&utm_campaign=cloud_promotion&utm_content=signup_link),
+offers a SaaS version which comes with a control plane to create and manage multiple ZenML servers.
+These servers are managed and maintained by ZenML’s dedicated team, alleviating
+the burden of server management from your end. They also come with additional features like RBAC, a Model Control Plane
+and [much more](https://zenml-io.gitbook.io/zenml-documentation/getting-started/zenml-pro.
 - **Self-hosted deployment**: Alternatively, you have the flexibility to [deploy
 ZenML on your own self-hosted environment](https://docs.zenml.io/getting-started/deploying-zenml#deploying-a-zenml-server).
 This can be achieved through various methods, including using our CLI, Docker,
@@ -259,6 +268,14 @@ Or, if you
 prefer, [open an issue](https://github.com/zenml-io/zenml/issues/new/choose) on
 our GitHub repo.
 
+## ⭐️ Show Your Support
+
+If you find ZenML helpful or interesting, please consider giving us a star on GitHub. Your support helps promote the project and lets others know that it's worth checking out. 
+
+Thank you for your support! 🌟
+
+[![Star this project](https://img.shields.io/github/stars/zenml-io/zenml?style=social)](https://github.com/zenml-io/zenml/stargazers)
+
 ## 📜 License
 
 ZenML is distributed under the terms of the Apache License Version 2.0.
@@ -280,7 +297,7 @@ the Apache License Version 2.0.
     ·
     <a href="https://github.com/zenml-io/zenml/issues">Report Bug</a>
     ·
-    <a href="https://zenml.io/cloud">Sign up for Cloud</a>
+    <a href="https://zenml.io/cloud">Sign up for ZenML Pro</a>
     ·
     <a href="https://www.zenml.io/blog">Read Blog</a>
     ·

{zenml_nightly-0.58.2.dev20240612 → zenml_nightly-0.58.2.dev20240615}/RELEASE_NOTES.md

@@ -233,9 +233,9 @@ zenml up --legacy
 
 Note that you can’t use both the new and old dashboard at the same time.
 
-If you’re self-hosting ZenML instead of using ZenML Cloud, you can specify which dashboard you want to use by setting the `ZENML_SERVER_USE_LEGACY_DASHBOARD` environment variable pre-deployment. Specifying a boolean value for this variable will determine which dashboard gets served for your deployment. (There’s no dynamic switching between dashboards allowed, so if you wish to change which dashboard is used for a deployed server, you’ll need to redeploy the server after updating the environment variable.)
+If you’re self-hosting ZenML instead of using ZenML Pro, you can specify which dashboard you want to use by setting the `ZENML_SERVER_USE_LEGACY_DASHBOARD` environment variable pre-deployment. Specifying a boolean value for this variable will determine which dashboard gets served for your deployment. (There’s no dynamic switching between dashboards allowed, so if you wish to change which dashboard is used for a deployed server, you’ll need to redeploy the server after updating the environment variable.)
 
-If you’re using [ZenML Cloud](https://cloud.zenml.io/), your experience won’t change with this release and your use of the dashboard remains the same.
+If you’re using [SaaS ZenML Pro](https://cloud.zenml.io/), your experience won’t change with this release and your use of the dashboard remains the same.
 
 ## What's Changed
 * Add Comet to Experiment Trackers in TOC by @strickvl in https://github.com/zenml-io/zenml/pull/2637

{zenml_nightly-0.58.2.dev20240612 → zenml_nightly-0.58.2.dev20240615}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zenml-nightly"
-version = "0.58.2.dev20240612"
+version = "0.58.2.dev20240615"
 packages = [{ include = "zenml", from = "src" }]
 description = "ZenML: Write production-ready ML code."
 authors = ["ZenML GmbH <info@zenml.io>"]

zenml_nightly-0.58.2.dev20240615/src/zenml/VERSION

@@ -0,0 +1 @@
+0.58.2.dev20240615

{zenml_nightly-0.58.2.dev20240612 → zenml_nightly-0.58.2.dev20240615}/src/zenml/cli/base.py

@@ -87,7 +87,7 @@ ZENML_PROJECT_TEMPLATES = dict(
     ),
     nlp=ZenMLProjectTemplateLocation(
         github_url="zenml-io/template-nlp",
-        github_tag="2024.06.06",  # Make sure it is aligned with .github/workflows/update-templates-to-examples.yml
+        github_tag="2024.06.14",  # Make sure it is aligned with .github/workflows/update-templates-to-examples.yml
     ),
     llm_finetuning=ZenMLProjectTemplateLocation(
         github_url="zenml-io/template-llm-finetuning",

{zenml_nightly-0.58.2.dev20240612 → zenml_nightly-0.58.2.dev20240615}/src/zenml/new/pipelines/run_utils.py

@@ -255,6 +255,6 @@ def _validate_new_version_requests(
         )
     if not is_cloud_model:
         logger.info(
-            "Models can be viewed in the dashboard using ZenML Cloud. Sign up "
+            "Models can be viewed in the dashboard using ZenML Pro. Sign up "
             "for a free trial at https://www.zenml.io/cloud/"
         )

{zenml_nightly-0.58.2.dev20240612 → zenml_nightly-0.58.2.dev20240615}/src/zenml/stack/flavor.py

@@ -26,6 +26,7 @@ from zenml.models import (
 from zenml.models.v2.core.flavor import InternalFlavorRequest
 from zenml.stack.stack_component import StackComponent, StackComponentConfig
 from zenml.utils import source_utils
+from zenml.utils.package_utils import is_latest_zenml_version
 
 
 class Flavor:
@@ -187,16 +188,12 @@ class Flavor:
         )
         return model
 
-    def generate_default_docs_url(self, component_name: str = "") -> str:
+    def generate_default_docs_url(self) -> str:
         """Generate the doc urls for all inbuilt and integration flavors.
 
         Note that this method is not going to be useful for custom flavors,
         which do not have any docs in the main zenml docs.
 
-        Args:
-            component_name: The name of the component for docs generation. Used
-                for legacy documentation before ZenML v0.34.0.
-
         Returns:
             The complete url to the zenml documentation
         """
@@ -204,11 +201,18 @@ class Flavor:
 
         component_type = self.type.plural.replace("_", "-")
         name = self.name.replace("_", "-")
-        docs_component_name = component_name or name
-        base = f"https://docs.zenml.io/v/{__version__}"
-        return (
-            f"{base}/stack-components/{component_type}/{docs_component_name}"
-        )
+
+        try:
+            is_latest = is_latest_zenml_version()
+        except RuntimeError:
+            # We assume in error cases that we are on the latest version
+            is_latest = True
+
+        if is_latest:
+            base = "https://docs.zenml.io"
+        else:
+            base = f"https://zenml-io.gitbook.io/zenml-legacy-documentation/v/{__version__}"
+        return f"{base}/stack-components/{component_type}/{name}"
 
     def generate_default_sdk_docs_url(self) -> str:
         """Generate SDK docs url for a flavor.

{zenml_nightly-0.58.2.dev20240612 → zenml_nightly-0.58.2.dev20240615}/src/zenml/utils/cloud_utils.py

@@ -11,7 +11,7 @@
 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 #  or implied. See the License for the specific language governing
 #  permissions and limitations under the License.
-"""Utilities for ZenML Cloud."""
+"""Utilities for ZenML Pro."""
 
 from zenml.logger import get_logger
 from zenml.models.v2.core.model_version import ModelVersionResponse
@@ -21,13 +21,13 @@ logger = get_logger(__name__)
 
 
 def is_cloud_model_version(model_version: ModelVersionResponse) -> bool:
-    """Check if a model version is from a ZenML Cloud server.
+    """Check if a model version is from a ZenML Pro server.
 
     Args:
         model_version: The model version to check.
 
     Returns:
-        True if the model version is from a ZenML Cloud server, else False.
+        True if the model version is from a ZenML Pro server, else False.
     """
     model_version_url = get_model_version_url(model_version.id)
     if model_version_url:

{zenml_nightly-0.58.2.dev20240612 → zenml_nightly-0.58.2.dev20240615}/src/zenml/utils/dashboard_utils.py

@@ -32,13 +32,13 @@ logger = get_logger(__name__)
 
 
 def is_cloud_server(server_info: ServerModel) -> bool:
-    """Checks whether the server info refers to a ZenML Cloud server.
+    """Checks whether the server info refers to a ZenML Pro server.
 
     Args:
         server_info: Info of the server.
 
     Returns:
-        Whether the server info refers to a ZenML Cloud server.
+        Whether the server info refers to a ZenML Pro server.
     """
     return (
         "organization_id" in server_info.metadata

zenml_nightly-0.58.2.dev20240615/src/zenml/utils/package_utils.py

@@ -0,0 +1,50 @@
+#  Copyright (c) ZenML GmbH 2024. All Rights Reserved.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at:
+#
+#       https://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
+#  or implied. See the License for the specific language governing
+#  permissions and limitations under the License.
+"""Utility functions for the package."""
+
+import requests
+from packaging import version
+
+
+def is_latest_zenml_version() -> bool:
+    """Checks if the currently running ZenML package is on the latest version.
+
+    Returns:
+        True in case the current running zenml code is the latest available version on PYPI, otherwise False.
+
+    Raises:
+        RuntimeError: In case something goe wrong
+    """
+    from zenml import __version__
+
+    # Get the current version of the package
+    current_local_version = __version__
+
+    # Get the latest version from PyPI
+    try:
+        response = requests.get("https://pypi.org/pypi/zenml/json", timeout=60)
+        response.raise_for_status()
+        latest_published_version = response.json()["info"]["version"]
+    except Exception as e:
+        raise RuntimeError(
+            f"Failed to fetch the latest version from PyPI: {e}"
+        )
+
+    # Compare versions
+    if version.parse(latest_published_version) > version.parse(
+        current_local_version
+    ):
+        return False
+    else:
+        return True

{zenml_nightly-0.58.2.dev20240612 → zenml_nightly-0.58.2.dev20240615}/src/zenml/zen_server/cloud_utils.py

@@ -14,7 +14,7 @@ ZENML_CLOUD_RBAC_ENV_PREFIX = "ZENML_CLOUD_"
 
 
 class ZenMLCloudConfiguration(BaseModel):
-    """ZenML Cloud RBAC configuration."""
+    """ZenML Pro RBAC configuration."""
 
     api_url: str
 
@@ -141,7 +141,7 @@ class ZenMLCloudSession:
             response.raise_for_status()
         except requests.HTTPError as e:
             raise RuntimeError(
-                f"Failed while trying to contact the central zenml cloud "
+                f"Failed while trying to contact the central zenml pro "
                 f"service: {e}"
             )
 
@@ -149,7 +149,7 @@ class ZenMLCloudSession:
 
     @property
     def session(self) -> requests.Session:
-        """Authenticate to the ZenML Cloud API.
+        """Authenticate to the ZenML Pro Management Plane.
 
         Returns:
             A requests session with the authentication token.

{zenml_nightly-0.58.2.dev20240612 → zenml_nightly-0.58.2.dev20240615}/src/zenml/zen_server/feature_gate/zenml_cloud_feature_gate.py

@@ -11,7 +11,7 @@
 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 #  or implied. See the License for the specific language governing
 #  permissions and limitations under the License.
-"""ZenML Cloud implementation of the feature gate."""
+"""ZenML Pro implementation of the feature gate."""
 
 from typing import Any, Dict
 from uuid import UUID

{zenml_nightly-0.58.2.dev20240612 → zenml_nightly-0.58.2.dev20240615}/src/zenml/zen_server/pipeline_deployment/utils.py

@@ -7,6 +7,7 @@ from typing import List, Optional, Set, Tuple
 from uuid import UUID
 
 from fastapi import BackgroundTasks
+from packaging import version
 
 from zenml.config.pipeline_configurations import PipelineConfiguration
 from zenml.config.pipeline_run_configuration import PipelineRunConfiguration
@@ -116,7 +117,14 @@ def run_pipeline(
             stack=stack
         )
 
-        python_version = f"{sys.version_info.major}.{sys.version_info.minor}"
+        if build.python_version:
+            version_info = version.parse(build.python_version)
+            python_version = f"{version_info.major}.{version_info.minor}"
+        else:
+            python_version = (
+                f"{sys.version_info.major}.{sys.version_info.minor}"
+            )
+
         dockerfile = generate_dockerfile(
             pypi_requirements=pypi_requirements,
             apt_packages=apt_packages,

{zenml_nightly-0.58.2.dev20240612 → zenml_nightly-0.58.2.dev20240615}/src/zenml/zen_server/rbac/zenml_cloud_rbac.py

@@ -34,7 +34,7 @@ SERVER_ID = server_config().external_server_id
 
 
 def _convert_to_cloud_resource(resource: Resource) -> str:
-    """Convert a resource to a ZenML Cloud API resource.
+    """Convert a resource to a ZenML Pro Management Plane resource.
 
     Args:
         resource: The resource to convert.
@@ -75,7 +75,7 @@ def _convert_from_cloud_resource(cloud_resource: str) -> Resource:
 
 
 class ZenMLCloudRBAC(RBACInterface, ZenMLCloudSession):
-    """RBAC implementation that uses the ZenML Cloud API as a backend."""
+    """RBAC implementation that uses the ZenML Pro Management Plane as a backend."""
 
     def check_permissions(
         self, user: "UserResponse", resources: Set[Resource], action: Action
@@ -99,7 +99,7 @@ class ZenMLCloudRBAC(RBACInterface, ZenMLCloudSession):
             # Service accounts have full permissions for now
             return {resource: True for resource in resources}
 
-        # At this point it's a regular user, which in the ZenML cloud with RBAC
+        # At this point it's a regular user, which in a ZenML Pro with RBAC
         # enabled is always authenticated using external authentication
         assert user.external_user_id
 
@@ -139,7 +139,7 @@ class ZenMLCloudRBAC(RBACInterface, ZenMLCloudSession):
             # Service accounts have full permissions for now
             return True, []
 
-        # At this point it's a regular user, which in the ZenML cloud with RBAC
+        # At this point it's a regular user, which in the ZenML Pro with RBAC
         # enabled is always authenticated using external authentication
         assert user.external_user_id
         params = {

zenml_nightly-0.58.2.dev20240612/src/zenml/VERSION

@@ -1 +0,0 @@
-0.58.2.dev20240612