finish pydantic migration

vanderschaarlab · Sep 27, 2024 · a25fa5c · a25fa5c
1 parent 7708426
commit a25fa5c
Show file tree

Hide file tree

Showing 8 changed files with 819 additions and 258 deletions.
diff --git a/src/synthcity/plugins/core/constraints.py b/src/synthcity/plugins/core/constraints.py
@@ -9,6 +9,8 @@
 # synthcity absolute
 import synthcity.logger as log
 
+Rule = Tuple[str, str, Any]  # Define a type alias for clarity
+
 
 class Constraints(BaseModel):
     """
@@ -41,7 +43,7 @@ class Constraints(BaseModel):
             and thresh is the threshold or data type.
     """
 
-    rules: list = []
+    rules: list[Rule] = []
 
     @field_validator("rules", mode="before")
     def _validate_rules(cls: Any, rules: List) -> List: