satcube 0.1.11__py3-none-any.whl → 0.1.13__py3-none-any.whl

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] = []

{satcube-0.1.11.dist-info → satcube-0.1.13.dist-info}/METADATA

@@ -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.dist-info → satcube-0.1.13.dist-info}/RECORD

@@ -1,5 +1,5 @@
 satcube/__init__.py,sha256=WxfiBldO7YXWJFATRjbhWLOnD1mk0b8Pc6HGQYOJZJA,221
-satcube/cloud_detection.py,sha256=QSzeqCF63kJHzejbrXKL10x-kLxK0iorHvJNALoHYbA,5848
+satcube/cloud_detection.py,sha256=nSDwBd4oboPy5JKsbI9hUD2cYrIi2RQd27esl_eHvBs,6325
 satcube/cloud_detection_old.py,sha256=7MviF8QlT2tj6QC3seepr8SZU0nHOK9Pji8AU94Z2q0,751
 satcube/dataclass.py,sha256=TAAKouyTts5eMtVXRmwWgJb5EaUXryEtkKMtt1O8TKM,934
 satcube/download.py,sha256=xwPIm6SWN_cev2wM0OzqN18ejPgnjuNVPxfeF7FhI9c,1397
@@ -7,7 +7,7 @@ satcube/download_old.py,sha256=Y7dUgq7Gl7jVKHZl5x9cpxALN9T6dCjtZwNjBiq1CAA,2647
 satcube/main.py,sha256=BpQJbPXl6Ydj6X3pX2lFepH7w1-cfJ3LeTTsTmQih6s,14841
 satcube/utils.py,sha256=wQl4ZSrocSZSSU6hjwCA2CYIv8p3deiX8PUKxJRr3yc,1952
 satcube/utils_old.py,sha256=UBCI2oaL7E5MEjebobnyqGqgOtK6jU9O3t-c58JqZ0k,35057
-satcube-0.1.11.dist-info/LICENSE,sha256=YdB4BQMkMzWuKvXRIpQR4g91IQ_pwA5PSH2lNM97zFI,1070
-satcube-0.1.11.dist-info/METADATA,sha256=pqGa2vAI5SKhyZKXaJMxg5Uzw5DfnRsESoLbncZgG1k,6686
-satcube-0.1.11.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-satcube-0.1.11.dist-info/RECORD,,
+satcube-0.1.13.dist-info/LICENSE,sha256=YdB4BQMkMzWuKvXRIpQR4g91IQ_pwA5PSH2lNM97zFI,1070
+satcube-0.1.13.dist-info/METADATA,sha256=4HgT2yGq_gX8ky_inhlCZsWyylFDJS3B71wH5HZXT-0,6686
+satcube-0.1.13.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+satcube-0.1.13.dist-info/RECORD,,