From fb5792b42000e37fa546a62bdd2c9c8ba6a448bc Mon Sep 17 00:00:00 2001
From: Baruch Tabanpour <btaba@google.com>
Date: Wed, 6 Aug 2025 16:49:09 -0700
Subject: [PATCH] Add back warp as an optional dep.

PiperOrigin-RevId: 791896458
Change-Id: I614d4d766d2c2d1ad79d5c3db9bbcbf5778f2183
---
 mjx/pyproject.toml | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/mjx/pyproject.toml b/mjx/pyproject.toml
index 368a9c06..a264dca3 100644
--- a/mjx/pyproject.toml
+++ b/mjx/pyproject.toml
@@ -32,6 +32,10 @@ dependencies = [
     "mujoco>=3.3.5.dev0",
     "scipy",
     "trimesh",
+]
+
+[project.optional-dependencies]
+warp = [
     "warp-lang==1.8.1",
 ]