Adding torchgeo datasets install

microsoft · Feb 23, 2024 · 9af5e8d · 9af5e8d
1 parent 1681fe5
commit 9af5e8d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docs/tutorials/custom_segmentation_trainer.ipynb b/docs/tutorials/custom_segmentation_trainer.ipynb
@@ -38,7 +38,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "%pip install torchgeo"
+    "%pip install torchgeo[datasets]"
    ]
   },
   {

diff --git a/docs/tutorials/custom_segmentation_trainer.py b/docs/tutorials/custom_segmentation_trainer.py
@@ -30,7 +30,7 @@
 #
 # As always, we install TorchGeo.
 
-# %pip install torchgeo
+# %pip install torchgeo[datasets]
 
 # ## Imports
 #