fix: outputs

based-robotics · Sep 17, 2024 · 3cb18e5 · 3cb18e5
1 parent 88876e0
commit 3cb18e5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jaxadi/_ops.py b/jaxadi/_ops.py
@@ -95,5 +95,5 @@
     OP_ATAN2: "jnp.arctan2({0}, {1})",
     OP_CONST: "{0:.16f}",
     OP_INPUT: "inputs[{0}][{1}, {2}]",
-    OP_OUTPUT: "work[{0}][0]",
+    OP_OUTPUT: "{0}[0]",
 }