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,