satcube 0.1.11__tar.gz → 0.1.13__tar.gz

{satcube-0.1.11 → satcube-0.1.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: satcube
-Version: 0.1.11
+Version: 0.1.13
 Summary: A Python package to create cloud-free monthly composites by fusing Landsat and Sentinel-2 data.
 Home-page: https://github.com/IPL-UV/satcube
 Author: Cesar Aybar
@@ -11,7 +11,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Provides-Extra: full
-Requires-Dist: cubexpress (>=0.1.4)
+Requires-Dist: cubexpress (>=0.1.7)
 Requires-Dist: earthengine-api (>=1.5.12)
 Requires-Dist: mlstac (>=0.4.0)
 Requires-Dist: numpy (>=1.25.0)

{satcube-0.1.11 → satcube-0.1.13}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name    = "satcube"
-version = "0.1.11"
+version = "0.1.13"
 description = "A Python package to create cloud-free monthly composites by fusing Landsat and Sentinel-2 data."
 authors = ["Cesar Aybar <fcesar.aybar@uv.es>"]
 repository     = "https://github.com/IPL-UV/satcube"
@@ -10,7 +10,7 @@ packages       = [{ include = "satcube" }]
 
 [tool.poetry.dependencies]
 python          = ">=3.10,<4.0"
-cubexpress      = ">=0.1.4"
+cubexpress      = ">=0.1.7"
 mlstac          = ">=0.4.0"
 satalign        = ">=0.1.11"         
 segmentation-models-pytorch = ">=0.3.0"

{satcube-0.1.11 → satcube-0.1.13}/satcube/cloud_detection.py

@@ -22,8 +22,16 @@ import rasterio as rio
 import torch
 
 from satcube.utils import DeviceManager, _reset_gpu
+import warnings, re
 
 
+warnings.filterwarnings(
+    "ignore",
+    message=re.escape("The secret `HF_TOKEN` does not exist in your Colab secrets."),
+    category=UserWarning,
+    module="huggingface_hub.utils._auth",
+)
+
 def cloud_masking(
     input: str | Path,              # noqa: A002 (shadowing built-in is OK here)
     output: str | Path,
@@ -32,7 +40,7 @@ def cloud_masking(
     pad: int = 64,
     save_mask: bool = False,
     device: str = "cpu",
-    max_pix_cpu: float = 7.0e7,
+    max_pix_cpu: float = 7.0e7
 ) -> List[Path]:
     """Write cloud-masked Sentinel-2 images.
 
@@ -52,7 +60,7 @@ def cloud_masking(
         Tile images larger than this when running on CPU.
 
     Returns
-    -------
+    ------
     list[pathlib.Path]
         Paths to the generated masked images.
     """
@@ -75,8 +83,18 @@ def cloud_masking(
     if not tif_paths:
         print(f"[cloud_masking] No .tif files found in {src}")
         return []
+    
+    dir = Path("SEN2CloudEnsemble")
+
+    if not dir.exists():
+
+        mlstac.download(
+            file = "https://huggingface.co/tacofoundation/CloudSEN12-models/resolve/main/SEN2CloudEnsemble/mlm.json",
+            output_dir = "SEN2CloudEnsemble",
+        )
+
+    experiment = mlstac.load(dir.as_posix())
 
-    experiment = mlstac.load("SEN2CloudEnsemble")
     dm = DeviceManager(experiment, init_device=device)
 
     masked_paths: list[Path] = []