From 914f685ed826d9adf931b60348cb587e7bec35e5 Mon Sep 17 00:00:00 2001
From: Akash Kwatra <akashkw@gmail.com>
Date: Mon, 14 Feb 2022 01:16:11 -0800
Subject: [PATCH] Update typing for CometLogger.experiment (#11836)

---
 pytorch_lightning/loggers/comet.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pytorch_lightning/loggers/comet.py b/pytorch_lightning/loggers/comet.py
index a98b2c7fa5..a4fca47923 100644
--- 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.