From 89152fb39e8f7a8eddfa7ec2622270b9dcf19521 Mon Sep 17 00:00:00 2001
From: Taylor Howell <taylorhowell@google.com>
Date: Thu, 11 Sep 2025 15:06:06 -0700
Subject: [PATCH] Import NVIDIA/warp from GitHub.

PiperOrigin-RevId: 806000657
Change-Id: I5127baf8d212c5fc7bd597c843957f4b279d52db
---
 mjx/mujoco/mjx/third_party/warp/jax_experimental/ffi.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/mjx/mujoco/mjx/third_party/warp/jax_experimental/ffi.py b/mjx/mujoco/mjx/third_party/warp/jax_experimental/ffi.py
index 7b10099f..05165b49 100644
--- a/mjx/mujoco/mjx/third_party/warp/jax_experimental/ffi.py
+++ b/mjx/mujoco/mjx/third_party/warp/jax_experimental/ffi.py
@@ -381,6 +381,7 @@ class FfiCallable:
             if arg_name == "return":
                 if arg_type is not None:
                     raise TypeError("Function must not return a value")
+                continue
             else:
                 arg = FfiArg(arg_name, arg_type, arg_name in in_out_argnames)
                 if arg_name in in_out_argnames: