Merge 0ad2aaf into f353441

DeepRank · Jan 21, 2021 · 3049c97 · 3049c97
2 parents f353441 + 0ad2aaf
commit 3049c97
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_generate_cuda.py b/test/test_generate_cuda.py
@@ -24,15 +24,15 @@ class TestGenerateCUDA(unittest.TestCase):
     pdb_native = ['./1AK4/native/']
 
     @unittest.skipIf(skip, "torch fails on Travis")
-    @staticmethod
-    def test_generate_cuda():
+    def test_generate_cuda(self):
 
         # init the data assembler
         database = DataGenerator(
             chain1='C',
             chain2='D',
             pdb_source=self.pdb_source,
             pdb_native=self.pdb_native,
+            pssm_source='./1AK4/pssm_new/',
             compute_targets=['deeprank.targets.dockQ'],
             compute_features=[
                 'deeprank.features.AtomicFeature',