zenml-nightly 0.73.0.dev20250124__py3-none-any.whl → 0.73.0.dev20250125__py3-none-any.whl

zenml/zen_stores/schemas/secret_schemas.py

@@ -15,7 +15,6 @@
 
 import base64
 import json
-from datetime import datetime, timezone
 from typing import Any, Dict, Optional, cast
 from uuid import UUID
 
@@ -35,6 +34,7 @@ from zenml.models import (
     SecretResponseMetadata,
     SecretUpdate,
 )
+from zenml.utils.time_utils import utc_now
 from zenml.zen_stores.schemas.base_schemas import NamedSchema
 from zenml.zen_stores.schemas.schema_utils import build_foreign_key_field
 from zenml.zen_stores.schemas.user_schemas import UserSchema
@@ -209,7 +209,7 @@ class SecretSchema(NamedSchema, table=True):
             else:
                 setattr(self, field, value)
 
-        self.updated = datetime.now(timezone.utc)
+        self.updated = utc_now()
         return self
 
     def to_model(

zenml/zen_stores/schemas/server_settings_schemas.py

@@ -14,7 +14,7 @@
 """SQLModel implementation for the server settings table."""
 
 import json
-from datetime import datetime, timezone
+from datetime import datetime
 from typing import Any, Optional, Set
 from uuid import UUID
 
@@ -27,6 +27,7 @@ from zenml.models import (
     ServerSettingsResponseResources,
     ServerSettingsUpdate,
 )
+from zenml.utils.time_utils import utc_now
 
 
 class ServerSettingsSchema(SQLModel, table=True):
@@ -42,12 +43,8 @@ class ServerSettingsSchema(SQLModel, table=True):
     display_announcements: Optional[bool] = Field(nullable=True)
     display_updates: Optional[bool] = Field(nullable=True)
     onboarding_state: Optional[str] = Field(nullable=True)
-    last_user_activity: datetime = Field(
-        default_factory=lambda: datetime.now(timezone.utc)
-    )
-    updated: datetime = Field(
-        default_factory=lambda: datetime.now(timezone.utc)
-    )
+    last_user_activity: datetime = Field(default_factory=utc_now)
+    updated: datetime = Field(default_factory=utc_now)
 
     def update(
         self, settings_update: ServerSettingsUpdate
@@ -67,7 +64,7 @@ class ServerSettingsSchema(SQLModel, table=True):
             if hasattr(self, field):
                 setattr(self, field, value)
 
-        self.updated = datetime.now(timezone.utc)
+        self.updated = utc_now()
 
         return self
 
@@ -87,7 +84,7 @@ class ServerSettingsSchema(SQLModel, table=True):
         )
         new_state = old_state.union(completed_steps)
         self.onboarding_state = json.dumps(list(new_state))
-        self.updated = datetime.now(timezone.utc)
+        self.updated = utc_now()
 
         return self
 

zenml/zen_stores/schemas/service_connector_schemas.py

@@ -15,7 +15,7 @@
 
 import base64
 import json
-from datetime import datetime, timezone
+from datetime import datetime
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, cast
 from uuid import UUID
 
@@ -29,6 +29,7 @@ from zenml.models import (
     ServiceConnectorResponseMetadata,
     ServiceConnectorUpdate,
 )
+from zenml.utils.time_utils import utc_now
 from zenml.zen_stores.schemas.base_schemas import NamedSchema
 from zenml.zen_stores.schemas.schema_utils import build_foreign_key_field
 from zenml.zen_stores.schemas.user_schemas import UserSchema
@@ -227,7 +228,7 @@ class ServiceConnectorSchema(NamedSchema, table=True):
             else:
                 setattr(self, field, value)
         self.secret_id = secret_id
-        self.updated = datetime.now(timezone.utc)
+        self.updated = utc_now()
         return self
 
     def to_model(

zenml/zen_stores/schemas/service_schemas.py

@@ -15,7 +15,6 @@
 
 import base64
 import json
-from datetime import datetime, timezone
 from typing import Any, Optional
 from uuid import UUID
 
@@ -32,6 +31,7 @@ from zenml.models.v2.core.service import (
     ServiceUpdate,
 )
 from zenml.utils.dict_utils import dict_to_bytes
+from zenml.utils.time_utils import utc_now
 from zenml.zen_stores.schemas.base_schemas import NamedSchema
 from zenml.zen_stores.schemas.model_schemas import ModelVersionSchema
 from zenml.zen_stores.schemas.pipeline_run_schemas import PipelineRunSchema
@@ -210,7 +210,7 @@ class ServiceSchema(NamedSchema, table=True):
                 )
             else:
                 setattr(self, field, value)
-        self.updated = datetime.now(timezone.utc)
+        self.updated = utc_now()
         return self
 
     @classmethod

zenml/zen_stores/schemas/stack_schemas.py

@@ -15,7 +15,6 @@
 
 import base64
 import json
-from datetime import datetime, timezone
 from typing import TYPE_CHECKING, Any, List, Optional
 from uuid import UUID
 
@@ -27,6 +26,7 @@ from zenml.models import (
     StackResponseMetadata,
     StackUpdate,
 )
+from zenml.utils.time_utils import utc_now
 from zenml.zen_stores.schemas.base_schemas import NamedSchema
 from zenml.zen_stores.schemas.schema_utils import build_foreign_key_field
 from zenml.zen_stores.schemas.user_schemas import UserSchema
@@ -134,7 +134,7 @@ class StackSchema(NamedSchema, table=True):
             else:
                 setattr(self, field, value)
 
-        self.updated = datetime.now(timezone.utc)
+        self.updated = utc_now()
         return self
 
     def to_model(

zenml/zen_stores/schemas/step_run_schemas.py

@@ -14,7 +14,7 @@
 """SQLModel implementation of step run tables."""
 
 import json
-from datetime import datetime, timezone
+from datetime import datetime
 from typing import TYPE_CHECKING, Any, Dict, List, Optional
 from uuid import UUID
 
@@ -43,6 +43,7 @@ from zenml.models.v2.core.step_run import (
     StepRunInputResponse,
     StepRunResponseResources,
 )
+from zenml.utils.time_utils import utc_now
 from zenml.zen_stores.schemas.base_schemas import NamedSchema
 from zenml.zen_stores.schemas.constants import MODEL_VERSION_TABLENAME
 from zenml.zen_stores.schemas.pipeline_deployment_schemas import (
@@ -358,7 +359,7 @@ class StepRunSchema(NamedSchema, RunMetadataInterface, table=True):
                 if value and self.model_version_id is None:
                     self.model_version_id = value
 
-        self.updated = datetime.now(timezone.utc)
+        self.updated = utc_now()
 
         return self
 

zenml/zen_stores/schemas/tag_schemas.py

@@ -13,7 +13,6 @@
 #  permissions and limitations under the License.
 """SQLModel implementation of tag tables."""
 
-from datetime import datetime, timezone
 from typing import Any, List
 from uuid import UUID
 
@@ -30,6 +29,7 @@ from zenml.models import (
     TagResponseBody,
     TagUpdate,
 )
+from zenml.utils.time_utils import utc_now
 from zenml.zen_stores.schemas.base_schemas import BaseSchema, NamedSchema
 from zenml.zen_stores.schemas.schema_utils import build_foreign_key_field
 
@@ -103,7 +103,7 @@ class TagSchema(NamedSchema, table=True):
             else:
                 setattr(self, field, value)
 
-        self.updated = datetime.now(timezone.utc)
+        self.updated = utc_now()
         return self
 
 

zenml/zen_stores/schemas/trigger_schemas.py

@@ -15,7 +15,6 @@
 
 import base64
 import json
-from datetime import datetime, timezone
 from typing import Any, List, Optional, cast
 from uuid import UUID
 
@@ -38,6 +37,7 @@ from zenml.models import (
     TriggerUpdate,
 )
 from zenml.utils.json_utils import pydantic_encoder
+from zenml.utils.time_utils import utc_now
 from zenml.zen_stores.schemas.action_schemas import ActionSchema
 from zenml.zen_stores.schemas.base_schemas import BaseSchema, NamedSchema
 from zenml.zen_stores.schemas.event_source_schemas import EventSourceSchema
@@ -133,7 +133,7 @@ class TriggerSchema(NamedSchema, table=True):
             else:
                 setattr(self, field, value)
 
-        self.updated = datetime.now(timezone.utc)
+        self.updated = utc_now()
         return self
 
     @classmethod

zenml/zen_stores/schemas/user_schemas.py

@@ -14,7 +14,6 @@
 """SQLModel implementation of user tables."""
 
 import json
-from datetime import datetime, timezone
 from typing import TYPE_CHECKING, Any, List, Optional
 from uuid import UUID
 
@@ -33,6 +32,7 @@ from zenml.models import (
     UserResponseMetadata,
     UserUpdate,
 )
+from zenml.utils.time_utils import utc_now
 from zenml.zen_stores.schemas.base_schemas import NamedSchema
 
 if TYPE_CHECKING:
@@ -225,7 +225,7 @@ class UserSchema(NamedSchema, table=True):
             else:
                 setattr(self, field, value)
 
-        self.updated = datetime.now(timezone.utc)
+        self.updated = utc_now()
         return self
 
     def update_service_account(
@@ -245,7 +245,7 @@ class UserSchema(NamedSchema, table=True):
         ).items():
             setattr(self, field, value)
 
-        self.updated = datetime.now(timezone.utc)
+        self.updated = utc_now()
         return self
 
     def to_model(

zenml/zen_stores/schemas/workspace_schemas.py

@@ -13,7 +13,6 @@
 #  permissions and limitations under the License.
 """SQL Model Implementations for Workspaces."""
 
-from datetime import datetime, timezone
 from typing import TYPE_CHECKING, Any, List
 
 from sqlmodel import Relationship
@@ -25,6 +24,7 @@ from zenml.models import (
     WorkspaceResponseMetadata,
     WorkspaceUpdate,
 )
+from zenml.utils.time_utils import utc_now
 from zenml.zen_stores.schemas.base_schemas import NamedSchema
 
 if TYPE_CHECKING:
@@ -168,7 +168,7 @@ class WorkspaceSchema(NamedSchema, table=True):
         ).items():
             setattr(self, field, value)
 
-        self.updated = datetime.now(timezone.utc)
+        self.updated = utc_now()
         return self
 
     def to_model(

zenml/zen_stores/sql_zen_store.py

@@ -22,7 +22,7 @@ import random
 import re
 import sys
 import time
-from datetime import datetime, timezone
+from datetime import datetime
 from functools import lru_cache
 from pathlib import Path
 from typing import (
@@ -309,6 +309,7 @@ from zenml.utils.string_utils import (
     random_str,
     validate_name,
 )
+from zenml.utils.time_utils import utc_now
 from zenml.zen_stores import template_utils
 from zenml.zen_stores.base_zen_store import (
     BaseZenStore,
@@ -795,7 +796,7 @@ class SqlZenStoreConfiguration(StoreConfiguration):
                     logger.warning(
                         f"Database SSL setting `{key}` is not a file. "
                     )
-                sqlalchemy_ssl_args[key.lstrip("ssl_")] = ssl_setting
+                sqlalchemy_ssl_args[key.removeprefix("ssl_")] = ssl_setting
             if len(sqlalchemy_ssl_args) > 0:
                 sqlalchemy_ssl_args["check_hostname"] = (
                     self.ssl_verify_server_cert
@@ -1743,7 +1744,7 @@ class SqlZenStore(BaseZenStore):
             settings = self._get_server_settings(session=session)
 
             if last_user_activity < settings.last_user_activity.replace(
-                tzinfo=timezone.utc
+                tzinfo=None
             ):
                 return
 
@@ -4048,7 +4049,7 @@ class SqlZenStore(BaseZenStore):
                 # Delete devices that have expired
                 if (
                     device.expires is not None
-                    and device.expires < datetime.now()
+                    and device.expires < utc_now()
                     and device.user_id is None
                 ):
                     session.delete(device)
@@ -8634,19 +8635,10 @@ class SqlZenStore(BaseZenStore):
                 ExecutionStatus.COMPLETED,
                 ExecutionStatus.FAILED,
             }:
-                run_update.end_time = datetime.now(timezone.utc)
+                run_update.end_time = utc_now()
                 if pipeline_run.start_time and isinstance(
                     pipeline_run.start_time, datetime
                 ):
-                    # We need to ensure both datetimes are timezone-aware to avoid TypeError when subtracting
-                    # Now calculate the duration time
-                    if pipeline_run.start_time.tzinfo is None:
-                        pipeline_run.start_time = (
-                            pipeline_run.start_time.replace(
-                                tzinfo=timezone.utc
-                            )
-                        )
-
                     duration_time = (
                         run_update.end_time - pipeline_run.start_time
                     )

{zenml_nightly-0.73.0.dev20250124.dist-info → zenml_nightly-0.73.0.dev20250125.dist-info}/METADATA

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