Apply suggestions from code review

pyg-team · Sep 26, 2024 · 9436ac0 · 9436ac0
1 parent d9c75b7
commit 9436ac0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/nn/models/test_compile.py b/test/nn/models/test_compile.py
@@ -14,7 +14,7 @@
 from torch_frame.testing import withPackage
 
 
-@withPackage("torch>=2.4.0")
+@withPackage("torch>=2.5.0")
 @pytest.mark.parametrize(
     "model_cls, model_kwargs, stypes, expected_graph_breaks",
     [
@@ -54,7 +54,7 @@
             Trompt,
             dict(channels=8, num_prompts=2),
             None,
-            7,
+            4,
             id="Trompt",
         ),
         pytest.param(