handle even a missing targets group

DeepRank · Dec 18, 2023 · 7b33837 · 7b33837
1 parent edfed88
commit 7b33837
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/deeprank/learn/DataSet.py b/deeprank/learn/DataSet.py
@@ -936,9 +936,8 @@ def load_one_variant(self, fname, variant_name=None):
         # get the target value
         if self.select_target is not None:
 
-            target_group = variant_data['targets']
-            if self.select_target in target_group:
-                target = target_group[self.select_target][()]
+            if 'targets' in variant_data and self.select_target in variant_data['targets']:
+                target = variant_data['targets'][self.select_target][()]
                 logger.debug("{} has target {}".format(variant_name, target))
 
                 target = np.array([target]).astype(outtype)