diff --git a/src/dispatch/any.py b/src/dispatch/any.py index 92dda7d..4933990 100644 --- 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 index 28fd2e5..4cb23e7 100644 --- 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)