From 6d8a2aabe90d2d44beb18694af52f00ab07e3125 Mon Sep 17 00:00:00 2001
From: "Adam J. Stewart" <ajstewart426@gmail.com>
Date: Wed, 24 Nov 2021 22:23:09 -0600
Subject: [PATCH] ResNet is not documented, nothing to link to

---
 docs/conf.py              | 2 ++
 torchgeo/models/resnet.py | 4 ----
 2 files changed, 2 insertions(+), 4 deletions(-)

diff --git a/docs/conf.py b/docs/conf.py
index 4f0d7706c27..41a0e00f17c 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -59,6 +59,8 @@
     ("py:class", ".."),
     # TODO: can't figure out why this isn't found
     ("py:class", "LightningDataModule"),
+    # Undocumented class
+    ("py:class", "torchvision.models.resnet.ResNet"),
 ]
 
 
diff --git a/torchgeo/models/resnet.py b/torchgeo/models/resnet.py
index af3b0e846c8..7c47a4c124b 100644
--- a/torchgeo/models/resnet.py
+++ b/torchgeo/models/resnet.py
@@ -22,10 +22,6 @@
 
 NUM_CLASSES = {"sentinel2": 17}
 
-# https://github.com/pytorch/pytorch/issues/60979
-# https://github.com/pytorch/pytorch/pull/61045
-ResNet.__module__ = "torchvision.models"
-
 
 def _resnet(
     sensor: str,