Skip broken tests at HEAD

PiperOrigin-RevId: 723321880
ksebaz · Feb 5, 2025 · c07b6b5 · c07b6b5
1 parent 9aa9813
commit c07b6b5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/random_lax_test.py b/tests/random_lax_test.py
@@ -1271,6 +1271,8 @@ def testMultinomialExample(self):
       for outcomes in [2, 3, 4]
   ])
   def testMultinomialShape(self, shape, outcomes):
+    if outcomes == 2:
+      self.skipTest('Broken at HEAD b/394428711')
     key = random.key(0)
 
     key, subkey = random.split(key)