Update typing for CometLogger.experiment (#11836)

2022-02-14 01:16:11 -08:00 · 2022-02-14 01:16:11 -08:00 · 914f685ed8
parent 79c4e5de60
commit 914f685ed8
1 changed files with 1 additions and 1 deletions
--- a/pytorch_lightning/loggers/comet.py
+++ b/pytorch_lightning/loggers/comet.py
@ -188,7 +188,7 @@ class CometLogger(LightningLoggerBase):

    @property
    @rank_zero_experiment
-    def experiment(self):
+    def experiment(self) -> Union[CometExperiment, CometExistingExperiment, CometOfflineExperiment]:
        r"""
        Actual Comet object. To use Comet features in your
        :class:`~pytorch_lightning.core.lightning.LightningModule` do the following.