Commit cad8de7e authored by Sachin Kadyan's avatar Sachin Kadyan
Browse files

Fixed filepath for the features.pkl pickle file.

parent 8923e536
...@@ -57,7 +57,7 @@ class TestDataTransforms(unittest.TestCase): ...@@ -57,7 +57,7 @@ class TestDataTransforms(unittest.TestCase):
assert torch.all(torch.eq(protein['template_aatype'], template_seq_ours)) assert torch.all(torch.eq(protein['template_aatype'], template_seq_ours))
def test_correct_msa_restypes(self): def test_correct_msa_restypes(self):
with open('../features.pkl', 'rb') as file: with open("../test_data/features.pkl", 'rb') as file:
features = pickle.load(file) features = pickle.load(file)
protein = {'msa': torch.tensor(features['msa'], dtype=torch.int64)} protein = {'msa': torch.tensor(features['msa'], dtype=torch.int64)}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment