Merge pull request #26238 from hawkinsp:coretest

PiperOrigin-RevId: 721884935

Author: Google-ML-Automation

tests/core_test.py

@@ -295,6 +295,8 @@ def test_jvp_repeated_fwd(self):
 
   @jtu.thread_unsafe_test()  # gc isn't predictable when threaded
   def test_reference_cycles(self):
+    if jtu.TEST_NUM_THREADS.value > 1:
+      self.skipTest("Test does not work with multiple threads")
     gc.collect()
 
     def f(x):
@@ -313,6 +315,8 @@ def f(x):
 
   @jtu.thread_unsafe_test()  # gc isn't predictable when threaded
   def test_reference_cycles_jit(self):
+    if jtu.TEST_NUM_THREADS.value > 1:
+      self.skipTest("Test does not work with multiple threads")
     gc.collect()
 
     def f(x):