process features extras

eolearn/features/extra/__init__.py

@@ -0,0 +1,4 @@
+"""
+A collection of EOTasks with non-standard dependencies. Use the extra `FEATURES_EXTRA` to install dependencies or
+install the module-specific dependencies by hand.
+"""

pyproject.toml

@@ -59,15 +59,15 @@ dependencies = [
 all = [
     "eo-learn[COREGISTRATION,FEATURES,GEOMETRY,IO,MASK,MLTOOLS,VISUALIZATION]",
 ]
-full = ["eo-learn[ALL,RAY,ZARR,METEOBLUE,TDIGEST]"]
+full = ["eo-learn[ALL,RAY,ZARR,METEOBLUE,TDIGEST,FEATURES_EXTRA]"]
 ray = ["ray[default]"]
 zarr = ["s3fs", "zarr"]
 coregistration = [
     "opencv-python-headless",
-] # using headless version of opencv, otherwise gitlab runners fail
-features = [
+] # using headless, otherwise gitlab runners fail
+features = ["opencv-python-headless"]
+features_extra = [
     "numba>=0.53.0",
-    "opencv-python-headless",
     "scikit-image>=0.19.0",
     "scikit-learn",
     "scipy",
@@ -79,7 +79,12 @@ geodb = [
     "xcube-geodb @ git+git://github.com/dcs4cop/xcube-geodb.git",
 ]
 meteoblue = ["meteoblue_dataset_sdk>=1,<2"]
-mask = ["lightgbm>=2.0.11, <4", "opencv-python-headless", "s2cloudless", "scikit-image>=0.13.0"]
+mask = [
+    "lightgbm>=2.0.11, <4",
+    "opencv-python-headless",
+    "s2cloudless",
+    "scikit-image>=0.13.0",
+]
 mltools = ["shapely"]
 tdigest = ["tdigest==0.5.2.2"]
 mltoolsplotting = ["matplotlib"]
@@ -174,7 +179,13 @@ ignore = [
     "B028",  # always demands a stacklevel argument when warning
     "PT011", # complains for `pytest.raises(ValueError)` but we use it a lot
 ]
-per-file-ignores = { "__init__.py" = ["F401", "I002"], "conf.py" = ["I002", "FA100"] }
+per-file-ignores = { "__init__.py" = [
+    "F401",
+    "I002",
+], "conf.py" = [
+    "I002",
+    "FA100",
+] }
 exclude = [".git", "__pycache__", "build", "dist"]
 
 

tests/features/test_clustering.py → tests/features/extra/test_clustering.py

@@ -13,7 +13,7 @@
 import pytest
 
 from eolearn.core import FeatureType
-from eolearn.features import ClusteringTask
+from eolearn.features.extra.clustering import ClusteringTask
 
 logging.basicConfig(level=logging.DEBUG)
 

tests/features/test_interpolation.py → tests/features/extra/test_interpolation.py

@@ -16,7 +16,7 @@
 from sentinelhub.testing_utils import assert_statistics_match
 
 from eolearn.core import EOPatch, EOTask, FeatureType
-from eolearn.features import (
+from eolearn.features.extra.interpolation import (
     InterpolationTask,
     KrigingInterpolationTask,
     LinearInterpolationTask,