zenml-nightly 0.72.0.dev20250121__py3-none-any.whl → 0.73.0.dev20250124__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.72.0.dev20250121
+0.73.0.dev20250124

zenml/cli/login.py

@@ -249,6 +249,7 @@ def connect_to_pro_server(
     from zenml.login.pro.tenant.models import TenantStatus
 
     pro_api_url = pro_api_url or ZENML_PRO_API_URL
+    pro_api_url = pro_api_url.rstrip("/")
 
     server_id, server_url, server_name = None, None, None
     login = False
@@ -434,6 +435,7 @@ def is_pro_server(
     from zenml.login.credentials_store import get_credentials_store
     from zenml.login.server_info import get_server_info
 
+    url = url.rstrip("/")
     # First, check the credentials store
     credentials_store = get_credentials_store()
     credentials = credentials_store.get_credentials(url)

zenml/cli/server.py

@@ -578,6 +578,7 @@ def server_list(
     from zenml.login.pro.tenant.models import TenantRead, TenantStatus
 
     pro_api_url = pro_api_url or ZENML_PRO_API_URL
+    pro_api_url = pro_api_url.rstrip("/")
 
     credentials_store = get_credentials_store()
     pro_token = credentials_store.get_pro_token(

zenml/cli/service_connectors.py

@@ -13,7 +13,7 @@
 #  permissions and limitations under the License.
 """Service connector CLI commands."""
 
-from datetime import datetime
+from datetime import datetime, timezone
 from typing import Any, Dict, List, Optional, Union, cast
 from uuid import UUID
 
@@ -291,7 +291,9 @@ def prompt_expires_at(
     while True:
         default_str = ""
         if default is not None:
-            seconds = int((default - datetime.utcnow()).total_seconds())
+            seconds = int(
+                (default - datetime.now(timezone.utc)).total_seconds()
+            )
             default_str = (
                 f" [{str(default)} i.e. in "
                 f"{seconds_to_human_readable(seconds)}]"
@@ -307,14 +309,16 @@ def prompt_expires_at(
 
         assert expires_at is not None
         assert isinstance(expires_at, datetime)
-        if expires_at < datetime.utcnow():
+        if expires_at < datetime.now(timezone.utc):
             cli_utils.warning(
                 "The expiration time must be in the future. Please enter a "
                 "later date and time."
             )
             continue
 
-        seconds = int((expires_at - datetime.utcnow()).total_seconds())
+        seconds = int(
+            (expires_at - datetime.now(timezone.utc)).total_seconds()
+        )
 
         confirm = click.confirm(
             f"Credentials will be valid until {str(expires_at)} UTC (i.e. "

zenml/cli/stack.py

@@ -17,7 +17,7 @@ import getpass
 import re
 import time
 import webbrowser
-from datetime import datetime
+from datetime import datetime, timezone
 from typing import (
     TYPE_CHECKING,
     Any,
@@ -1575,7 +1575,7 @@ def deploy(
         ):
             raise click.Abort()
 
-        date_start = datetime.utcnow()
+        date_start = datetime.now(timezone.utc)
 
         webbrowser.open(deployment_config.deployment_url)
         console.print(

zenml/config/pipeline_configurations.py

@@ -13,7 +13,7 @@
 #  permissions and limitations under the License.
 """Pipeline configuration classes."""
 
-from datetime import datetime
+from datetime import datetime, timezone
 from typing import TYPE_CHECKING, Any, Dict, List, Optional
 
 from pydantic import SerializeAsAny, field_validator
@@ -61,7 +61,7 @@ class PipelineConfigurationUpdate(StrictBaseModel):
             The full substitutions dict including date and time.
         """
         if start_time is None:
-            start_time = datetime.utcnow()
+            start_time = datetime.now(timezone.utc)
         ret = self.substitutions.copy()
         ret.setdefault("date", start_time.strftime("%Y_%m_%d"))
         ret.setdefault("time", start_time.strftime("%H_%M_%S_%f"))

zenml/config/server_config.py

@@ -432,6 +432,26 @@ class ServerConfiguration(BaseModel):
 
         return data
 
+    @field_validator("reportable_resources", mode="before")
+    @classmethod
+    def _convert_reportable_resources(cls, value: Any) -> Any:
+        """Convert reportable resources value if necessary.
+
+        This was previously set via an environment variable as a JSON-formatted
+        string. In case this still exists somewhere, this method converts the
+        JSON string to a list.
+
+        Args:
+            value: The reportable resources values.
+
+        Returns:
+            The potentially converted value.
+        """
+        if isinstance(value, str):
+            value = json.loads(value)
+
+        return value
+
     @property
     def deployment_id(self) -> UUID:
         """Get the ZenML server deployment ID.

zenml/enums.py

@@ -376,6 +376,7 @@ class MetadataResourceTypes(StrEnum):
     STEP_RUN = "step_run"
     ARTIFACT_VERSION = "artifact_version"
     MODEL_VERSION = "model_version"
+    SCHEDULE = "schedule"
 
 
 class DatabaseBackupStrategy(StrEnum):

zenml/event_hub/base_event_hub.py

@@ -14,7 +14,7 @@
 """Base class for event hub implementations."""
 
 from abc import ABC, abstractmethod
-from datetime import datetime, timedelta
+from datetime import datetime, timedelta, timezone
 from typing import TYPE_CHECKING, Any, Callable, Dict, Optional, Tuple
 
 from zenml import EventSourceResponse
@@ -134,7 +134,7 @@ class BaseEventHub(ABC):
         )
         expires: Optional[datetime] = None
         if trigger.action.auth_window:
-            expires = datetime.utcnow() + timedelta(
+            expires = datetime.now(timezone.utc) + timedelta(
                 minutes=trigger.action.auth_window
             )
         encoded_token = token.encode(expires=expires)

zenml/integrations/airflow/orchestrators/airflow_orchestrator.py

@@ -408,7 +408,8 @@ class AirflowOrchestrator(ContainerizedOrchestrator):
         if schedule:
             if schedule.cron_expression:
                 start_time = schedule.start_time or (
-                    datetime.datetime.utcnow() - datetime.timedelta(7)
+                    datetime.datetime.now(datetime.timezone.utc)
+                    - datetime.timedelta(7)
                 )
                 return {
                     "schedule": schedule.cron_expression,
@@ -428,6 +429,7 @@ class AirflowOrchestrator(ContainerizedOrchestrator):
             "schedule": "@once",
             # set a start time in the past and disable catchup so airflow
             # runs the dag immediately
-            "start_date": datetime.datetime.utcnow() - datetime.timedelta(7),
+            "start_date": datetime.datetime.now(datetime.timezone.utc)
+            - datetime.timedelta(7),
             "catchup": False,
         }

zenml/integrations/aws/__init__.py

@@ -35,12 +35,13 @@ AWS_RESOURCE_TYPE = "aws-generic"
 S3_RESOURCE_TYPE = "s3-bucket"
 AWS_IMAGE_BUILDER_FLAVOR = "aws"
 
+
 class AWSIntegration(Integration):
     """Definition of AWS integration for ZenML."""
 
     NAME = AWS
     REQUIREMENTS = [
-        "sagemaker>=2.117.0",
+        "sagemaker>=2.199.0",
         "kubernetes",
         "aws-profile-manager",
     ]

zenml/integrations/aws/flavors/sagemaker_orchestrator_flavor.py

@@ -85,6 +85,7 @@ class SagemakerOrchestratorSettings(BaseSettings):
             to the container is configured with input_data_s3_mode. Two possible
             input types:
                 - str: S3 location where training data is saved.
+                - Dict[str, str]: (ChannelName, S3Location) which represent
                 - Dict[str, str]: (ChannelName, S3Location) which represent
                     channels (e.g. training, validation, testing) where
                     specific parts of the data are saved in S3.
@@ -184,6 +185,10 @@ class SagemakerOrchestratorConfig(
 
     Attributes:
         execution_role: The IAM role ARN to use for the pipeline.
+        scheduler_role: The ARN of the IAM role that will be assumed by
+            the EventBridge service to launch Sagemaker pipelines
+            (For more details regarding the required permissions, please check:
+            https://docs.zenml.io/stack-components/orchestrators/sagemaker#required-iam-permissions-for-schedules)
         aws_access_key_id: The AWS access key ID to use to authenticate to AWS.
             If not provided, the value from the default AWS config will be used.
         aws_secret_access_key: The AWS secret access key to use to authenticate
@@ -203,6 +208,7 @@ class SagemakerOrchestratorConfig(
     """
 
     execution_role: str
+    scheduler_role: Optional[str] = None
     aws_access_key_id: Optional[str] = SecretField(default=None)
     aws_secret_access_key: Optional[str] = SecretField(default=None)
     aws_profile: Optional[str] = None
@@ -232,6 +238,15 @@ class SagemakerOrchestratorConfig(
         """
         return self.synchronous
 
+    @property
+    def is_schedulable(self) -> bool:
+        """Whether the orchestrator is schedulable or not.
+
+        Returns:
+            Whether the orchestrator is schedulable or not.
+        """
+        return True
+
 
 class SagemakerOrchestratorFlavor(BaseOrchestratorFlavor):
     """Flavor for the Sagemaker orchestrator."""