More tacoreader to optional dependencies

rosielickorish · rosielickorish · commit ad266fb08ad2 · 2026-05-19T20:35:08.000+01:00
diff --git a/README.md b/README.md
@@ -102,7 +102,22 @@ git clone https://github.com/torchgeo/terratorch.git
 cd terratorch
 pip install -e .[test]
 ```
-To install terratorch with partial (work in development) support for Weather Foundation Models, `pip install -e .[wxc]`, which currently works just for `Python >= 3.11`. 
+
+### Optional Dependencies
+
+TerraTorch supports several optional features that can be installed separately:
+
+- **VLLM support**: `pip install terratorch[vllm]`
+- **Weather Foundation Models**: `pip install terratorch[wxc]` (Python >= 3.11 only)
+- **PEFT (Parameter-Efficient Fine-Tuning)**: `pip install terratorch[peft]`
+- **Visualization tools**: `pip install terratorch[visualize]`
+- **GeoBench v2**: `pip install terratorch[geobenchv2]`
+- **Logging with Weights & Biases**: `pip install terratorch[logging]`
+- **MMSegmentation support**: `pip install terratorch[mmseg]`
+- **Surya support**: `pip install terratorch[surya]`
+- **Tortilla file support**: `pip install terratorch[tortilla]` - Required for loading datasets from tortilla files
+
+You can install multiple optional dependencies at once: `pip install terratorch[vllm,peft,logging]`
 
 ## Documentation
 
diff --git a/pyproject.toml b/pyproject.toml
@@ -66,7 +66,6 @@ dependencies = [
   "tifffile",
   "python-box",
   "tqdm",
-  "tacoreader==0.5.6",
   "tensorboard",
   "diffusers",
   "scikit-learn>=1.3.2",
@@ -156,6 +155,10 @@ vllm_test = [
   "imagehash",
 ]
 
+tortilla = [
+  "tacoreader==0.5.6",
+]
+
 [project.urls]
 Documentation = "https://torchgeo.github.io/terratorch/"
 Issues = "https://github.com/torchgeo/terratorch/issues"
diff --git a/terratorch/datamodules/generic_pixel_wise_data_module.py b/terratorch/datamodules/generic_pixel_wise_data_module.py
@@ -13,9 +13,16 @@
 import albumentations as A
 import kornia.augmentation as K
 import numpy as np
-import tacoreader
 import torch
 from albumentations.core.composition import BaseCompose, Compose
+
+try:
+    import tacoreader
+
+    HAS_TACOREADER = True
+except ImportError:
+    HAS_TACOREADER = False
+    tacoreader = None  # type: ignore
 from kornia.augmentation import AugmentationSequential
 from torch import Tensor
 from torch.utils.data import DataLoader
@@ -221,7 +228,12 @@ def __init__(
                 "Use tortilla_file OR data roots (not both)."
             )
 
-        self.tortilla_df = tacoreader.load(str(tortilla_file)) if tortilla_file is not None else None
+        if tortilla_file is not None and not HAS_TACOREADER:
+            raise ImportError(
+                "tacoreader is required to use tortilla files. Install it with: pip install terratorch[tortilla]"
+            )
+
+        self.tortilla_df = tacoreader.load(str(tortilla_file)) if tortilla_file is not None else None  # type: ignore
 
     def _get_tortilla_indices(self, stage: str) -> list[Hashable] | None:
         if self.tortilla_df is None:
