ResNet is not documented, nothing to link to

microsoft · Nov 25, 2021 · 6d8a2aa · 6d8a2aa
1 parent 3fd9bcc
commit 6d8a2aa
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git 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
@@ -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,