spai 2026.3.18.dev0__tar.gz → 2026.3.35.dev0__tar.gz

{spai-2026.3.18.dev0 → spai-2026.3.35.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: spai
-Version: 2026.3.18.dev0
+Version: 2026.3.35.dev0
 Author-email: Jan Sensio <it@earthpulse.es>
 Requires-Python: >=3.12
 Requires-Dist: bson>=0.5.10

{spai-2026.3.18.dev0 → spai-2026.3.35.dev0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "spai"
-version = "2026.03.18.dev0"
+version = "2026.03.35.dev0"
 description = ""
 authors = [{name = "Jan Sensio", email = "it@earthpulse.es"}]
 readme = "README.md"

spai-2026.3.35.dev0/spai/__init__.py

@@ -0,0 +1 @@
+__version__ = "2026.03.35.dev0"

{spai-2026.3.18.dev0 → spai-2026.3.35.dev0}/spai/data/satellite/load.py

@@ -7,24 +7,10 @@ from datetime import datetime
 
 from . import DOWNLOADERS, AVAILABLE_COLLECTIONS, STATIC_COLLECTIONS
 from .stac.STACDownloader import STACDownloader
-from .stac.pc.PCDownloader import PCDownloader
 from .utils import create_aoi_geodataframe
 from .explore import explore_satellite_imagery
 
 
-# when running for more than 45 mins, a script that downloads data from planetary computer will fail with a token error
-# this function clears the token cache to avoid this error
-def _refresh_planetary_computer_tokens():
-    try:
-        from planetary_computer.sas import TOKEN_CACHE
-
-        TOKEN_CACHE.clear()
-
-        print("Planetary Computer tokens refreshed", flush=True)
-    except ImportError:
-        pass
-
-
 def load_satellite_imagery(
     aoi: Any,
     date: Optional[List[Union[str, datetime]]] = None,
@@ -85,11 +71,6 @@ def load_satellite_imagery(
                 print("Try introducing a date parameter to load a specific image.")
                 return
 
-    if issubclass(
-        DOWNLOADERS[collection], PCDownloader
-    ):  # not sure if should apply to other Downloaders as well
-        _refresh_planetary_computer_tokens()
-
     downloader = DOWNLOADERS[collection](bbox, date, **kwargs)
     data = downloader.load_stac()
 

{spai-2026.3.18.dev0 → spai-2026.3.35.dev0}/spai/models/Config.py

@@ -17,6 +17,7 @@ class Resources(BaseModel):
     cpu: Optional[Union[str, float]] = None
     memory: Optional[str] = None
     gpu: Optional[int] = None
+    ephemeral_storage: Optional[str] = None
 
     def __init__(self, **data):
         super().__init__(**data)
@@ -32,6 +33,12 @@ class Resources(BaseModel):
             raise ValueError("CPU must be positive.")
         if self.gpu is not None and self.gpu < 0:
             raise ValueError("GPU must be positive.")
+        if self.ephemeral_storage is not None:
+            pattern = r"^\d+(Ei|Pi|Ti|Gi|Mi|Ki)$"
+            if not re.match(pattern, self.ephemeral_storage):
+                raise ValueError(
+                    "Ephemeral storage must be a string with a number followed by a unit (Ei, Pi, Ti, Gi, Mi, Ki), such as 2Gi. "
+                )
 
 
 class Autoscaling(BaseModel):

{spai-2026.3.18.dev0 → spai-2026.3.35.dev0}/spai/storage/S3Storage.py

@@ -154,18 +154,16 @@ class S3Storage(BaseStorage):
 
     @with_rioxarray
     def create_from_rioxarray(self, rxr, data, name):
-        # TODO don't save to disk
         content_type = "image/tiff"
-        tmp_path = f"/tmp/{name}"
-        self._create_dir(tmp_path)
-        data.rio.to_raster(tmp_path)
-        self.client.fput_object(
-            self.bucket,
-            name,
-            tmp_path,
-            content_type=content_type,
-        )
-        # shutil.rmtree(os.path.dirname(tmp_path))
+        with tempfile.TemporaryDirectory() as tmp_dir:
+            tmp_path = os.path.join(tmp_dir, name)
+            data.rio.to_raster(tmp_path)
+            self.client.fput_object(
+                self.bucket,
+                name,
+                tmp_path,
+                content_type=content_type,
+            )
         return self.get_url(name)
 
     def create_from_array(self, data, name):
@@ -322,7 +320,6 @@ class S3Storage(BaseStorage):
 
     @with_xarray
     def create_from_netcdf(self, xr, data, name):
-
         with tempfile.NamedTemporaryFile() as temp_file:
             data.to_netcdf(temp_file.name)
             self.client.fput_object(

spai-2026.3.18.dev0/spai/__init__.py

@@ -1 +0,0 @@
-__version__ = "2026.03.18.dev0"