replace checks obj in ingest and dataset.py

NYCPlanning · Dec 19, 2024 · 919a826 · 919a826
1 parent 2ca8dd3
commit 919a826
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/dcpy/models/dataset.py b/dcpy/models/dataset.py
@@ -1,5 +1,5 @@
 from dcpy.models.base import SortedSerializedBase
-from typing import Literal
+from typing import Literal, Any
 
 COLUMN_TYPES = Literal[
     "text",
@@ -14,10 +14,10 @@
 ]
 
 
-# TODO: extend/modify Checks model
-class Checks(SortedSerializedBase):
-    is_primary_key: bool | None = None
-    non_nullable: bool | None = None
+# TODO: DELETE
+# class Checks(SortedSerializedBase):
+#     is_primary_key: bool | None = None
+#     non_nullable: bool | None = None
 
 
 class Column(SortedSerializedBase, extra="forbid"):
@@ -29,4 +29,4 @@ class Column(SortedSerializedBase, extra="forbid"):
     data_type: COLUMN_TYPES | None = None
     description: str | None = None
     is_required: bool = True
-    checks: Checks | None = None
+    checks: list[str | dict[str, Any]] | None = None
diff --git a/dcpy/models/lifecycle/ingest.py b/dcpy/models/lifecycle/ingest.py
@@ -82,7 +82,6 @@ class Column(BaseColumn):
     _head_sort_order = ["id", "data_type", "description"]
 
     data_type: COLUMN_TYPES  # override BaseColumn `data_type` to be required field
-    checks: list[str | dict[str, Any]] | None = None
 
 
 class Template(BaseModel, extra="forbid"):