From 08148a2f8ee39c6d96b127be10ca4cce6a66d738 Mon Sep 17 00:00:00 2001 From: Alex Morehead Date: Fri, 19 Apr 2024 18:42:50 -0500 Subject: [PATCH] Update go.py --- proteinworkshop/datasets/go.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/proteinworkshop/datasets/go.py b/proteinworkshop/datasets/go.py index c4fb0623..95cb4c0f 100644 --- 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"