Update go.py

a-r-j · Apr 19, 2024 · 08148a2 · 08148a2
1 parent 61294d4
commit 08148a2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/proteinworkshop/datasets/go.py b/proteinworkshop/datasets/go.py
@@ -87,7 +87,7 @@ def __init__(
 
         self.train_fname = self.data_dir / "nrPDB-GO_train.txt"
         self.val_fname = self.data_dir / "nrPDB-GO_valid.txt"
-        self.test_fname = self.data_dir / "nrPDB-GO_test.csv"
+        self.test_fname = self.data_dir / "nrPDB-GO_test.txt"
         self.label_fname = self.data_dir / "nrPDB-GO_annot.tsv"
         self.url = "https://zenodo.org/record/6622158/files/GeneOntology.zip"