Use datetime.utcnow() to avoid timezone issues

Signed-off-by: Judah Rand <[email protected]>
feast-dev · Feb 2, 2022 · 2de9394 · 2de9394
1 parent 592af75
commit 2de9394
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/sdk/python/feast/registry.py b/sdk/python/feast/registry.py
@@ -374,7 +374,7 @@ def apply_feature_view(
         """
         feature_view.ensure_valid()
         if not feature_view.created_timestamp:
-            feature_view.created_timestamp = datetime.now()
+            feature_view.created_timestamp = datetime.utcnow()
         feature_view_proto = feature_view.to_proto()
         feature_view_proto.spec.project = project
         self._prepare_registry_for_changes()
@@ -819,7 +819,7 @@ def _prepare_registry_for_changes(self):
             registry_proto = RegistryProto()
             registry_proto.registry_schema_version = REGISTRY_SCHEMA_VERSION
             self.cached_registry_proto = registry_proto
-            self.cached_registry_proto_created = datetime.now()
+            self.cached_registry_proto_created = datetime.utcnow()
         return self.cached_registry_proto
 
     def _get_registry_proto(self, allow_cache: bool = False) -> RegistryProto:
@@ -838,7 +838,7 @@ def _get_registry_proto(self, allow_cache: bool = False) -> RegistryProto:
                 self.cached_registry_proto_ttl.total_seconds()
                 > 0  # 0 ttl means infinity
                 and (
-                    datetime.now()
+                    datetime.utcnow()
                     > (
                         self.cached_registry_proto_created
                         + self.cached_registry_proto_ttl
@@ -852,7 +852,7 @@ def _get_registry_proto(self, allow_cache: bool = False) -> RegistryProto:
 
             registry_proto = self._registry_store.get_registry_proto()
             self.cached_registry_proto = registry_proto
-            self.cached_registry_proto_created = datetime.now()
+            self.cached_registry_proto_created = datetime.utcnow()
 
             return registry_proto