Check none context map glob

pyplati · Dec 6, 2023 · 83550f4 · 83550f4
1 parent fb74648
commit 83550f4
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/platipy/imaging/cnn/dataload.py b/platipy/imaging/cnn/dataload.py
@@ -349,9 +349,9 @@ def setup(self, stage=None):
             {
                 "id": case,
                 "image": self.data_dir.joinpath(self.image_glob.format(case=case)),
-                "context_map": self.data_dir.joinpath(
-                    self.context_map_glob.format(case=case)
-                ),
+                "context_map": None
+                if self.context_map_glob is None
+                else self.data_dir.joinpath(self.context_map_glob.format(case=case)),
                 "observers": {
                     observer: {
                         structure: self.data_dir.joinpath(
@@ -372,9 +372,9 @@ def setup(self, stage=None):
             {
                 "id": case,
                 "image": self.data_dir.joinpath(self.image_glob.format(case=case)),
-                "context_map": self.data_dir.joinpath(
-                    self.context_map_glob.format(case=case)
-                ),
+                "context_map": None
+                if self.context_map_glob is None
+                else self.data_dir.joinpath(self.context_map_glob.format(case=case)),
                 "observers": {
                     observer: {
                         structure: self.data_dir.joinpath(