feature.py: Add static methods.

PiperOrigin-RevId: 645137751
google-research · Jun 21, 2024 · 6cb376a · 6cb376a
1 parent 8ffb4bf
commit 6cb376a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/skai/earth_engine.py b/src/skai/earth_engine.py
@@ -31,7 +31,7 @@
 ShapelyGeometry = shapely.geometry.base.BaseGeometry
 
 
-def _shapely_to_ee_feature(shapely_geometry: ShapelyGeometry) -> ee.Feature:
+def _shapely_to_ee_feature(shapely_geometry: ShapelyGeometry) -> ee.Geometry:
   """Converts shapely geometry into Earth Engine Feature."""
   geojson = json.loads(gpd.GeoSeries([shapely_geometry]).to_json())
   return ee.Feature(geojson['features'][0]['geometry']).geometry()