Nit

Faisal-Alsrheed · Jan 26, 2024 · e149772 · e149772
1 parent 7f87209
commit e149772
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keras/backend/jax/core.py b/keras/backend/jax/core.py
@@ -265,7 +265,7 @@ def scatter(indices, values, shape):
 
 
 def scatter_update(inputs, indices, updates):
-    inputs = jnp.array(inputs)
+    inputs = convert_to_tensor(inputs)
     indices = jnp.array(indices)
     indices = jnp.transpose(indices)
     inputs = inputs.at[tuple(indices)].set(updates)