Activate test for TF and dca.vectorize_method support

PiperOrigin-RevId: 556714332
google-research · Aug 31, 2023 · b494832 · b494832
1 parent a64c854
commit b494832
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/visu3d/dc_arrays/camera_test.py b/visu3d/dc_arrays/camera_test.py
@@ -232,7 +232,7 @@ def test_camera_render(
   )
   points = points.as_xnp(xnp)
   points = points.broadcast_to((1,) * len(shape) + point_shape)
-  if shape and xnp is enp.lazy.jnp:
+  if shape and xnp in (enp.lazy.jnp, enp.lazy.tnp):
     pytest.skip('vmap not (yet) supported for cam.render')
   img = cam.render(points)
   assert img.shape == shape + (H, W, 3)