pytorch: no need to use SANDCASTLE=1 from 2.3.0, it actually causes m…

…ore problems than it solves.
akesandgren · May 15, 2024 · 8bf6d80 · 8bf6d80
1 parent 0b5f61d
commit 8bf6d80
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/easybuild/easyblocks/p/pytorch.py b/easybuild/easyblocks/p/pytorch.py
@@ -426,8 +426,9 @@ def _set_cache_dir(self):
     def test_step(self):
         """Run unit tests"""
         self._set_cache_dir()
-        # Pretend to be on FB CI which disables some tests, especially those which download stuff
-        env.setvar('SANDCASTLE', '1')
+        if LooseVersion(self.version) < LooseVersion('2.3.0'):
+            # Pretend to be on FB CI which disables some tests, especially those which download stuff
+            env.setvar('SANDCASTLE', '1')
         # Skip this test(s) which is very flaky
         env.setvar('SKIP_TEST_BOTTLENECK', '1')
         # Parse excluded_tests and flatten into space separated string