Remove neuflow from test_models

hmorimitsu · Mar 30, 2024 · dadfc7b · dadfc7b
1 parent 7e653c7
commit dadfc7b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/ptlflow/models/test_models.py b/tests/ptlflow/models/test_models.py
@@ -35,6 +35,7 @@
 EXCLUDE_MODELS = [
     "matchflow",
     "matchflow_raft",
+    "neuflow",  # requires torch 2.X
     "scv4",
     "scv8",
     "separableflow",
@@ -45,6 +46,7 @@
     "lcv_raft_small",
     "matchflow",
     "matchflow_raft",
+    "neuflow",  # requires torch 2.X
     "scv4",
     "scv8",
     "separableflow",