Skip to content

Commit b16537b

Browse files
committed
Merge branch 'jpb/batching' of github.com:pasteurlabs/tesseract-jax into jpb/batching
2 parents 47efbc6 + c8f3081 commit b16537b

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

tesseract_jax/primitive.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -359,7 +359,6 @@ def apply_tesseract(
359359
client = Jaxeract(tesseract_client)
360360

361361
flat_args, input_pytreedef = jax.tree.flatten(inputs)
362-
363362
is_static_mask = tuple(_is_static(arg) for arg in flat_args)
364363
array_args, static_args = split_args(flat_args, is_static_mask)
365364
static_args = tuple(_make_hashable(arg) for arg in static_args)

0 commit comments

Comments
 (0)