Merge 14b03c6 into 295b440

DeepRank · Jun 29, 2021 · 18978bf · 18978bf
2 parents 295b440 + 14b03c6
commit 18978bf
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/deeprank/learn/DataSet.py b/deeprank/learn/DataSet.py
@@ -283,9 +283,10 @@ def process_dataset(self):
 
         # get the pairing
         self.get_pairing_feature()
-
-        # get grid shape
-        self.get_grid_shape()
+
+        if self.grid_shape is None: 
+            # get grid shape
+            self.get_grid_shape()
 
         # get the input shape
         self.get_input_shape()