From 7324864366f9573e7a8b5d6b0ea98f247ef800b5 Mon Sep 17 00:00:00 2001
From: Baruch Tabanpour <btaba@google.com>
Date: Sun, 15 Mar 2026 11:23:03 -0700
Subject: [PATCH] Add random seed to SensorTest to prevent flakiness.

PiperOrigin-RevId: 884055306
Change-Id: Ia053d2661ed30835e66a145659c1e1b7275ab251
---
 mjx/mujoco/mjx/_src/sensor_test.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/mjx/mujoco/mjx/_src/sensor_test.py b/mjx/mujoco/mjx/_src/sensor_test.py
index 2843473c..8eabac97 100644
--- a/mjx/mujoco/mjx/_src/sensor_test.py
+++ b/mjx/mujoco/mjx/_src/sensor_test.py
@@ -42,6 +42,10 @@ def _assert_attr_eq(a, b, attr):
 
 class SensorTest(parameterized.TestCase):
 
+  def setUp(self):
+    super().setUp()
+    np.random.seed(314)
+
   @parameterized.product(
       filename=['sensor/model.xml', 'sensor/sensor.xml'],
       cone_type=list(ConeType),