Avoid using UTC alias added in Python 3.11

dispatchrun · Jun 26, 2024 · 6b283a4 · 6b283a4
1 parent 856bf3e
commit 6b283a4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/dispatch/any.py b/src/dispatch/any.py
@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 import pickle
-from datetime import UTC, datetime, timedelta
+from datetime import datetime, timedelta, timezone
 from typing import Any
 
 import google.protobuf.any_pb2
@@ -108,7 +108,7 @@ def unmarshal_any(any: google.protobuf.any_pb2.Any) -> Any:
             return proto.value
 
     elif isinstance(proto, google.protobuf.timestamp_pb2.Timestamp):
-        return proto.ToDatetime(tzinfo=UTC)
+        return proto.ToDatetime(tzinfo=timezone.utc)
 
     elif isinstance(proto, google.protobuf.duration_pb2.Duration):
         return proto.ToTimedelta()

diff --git a/tests/dispatch/test_any.py b/tests/dispatch/test_any.py
@@ -1,5 +1,5 @@
 import pickle
-from datetime import UTC, datetime, timedelta
+from datetime import datetime, timedelta, timezone
 
 from dispatch.any import INT64_MAX, INT64_MIN, marshal_any, unmarshal_any
 from dispatch.sdk.v1 import error_pb2 as error_pb
@@ -70,7 +70,7 @@ def test_unmarshal_bytes():
 
 
 def test_unmarshal_timestamp():
-    ts = datetime.fromtimestamp(1719372909.641448, UTC)
+    ts = datetime.fromtimestamp(1719372909.641448, timezone.utc)
     boxed = marshal_any(ts)
     assert "type.googleapis.com/google.protobuf.Timestamp" == boxed.type_url
     assert ts == unmarshal_any(boxed)