use self to invoke static method in the class

DeepRank · Nov 25, 2019 · aebf829 · aebf829
1 parent 344d730
commit aebf829
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deeprank/generate/DataGenerator.py b/deeprank/generate/DataGenerator.py
@@ -413,7 +413,7 @@ def create_database(
                     self.f5.copy(mol_name + '/features/', molgrp)
 
                     # rotate the feature
-                    DataGenerator._rotate_feature(molgrp, axis, angle, mol_center)
+                    self._rotate_feature(molgrp, axis, angle, mol_center)
 
                     # grid center
                     molgrp.require_group('grid_points')
@@ -579,7 +579,7 @@ def add_feature(self, remove_error=True, prog_bar=True):
                     aug_molgrp.create_dataset("features/" + k, data=data)
 
                     # rotate
-                    DataGenerator._rotate_feature(
+                    self._rotate_feature(
                         aug_molgrp, axis, angle, center, feat_name=[k])
 
         # find errored augmented molecules