Refactor sanitize_encoding function parameters

Author: devsjc

src/satellite_consumer/storage.py

@@ -8,7 +8,6 @@
 import fsspec
 import gcsfs
 import icechunk
-import numpy as np
 import pandas as pd
 import s3fs
 import xarray as xr
@@ -24,11 +23,10 @@
 
 def sanitize_encoding(
     ds: xr.Dataset,
+    dims: list[str],
     data: dict[str, Any],
 ) -> dict[str, Any]:
     """Get the encoding dictionary for writing the dataset to Zarr."""
-    dims: list[str] = encoding["_ARRAY_DIMENSIONS"]
-
     if isinstance(data, dict):
         for key, value in data.items():
             if key in ["chunks", "shards"] and isinstance(value, list):
@@ -85,7 +83,7 @@ def write_to_store(
                 obj=ds,
                 session=session,
                 mode="w-",
-                encoding=sanitize_encoding(ds=ds, encoding=encoding),
+                encoding=sanitize_encoding(ds=ds, dims=dims, encoding=encoding),
             )
             _ = session.commit(message="initial commit")
         elif isinstance(dst, str):
@@ -96,7 +94,7 @@ def write_to_store(
                 write_empty_chunks=False,
                 zarr_format=3,
                 compute=True,
-                encoding=sanitize_encoding(ds=ds, encoding=encoding),
+                encoding=sanitize_encoding(ds=ds, dims=dims, encoding=encoding),
             )
         return None