[FIX] truth value error

wbenbihi · Dec 6, 2020 · 1836d40 · 1836d40
1 parent c80ca86
commit 1836d40
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/dataset/dataset.py b/src/dataset/dataset.py
@@ -54,15 +54,15 @@ def __init__(self, config: HourglassConfig):
         self.train_dataset: tf.data.Dataset = self._create_dataset(self.summary_train)
         self.test_dataset: tf.data.Dataset = self._create_dataset(self.summary_test)
         self.val_dataset: Optional[tf.data.Dataset] = (
-            self._create_dataset(self.summary_val) if self.summary_val else None
+            self._create_dataset(self.summary_val) if self.summary_val is not None else None
         )
         self._has_datasets = True
 
         # Data Augmentation
         self.train_dataset = self._augment_dataset(self.train_dataset)
         self.test_dataset = self._augment_dataset(self.test_dataset)
         self.val_dataset = (
-            self._augment_dataset(self.val_dataset) if self.val_dataset else None
+            self._augment_dataset(self.val_dataset) if self.val_dataset is not None else None
         )
         self._has_augments = True