meow meow

Author: deependujha

tests/tests_fabric/test_connector.py

@@ -890,7 +890,7 @@ def test_precision_selection_model_parallel(_, precision, raises):
 
 
 def test_bitsandbytes_precision_cuda_required(monkeypatch):
-    monkeypatch.setattr("pytorch_lightning_enterprise.plugins.precision.bitsandbytes", "_BITSANDBYTES_AVAILABLE", True)
+    monkeypatch.setattr("pytorch_lightning_enterprise.plugins.precision.bitsandbytes._BITSANDBYTES_AVAILABLE", True)
     monkeypatch.setitem(sys.modules, "bitsandbytes", Mock())
     with pytest.raises(RuntimeError, match="Bitsandbytes is only supported on CUDA GPUs"):
         _Connector(accelerator="cpu", plugins=BitsandbytesPrecision(mode="int8"))

tests/tests_pytorch/trainer/connectors/test_accelerator_connector.py

@@ -968,7 +968,7 @@ def test_precision_selection(precision_str, strategy_str, expected_precision_cls
 
 def test_bitsandbytes_precision_cuda_required(monkeypatch):
     monkeypatch.setattr(lightning.fabric.plugins.precision.bitsandbytes, "_BITSANDBYTES_AVAILABLE", True)
-    monkeypatch.setattr("pytorch_lightning_enterprise.plugins.precision.bitsandbytes", "_BITSANDBYTES_AVAILABLE", True)
+    monkeypatch.setattr("pytorch_lightning_enterprise.plugins.precision.bitsandbytes._BITSANDBYTES_AVAILABLE", True)
     monkeypatch.setitem(sys.modules, "bitsandbytes", Mock())
     with pytest.raises(RuntimeError, match="Bitsandbytes is only supported on CUDA GPUs"):
         _AcceleratorConnector(accelerator="cpu", plugins=BitsandbytesPrecision(mode="int8"))