replace Checks in test_metadata file

NYCPlanning · Nov 26, 2024 · c790836 · c790836
1 parent b986056
commit c790836
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/dcpy/test/models/product/test_metadata.py b/dcpy/test/models/product/test_metadata.py
@@ -3,6 +3,7 @@
 
 from dcpy.models.product import metadata as md
 from dcpy.models.product.dataset import metadata_v2 as ds_md
+from dcpy.models import dataset
 
 
 @pytest.fixture
@@ -178,7 +179,7 @@ def test_column_defaults_applied(dataset_with_snippets: ds_md.Metadata):
             name="uid",
             data_type="text",
             data_source="Department of City Planning",
-            checks=ds_md.Checks(is_primary_key=True),
+            checks=dataset.Checks(is_primary_key=True),
         ),
         ds_md.DatasetColumn(
             id="bbl",