From 9af5e8d9b05416b5528172405967d84b027d61b1 Mon Sep 17 00:00:00 2001
From: Caleb Robinson <calebrob6@gmail.com>
Date: Fri, 23 Feb 2024 21:43:49 +0000
Subject: [PATCH] Adding torchgeo datasets install

---
 docs/tutorials/custom_segmentation_trainer.ipynb | 2 +-
 docs/tutorials/custom_segmentation_trainer.py    | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/docs/tutorials/custom_segmentation_trainer.ipynb b/docs/tutorials/custom_segmentation_trainer.ipynb
index bdcac130ef5..5c04d64333a 100644
--- 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
index db77850b795..475456ffd83 100644
--- 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
 #