"tools/cfgs/git@developer.sourcefind.cn:OpenDAS/openpcdet.git" did not exist on "557d463793f6ff63f0301e851a047d29924fb1fe"
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):
assert torch.all(torch.eq(protein['template_aatype'], template_seq_ours))
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)
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