Ensure types for tree attributes

hasadna · Dec 2, 2023 · 6ebd9a1 · 6ebd9a1
1 parent c4a60a2
commit 6ebd9a1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/operators/trees/__init__.py b/operators/trees/__init__.py
@@ -236,8 +236,10 @@ def main(local=False):
             road_id='road_id',
         )),
         DF.set_type('attributes-canopy-area', type='number', on_error=DF.schema_validator.clear),
+        DF.set_type('attributes-canopy-diameter', type='number', on_error=DF.schema_validator.clear),
         DF.set_type('attributes-height', type='number', on_error=DF.schema_validator.clear),
         DF.set_type('attributes-bark-diameter', type='number', on_error=DF.schema_validator.clear),
+        DF.set_type('attributes-bark-circumference', type='number', on_error=DF.schema_validator.clear),
         DF.checkpoint('tree-processing-clusters', CHECKPOINT_PATH)
     ).process()