From 79fb5c6470a6afc0ea9cf8a55374ac20a6731616 Mon Sep 17 00:00:00 2001
From: qazal <77887910+Qazalin@users.noreply.github.com>
Date: Thu, 6 Feb 2025 15:27:59 +0100
Subject: [PATCH] hotfix: test_shard_no_recompile shouldn't rely on schedule
 order [pr] (#8928)

---
 test/test_multitensor.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/test_multitensor.py b/test/test_multitensor.py
index 28b118b619..2fe5e2e71f 100644
--- a/test/test_multitensor.py
+++ b/test/test_multitensor.py
@@ -84,7 +84,7 @@ class TestMultiTensor(unittest.TestCase):
     for si, ei in zip(sched[:], lower_schedule(sched)):
       if isinstance(ei.prg, CompiledRunner): names.append(ei.prg.p.name)
       ei.run()
-    assert names[-2] == names[-1], "function was relinearized"
+    self.assertEqual(len(set(names)), 3), "function was relinearized"
 
   @unittest.skip("this doesn't fold because shard_ calls contiguous on all lbs")
   def test_sharded_memory(self):