Get rid of IPV4Address

Author: dandavison

tests/contrib/test_pydantic.py

@@ -1,6 +1,5 @@
+import datetime
 import uuid
-from datetime import datetime, timedelta
-from ipaddress import IPv4Address
 from typing import List
 
 from pydantic import BaseModel
@@ -12,8 +11,7 @@
 
 
 class MyPydanticModel(BaseModel):
-    some_ip: IPv4Address
-    some_date: datetime
+    some_date: datetime.datetime
 
 
 @activity.defn
@@ -28,7 +26,7 @@ class MyWorkflow:
     async def run(self, models: List[MyPydanticModel]) -> List[MyPydanticModel]:
         workflow.logger.info("Got models in workflow: %s" % models)
         return await workflow.execute_activity(
-            my_activity, models, start_to_close_timeout=timedelta(minutes=1)
+            my_activity, models, start_to_close_timeout=datetime.timedelta(minutes=1)
         )
 
 
@@ -41,12 +39,10 @@ async def test_workflow_with_pydantic_model(client: Client):
 
     orig_models = [
         MyPydanticModel(
-            some_ip=IPv4Address("127.0.0.1"),
-            some_date=datetime(2000, 1, 2, 3, 4, 5),
+            some_date=datetime.datetime(2000, 1, 2, 3, 4, 5),
         ),
         MyPydanticModel(
-            some_ip=IPv4Address("127.0.0.2"),
-            some_date=datetime(2001, 2, 3, 4, 5, 6),
+            some_date=datetime.datetime(2001, 2, 3, 4, 5, 6),
         ),
     ]