lint

chenhao388 · chenhao388 · commit f26a8f50fb3e · 2025-10-25T15:17:08.000+08:00
diff --git a/test/test_nn.py b/test/test_nn.py
@@ -2161,7 +2161,7 @@ def test_module_buffer():
         module.cuda()
         assert module.td.device.type == "cuda"
     elif is_npu_available():
-        module.npu()
+        module = module.to("npu:0")
         assert module.td.device.type == "npu"
 
 
