warp-lang 1.7.2rc1__py3-none-macosx_10_13_universal2.whl → 1.8.1__py3-none-macosx_10_13_universal2.whl

warp/tests/cuda/test_streams.py

@@ -492,7 +492,7 @@ class TestStreams(unittest.TestCase):
 
         # Can't get the stream on a CPU device
         with self.assertRaises(RuntimeError):
-            cpu_stream = cpu_device.stream  # noqa: F841
+            cpu_stream = cpu_device.stream
 
     @unittest.skipUnless(len(wp.get_cuda_devices()) > 1, "Requires at least two CUDA devices")
     @unittest.skipUnless(check_p2p(), "Peer-to-Peer transfers not supported")

warp/tests/geometry/test_volume.py

@@ -893,7 +893,7 @@ def test_volume_from_numpy_3d(test, device):
     nums = np.ceil((maxs - mins) / (voxel_size)).astype(dtype=int)
     centers = np.array([[-1.0, -1.0, -1.0], [0.0, 0.0, 0.0], [1.0, 1.0, 1.0]])
     rad = 2.5
-    sphere_sdf_np = np.zeros(tuple(nums) + (3,))
+    sphere_sdf_np = np.zeros((*tuple(nums), 3))
     for x in range(nums[0]):
         for y in range(nums[1]):
             for z in range(nums[2]):
@@ -935,7 +935,7 @@ def test_volume_aniso_transform(test, device):
 def test_volume_write(test, device):
     codecs = ["none", "zip", "blosc"]
     try:
-        import blosc  # noqa: F401 I001
+        import blosc  # noqa: F401
     except ImportError:
         codecs.pop()
 

warp/tests/interop/test_dlpack.py

@@ -412,8 +412,8 @@ def test_dlpack_warp_to_jax(test, device):
 
     test.assertEqual(a.ptr, j1.unsafe_buffer_pointer())
     test.assertEqual(a.ptr, j2.unsafe_buffer_pointer())
-    test.assertEqual(a.device, wp.device_from_jax(list(j1.devices())[0]))
-    test.assertEqual(a.device, wp.device_from_jax(list(j2.devices())[0]))
+    test.assertEqual(a.device, wp.device_from_jax(next(iter(j1.devices()))))
+    test.assertEqual(a.device, wp.device_from_jax(next(iter(j2.devices()))))
     test.assertEqual(a.shape, j1.shape)
     test.assertEqual(a.shape, j2.shape)
 
@@ -460,8 +460,8 @@ def test_dlpack_warp_to_jax_v2(test, device):
 
     test.assertEqual(a.ptr, j1.unsafe_buffer_pointer())
     test.assertEqual(a.ptr, j2.unsafe_buffer_pointer())
-    test.assertEqual(a.device, wp.device_from_jax(list(j1.devices())[0]))
-    test.assertEqual(a.device, wp.device_from_jax(list(j2.devices())[0]))
+    test.assertEqual(a.device, wp.device_from_jax(next(iter(j1.devices()))))
+    test.assertEqual(a.device, wp.device_from_jax(next(iter(j2.devices()))))
     test.assertEqual(a.shape, j1.shape)
     test.assertEqual(a.shape, j2.shape)
 
@@ -543,15 +543,15 @@ def test_dlpack_jax_to_warp(test, device):
         j = jax.numpy.arange(N, dtype=jax.numpy.float32)
 
         # use generic dlpack conversion
-        a1 = wp.from_dlpack(jax.dlpack.to_dlpack(j))
+        a1 = wp.from_dlpack(j)
 
         # use jax wrapper
         a2 = wp.from_jax(j)
 
         test.assertEqual(a1.ptr, j.unsafe_buffer_pointer())
         test.assertEqual(a2.ptr, j.unsafe_buffer_pointer())
-        test.assertEqual(a1.device, wp.device_from_jax(list(j.devices())[0]))
-        test.assertEqual(a2.device, wp.device_from_jax(list(j.devices())[0]))
+        test.assertEqual(a1.device, wp.device_from_jax(next(iter(j.devices()))))
+        test.assertEqual(a2.device, wp.device_from_jax(next(iter(j.devices()))))
         test.assertEqual(a1.shape, j.shape)
         test.assertEqual(a2.shape, j.shape)
 
@@ -586,8 +586,8 @@ def test_dlpack_jax_to_warp_v2(test, device):
 
         test.assertEqual(a1.ptr, j.unsafe_buffer_pointer())
         test.assertEqual(a2.ptr, j.unsafe_buffer_pointer())
-        test.assertEqual(a1.device, wp.device_from_jax(list(j.devices())[0]))
-        test.assertEqual(a2.device, wp.device_from_jax(list(j.devices())[0]))
+        test.assertEqual(a1.device, wp.device_from_jax(next(iter(j.devices()))))
+        test.assertEqual(a2.device, wp.device_from_jax(next(iter(j.devices()))))
         test.assertEqual(a1.shape, j.shape)
         test.assertEqual(a2.shape, j.shape)
 

warp/tests/interop/test_jax.py

@@ -319,7 +319,6 @@ try:
     os.environ["XLA_PYTHON_CLIENT_ALLOCATOR"] = "platform"
 
     import jax
-    import jax.dlpack
 
     # NOTE: we must enable 64-bit types in Jax to test the full gamut of types
     jax.config.update("jax_enable_x64", True)

warp/tests/run_coverage_serial.py

@@ -28,7 +28,7 @@ cover = coverage.Coverage(config_file=True, messages=True)
 cover.start()
 
 with cover.collect():
-    import unittest_serial  # noqa: E402
+    import unittest_serial
 
     unittest_serial.run_specified()
 

warp/tests/sim/disabled_kinematics.py

@@ -123,7 +123,7 @@ def build_complex_joint_mechanism(chain_length):
             parent_xform=tf1,
             child_xform=tf2,
         )
-        builder.joint_q[-7:] = [0.5, -0.9, 1.4] + list(wp.quat_rpy(0.3, -0.5, 0.7))
+        builder.joint_q[-7:] = [0.5, -0.9, 1.4, *wp.quat_rpy(0.3, -0.5, 0.7)]
         builder.joint_qd[-6:] = [1.0, -1.0, 0.5, 0.8, -0.3, 0.1]
 
         b8 = builder.add_body(com=com2)
@@ -133,7 +133,7 @@ def build_complex_joint_mechanism(chain_length):
             parent_xform=tf1,
             child_xform=tf2,
         )
-        builder.joint_q[-7:] = [-0.3, -0.7, 0.2] + list(wp.quat_rpy(0.1, 0.1, 0.4))
+        builder.joint_q[-7:] = [-0.3, -0.7, 0.2, *wp.quat_rpy(0.1, 0.1, 0.4)]
         builder.joint_qd[-6:] = [-0.34, 0.5, -0.6, -0.4, 0.2, 0.1]
 
         # D6 joint that behaves like a fixed joint