init target_formatter not only typed (#1665)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Lightning-Universe · Aug 9, 2023 · e76a61a · e76a61a
1 parent cad3cac
commit e76a61a
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 1 deletion.
diff --git a/src/flash/audio/classification/input.py b/src/flash/audio/classification/input.py
@@ -138,6 +138,7 @@ def load_data(
         # If we had binary multi-class targets then we also know the labels (column names)
         if (
             self.training
+            and hasattr(self, "target_formatter")
             and isinstance(self.target_formatter, MultiBinaryTargetFormatter)
             and isinstance(target_keys, List)
         ):

diff --git a/src/flash/image/classification/input.py b/src/flash/image/classification/input.py
@@ -157,6 +157,7 @@ def load_data(
         # If we had binary multi-class targets then we also know the labels (column names)
         if (
             self.training
+            and hasattr(self, "target_formatter")
             and isinstance(self.target_formatter, MultiBinaryTargetFormatter)
             and isinstance(target_keys, List)
         ):

diff --git a/src/flash/text/classification/input.py b/src/flash/text/classification/input.py
@@ -53,7 +53,11 @@ def load_data(
             self.load_target_metadata(targets, target_formatter=target_formatter)
 
             # If we had binary multi-class targets then we also know the labels (column names)
-            if isinstance(self.target_formatter, MultiBinaryTargetFormatter) and isinstance(target_keys, List):
+            if (
+                hasattr(self, "target_formatter")
+                and isinstance(self.target_formatter, MultiBinaryTargetFormatter)
+                and isinstance(target_keys, List)
+            ):
                 self.labels = target_keys
 
         # remove extra columns

diff --git a/src/flash/video/classification/input.py b/src/flash/video/classification/input.py
@@ -215,6 +215,7 @@ def load_data(
         # If we had binary multi-class targets then we also know the labels (column names)
         if (
             self.training
+            and hasattr(self, "target_formatter")
             and isinstance(self.target_formatter, MultiBinaryTargetFormatter)
             and isinstance(target_keys, List)
         ):
@@ -243,6 +244,7 @@ def load_data(
         # If we had binary multi-class targets then we also know the labels (column names)
         if (
             self.training
+            and hasattr(self, "target_formatter")
             and isinstance(self.target_formatter, MultiBinaryTargetFormatter)
             and isinstance(targets, List)
         ):