Trainers: fix support for non-TensorBoardLogger (#1145)

* Trainers: fix support for CSVLogger

* No need to check truthiness of attr

* Revert "No need to check truthiness of attr"

This reverts commit 450df6c53c.

torchgeo/trainers/classification.py

@@ -192,6 +192,7 @@ class ClassificationTask(pl.LightningModule):
             and hasattr(self.trainer, "datamodule")
             and self.logger
             and hasattr(self.logger, "experiment")
+            and hasattr(self.logger.experiment, "add_figure")
         ):
             try:
                 datamodule = self.trainer.datamodule
@@ -376,6 +377,7 @@ class MultiLabelClassificationTask(ClassificationTask):
             and hasattr(self.trainer, "datamodule")
             and self.logger
             and hasattr(self.logger, "experiment")
+            and hasattr(self.logger.experiment, "add_figure")
         ):
             try:
                 datamodule = self.trainer.datamodule

torchgeo/trainers/detection.py

@@ -230,6 +230,7 @@ class ObjectDetectionTask(pl.LightningModule):
             and hasattr(self.trainer, "datamodule")
             and self.logger
             and hasattr(self.logger, "experiment")
+            and hasattr(self.logger.experiment, "add_figure")
         ):
             try:
                 datamodule = self.trainer.datamodule

torchgeo/trainers/regression.py

@@ -149,6 +149,7 @@ class RegressionTask(pl.LightningModule):
             and hasattr(self.trainer, "datamodule")
             and self.logger
             and hasattr(self.logger, "experiment")
+            and hasattr(self.logger.experiment, "add_figure")
         ):
             try:
                 datamodule = self.trainer.datamodule

torchgeo/trainers/segmentation.py

@@ -202,6 +202,7 @@ class SemanticSegmentationTask(pl.LightningModule):
             and hasattr(self.trainer, "datamodule")
             and self.logger
             and hasattr(self.logger, "experiment")
+            and hasattr(self.logger.experiment, "add_figure")
         ):
             try:
                 datamodule = self.trainer.datamodule