Assert rehab costs are numeric type

nismod · Apr 2, 2024 · 2b3298a · 2b3298a
1 parent c2a7a29
commit 2b3298a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/open_gira/io.py b/src/open_gira/io.py
@@ -289,7 +289,7 @@ def read_rehab_costs(path: str) -> pd.DataFrame:
     assert costs.asset_type.dtype == object
 
     # check costs
-    assert costs.iloc[:, 1].dtype == float
+    assert pd.api.types.is_numeric_dtype(costs.iloc[:, 1].dtype)
     assert (costs.iloc[:, 1] >= 0).all()
 
-    return costs
+    return costs