Support pydantic v2

PrefectHQ · Nov 14, 2023 · 3b501e2 · 3b501e2
1 parent 68b02b8
commit 3b501e2
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/prefect_aws/glue_job.py b/prefect_aws/glue_job.py
@@ -7,7 +7,13 @@
 
 from prefect.infrastructure.base import Infrastructure, InfrastructureResult
 from prefect.utilities.asyncutils import run_sync_in_worker_thread, sync_compatible
-from pydantic import Field
+from pydantic import VERSION as PYDANTIC_VERSION
+
+if PYDANTIC_VERSION.startswith("2."):
+    from pydantic.v1 import Field
+else:
+    from pydantic import Field
+
 from typing_extensions import Literal
 
 from prefect_aws import AwsCredentials