Skip to content

GitLab

"examples/vscode:/vscode.git/clone" did not exist on "dddebc0df2a6952a508cd1a127c7aff0bc60d934"
Commit dd944f64 authored by Christina Floristean's avatar Christina Floristean
Browse files

Remove AF2 requirement from test

parent f0a320e0
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 3 deletions
tests/test_deepspeed_evo_attention.py
View file @ dd944f64
......@@ -96,15 +96,12 @@ class TestDeepSpeedKernel(unittest.TestCase):
self.assertTrue(torch.allclose(torch.abs(out_repro_msa), torch.abs(out_repro_msa_ds), atol=consts.eps))
self.assertTrue(torch.allclose(torch.abs(out_repro_pair), torch.abs(out_repro_pair_ds), atol=consts.eps))
@compare_utils.skip_unless_alphafold_installed()
def test_compare_evoformer_bf16(self):
self.compare_evoformer(torch.bfloat16)
@compare_utils.skip_unless_alphafold_installed()
def test_compare_evoformer_fp32(self):
self.compare_evoformer(torch.float32)
@compare_utils.skip_unless_alphafold_installed()
def test_dry_run(self):
with open("tests/test_data/sample_feats.pickle", "rb") as fp:
batch = pickle.load(fp)
......
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