support python 312 with pydantic validator

square · Jul 18, 2024 · 9a8b8c5 · 9a8b8c5
1 parent 70a30cb
commit 9a8b8c5
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/block_cascade/executors/vertex/resource.py b/block_cascade/executors/vertex/resource.py
@@ -1,4 +1,5 @@
 from dataclasses import field
+import sys
 from typing import Iterable, Optional, Type, TypeVar
 
 from pydantic import validator
@@ -11,6 +12,12 @@
 T = TypeVar("T", bound="GcpEnvironmentConfig")
 
 
+if sys.version_info.minor == 12:
+    pydantic_validator = lambda *args: validator(*args, allow_reuse=True)
+else:
+    pydantic_validator = validator
+
+
 @dataclass(frozen=True)
 class GcpAcceleratorConfig:
     """
@@ -101,7 +108,7 @@ class GcpEnvironmentConfig:
     network: Optional[str] = None
     image: Optional[str] = None
 
-    @validator("image", allow_resuse=True)
+    @pydantic_validator("image")
     def image_setter(cls, v, values):  # noqa: N805
         image = v
         # No image specified