Documented type is a combination of types

microsoft · Dec 24, 2024 · 56e95a3 · 56e95a3
1 parent 4728f5f
commit 56e95a3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/docs/conf.py b/docs/conf.py
@@ -59,7 +59,9 @@
     ('py:class', 'fiona.model.Feature'),
     ('py:class', 'kornia.augmentation._2d.intensity.base.IntensityAugmentationBase2D'),
     ('py:class', 'kornia.augmentation.base._AugmentationBase'),
-    ('py:class', 'lightning.pytorch.utilities.types.OptimizerLRScheduler'),
+    ('py:class', 'lightning.pytorch.utilities.types.LRSchedulerConfig'),
+    ('py:class', 'lightning.pytorch.utilities.types.OptimizerConfig'),
+    ('py:class', 'lightning.pytorch.utilities.types.OptimizerLRSchedulerConfig'),
     ('py:class', 'segmentation_models_pytorch.base.model.SegmentationModel'),
     ('py:class', 'timm.models.resnet.ResNet'),
     ('py:class', 'timm.models.vision_transformer.VisionTransformer'),