style(ruff) : automatic lint/format

earthdaily · Feb 10, 2025 · 5157b1d · 5157b1d
1 parent 434e886
commit 5157b1d
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 11 deletions.
diff --git a/earthdaily/accessor/__init__.py b/earthdaily/accessor/__init__.py
@@ -133,7 +133,7 @@ def zonal_stats(
         self,
         geometry,
         reducers: list = ["mean"],
-        preserve_columns: bool= True,
+        preserve_columns: bool = True,
         lazy_load=True,
         **kwargs,
     ):

diff --git a/earthdaily/earthdatastore/cube_utils/_zonal.py b/earthdaily/earthdatastore/cube_utils/_zonal.py
@@ -238,18 +238,22 @@ def _loop_time_chunks(dataset, method, smart_load, time_chunks):
             **kwargs,
         )
         zs = zs.drop("geometry")
-        zs = zs.assign_coords(geometry=('feature',geoms.geometry.to_wkt(rounding_precision=-1).values))
-        zs = zs.assign_coords(index=('feature',geoms.index))
-        zs = zs.set_index(feature=["geometry","index"])
-        zs = zs.transpose('time', 'feature', 'zonal_statistics')
-
+        zs = zs.assign_coords(
+            geometry=("feature", geoms.geometry.to_wkt(rounding_precision=-1).values)
+        )
+        zs = zs.assign_coords(index=("feature", geoms.index))
+        zs = zs.set_index(feature=["geometry", "index"])
+        zs = zs.transpose("time", "feature", "zonal_statistics")
+
     # store columns
     if preserve_columns:
-        coords = [col for col in geoms.columns if col not in geoms._geometry_column_name]
+        coords = [
+            col for col in geoms.columns if col not in geoms._geometry_column_name
+        ]
         values = geoms.loc[zs.index.values][coords].values.T
-        for coord, value in zip(coords,values):
-            zs = zs.assign_coords({coord:("feature",value)})
-        feature_index_names = list(zs['feature'].to_index().names)
+        for coord, value in zip(coords, values):
+            zs = zs.assign_coords({coord: ("feature", value)})
+        feature_index_names = list(zs["feature"].to_index().names)
         feature_index_names.extend(coords)
         zs = zs.set_index(feature=feature_index_names)
 

diff --git a/earthdaily/earthdatastore/parallel_search.py b/earthdaily/earthdatastore/parallel_search.py
@@ -116,7 +116,7 @@ def datetime_split(
     # Calculate or convert frequency
     if freq == "auto":
         # Calculate automatic frequency, 1 process per month
-        freq = Timedelta(days=np.ceil(date_diff.days/30)*31)
+        freq = Timedelta(days=np.ceil(date_diff.days / 30) * 31)
     elif isinstance(freq, (int, str)):
         freq = Timedelta(days=int(freq))
     elif not isinstance(freq, Timedelta):