Fix RunningStage properties for sanity checking (#16774)

Lightning-AI · Feb 16, 2023 · 57f2f1c · 57f2f1c
1 parent f6cad32
commit 57f2f1c
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/lightning/pytorch/trainer/states.py b/src/lightning/pytorch/trainer/states.py
@@ -63,13 +63,11 @@ class RunningStage(LightningEnum):
 
     @property
     def evaluating(self) -> bool:
-        return self in (self.VALIDATING, self.TESTING)
+        return self in (self.VALIDATING, self.TESTING, self.SANITY_CHECKING)
 
     @property
     def dataloader_prefix(self) -> Optional[str]:
-        if self == self.SANITY_CHECKING:
-            return None
-        if self == self.VALIDATING:
+        if self in (self.VALIDATING, self.SANITY_CHECKING):
             return "val"
         return self.value