HPU test update

Author: matthewdouglas

tests/test_modules.py

@@ -298,9 +298,8 @@ def test_kbit_backprop(device, module):
 
     kbit = nn.Sequential(*[torch.nn.Linear(dim1, dim2), module(dim2, 128)])
 
-    if device == "hpu":
-        if isinstance(kbit, bnb.nn.LinearFP4):
-            pytest.skip("FP4 is not supported on HPU")
+    if device == "hpu" and isinstance(kbit[1], bnb.nn.LinearFP4):
+        pytest.skip("FP4 is not supported on HPU")
 
     kbit[0].weight.detach().copy_(ref[0].weight)
     kbit[1].weight.detach().copy_(ref[1].weight)