Disable export for timm-eff-net

qubvel · qubvel · commit df2f48417b26 · 2025-01-13T22:46:19.000Z
diff --git a/tests/encoders/test_timm_ported_encoders.py b/tests/encoders/test_timm_ported_encoders.py
@@ -26,6 +26,9 @@ class TestTimmEfficientNetEncoder(base.BaseEncoderTester):
     )
     files_for_diff = ["encoders/timm_efficientnet.py"]
 
+    # works with torch 2.4.0, but not with torch 2.5.1
+    do_test_torch_export = False
+
 
 class TestTimmGERNetEncoder(base.BaseEncoderTester):
     encoder_names = (

Original file line number	Diff line number	Diff line change
`@@ -26,6 +26,9 @@ class TestTimmEfficientNetEncoder(base.BaseEncoderTester):`
`26`	`26`	`)`
`27`	`27`	`files_for_diff = ["encoders/timm_efficientnet.py"]`
`28`	`28`
	`29`	`+ # works with torch 2.4.0, but not with torch 2.5.1`
	`30`	`+ do_test_torch_export = False`
	`31`	`+`
`29`	`32`
`30`	`33`	`class TestTimmGERNetEncoder(base.BaseEncoderTester):`
`31`	`34`	`encoder_names = (`