add test for Smart Surface LULC

wri · Jun 3, 2024 · d053023 · d053023
1 parent 4329fb6
commit d053023
Show file tree

Hide file tree

Showing 4 changed files with 747 additions and 62 deletions.
diff --git a/city_metrix/layers/__init__.py b/city_metrix/layers/__init__.py
@@ -3,7 +3,7 @@
 from .land_surface_temperature import LandSurfaceTemperature
 from .tree_cover import TreeCover
 from .high_land_surface_temperature import HighLandSurfaceTemperature
-from .smart_cities_lulc import SmartCitiesLULC
+from .smart_surface_lulc import SmartSurfaceLULC
 from .open_street_map import OpenStreetMap, OpenStreetMapClass
 from .urban_land_use import UrbanLandUse
 from .natural_areas import NaturalAreas

diff --git a/city_metrix/layers/smart_cities_lulc.py → city_metrix/layers/smart_surface_lulc.py b/city_metrix/layers/smart_cities_lulc.py → city_metrix/layers/smart_surface_lulc.py
@@ -15,7 +15,7 @@
 from .building_classifier.building_classifier import BuildingClassifier
 
 
-class SmartCitiesLULC(Layer):
+class SmartSurfaceLULC(Layer):
     def __init__(self, land_cover_class=None, **kwargs):
         super().__init__(**kwargs)
         self.land_cover_class = land_cover_class
@@ -125,6 +125,7 @@ def get_data(self, bbox):
         # use chunk 512x512
         aligned_datasets = [ds.chunk({'x': 512, 'y': 512}) for ds in aligned_datasets]
         lulc = xr.concat(aligned_datasets, dim='Value').max(dim='Value')
+        lulc = lulc.compute()
 
         return lulc