From 0ad2aaf3d332982611b23ade34399ea86f9aeb78 Mon Sep 17 00:00:00 2001
From: CunliangGeng <c.geng@esciencecenter.nl>
Date: Thu, 21 Jan 2021 18:42:20 +0100
Subject: [PATCH] Fix bugs in test_generate_cuda.py

---
 test/test_generate_cuda.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/test/test_generate_cuda.py b/test/test_generate_cuda.py
index 6b505069..4c834c9d 100644
--- a/test/test_generate_cuda.py
+++ b/test/test_generate_cuda.py
@@ -24,8 +24,7 @@ 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(
@@ -33,6 +32,7 @@ def test_generate_cuda():
             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',