reenable more tests

Lightning-AI · Jan 10, 2025 · da14771 · da14771
1 parent 342b011
commit da14771
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 12 deletions.
diff --git a/thunder/tests/test_networks.py b/thunder/tests/test_networks.py
@@ -367,10 +367,6 @@ def test_quantization():
         assert_close(v, sd2[k])
 
 
-@pytest.mark.skipif(
-    version_between(torch.__version__, min_ver="2.6.0dev0", max_ver="2.6.0a99"),
-    reason="https://github.com/Lightning-AI/lightning-thunder/issues/1471",
-)
 @thunder.tests.framework.requiresCUDA
 def test_thunderfx_mistral_nemo_small():
     """
@@ -422,10 +418,6 @@ def test_thunderfx_mistral_nemo_small():
 
 
 # disabled "Qwen/Qwen2.5-7B-Instruct" see https://github.com/NVIDIA/Fuser/issues/3682
-@pytest.mark.skipif(
-    version_between(torch.__version__, min_ver="2.6.0dev0", max_ver="2.6.0a99"),
-    reason="https://github.com/Lightning-AI/lightning-thunder/issues/1471",
-)
 @thunder.tests.framework.requiresCUDA
 @pytest.mark.parametrize("model_id", ["microsoft/Phi-3-mini-128k-instruct"])
 def test_hf_for_nemo(model_id):

diff --git a/thunder/tests/test_recipes.py b/thunder/tests/test_recipes.py
@@ -24,10 +24,6 @@ def test_recipe_basic_bert():
     assert_close(actual, expected)
 
 
-@pytest.mark.skipif(
-    version_between(torch.__version__, min_ver="2.6.0dev0", max_ver="2.6.0a99"),
-    reason="https://github.com/Lightning-AI/lightning-thunder/issues/1471",
-)
 def test_recipe_basic_bert_dynamo():
     bert = transformers.BertForSequenceClassification(transformers.BertConfig())
     del bert.bert.encoder.layer[1:]