From 1ab27a90592d05ee690a51c915910774f9e810be Mon Sep 17 00:00:00 2001 From: Damian Stewart Date: Sat, 17 Jun 2023 10:38:34 +0200 Subject: [PATCH] better check for null manual_data_root --- data/every_dream_validation.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/data/every_dream_validation.py b/data/every_dream_validation.py index 97a99a4..e3fa770 100644 --- a/data/every_dream_validation.py +++ b/data/every_dream_validation.py @@ -97,7 +97,8 @@ class EveryDreamValidator: self.config.update({'manual_data_root': self.config['val_data_root']}) if self.config.get('val_split_mode') == 'manual': - if 'manual_data_root' in self.config: + manual_data_root = self.config.get('manual_data_root') + if manual_data_root is not None: self.config['extra_manual_datasets'].update({'val': self.config['manual_data_root']}) else: if len(self.config['extra_manual_datasets']) == 0: