Update DataSet.py

fix bug compute get_grid_shape only if grid_shape is not provided
DeepRank · Jun 29, 2021 · 14b03c6 · 14b03c6
1 parent 295b440
commit 14b03c6
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()