test_bitsandbytes_plugin

Author: Borda

tests/tests_fabric/plugins/precision/test_bitsandbytes.py

@@ -11,6 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License
+import platform
 import sys
 from unittest.mock import Mock
 
@@ -28,7 +29,7 @@
 
 
 @pytest.mark.skipif(_BITSANDBYTES_AVAILABLE, reason="bitsandbytes needs to be unavailable")
-@RunIf(mps=False)  # skip on MPS as Bitsandbytes is only supported on CUDA GPUs
+@pytest.mark.skipif(platform.system() == "Darwin")  # skip on Mac as Bitsandbytes is only supported on CUDA GPUs
 def test_bitsandbytes_plugin(monkeypatch):
     module = lightning.fabric.plugins.precision.bitsandbytes
     monkeypatch.setattr(module, "_BITSANDBYTES_AVAILABLE", lambda: True)

tests/tests_pytorch/plugins/precision/test_bitsandbytes.py

@@ -11,6 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License
+import platform
 import sys
 from unittest.mock import Mock
 
@@ -22,11 +23,10 @@
 from lightning.fabric.plugins.precision.bitsandbytes import _BITSANDBYTES_AVAILABLE
 from lightning.pytorch import LightningModule, Trainer
 from lightning.pytorch.plugins.precision.bitsandbytes import BitsandbytesPrecision
-from tests_pytorch.helpers.runif import RunIf
 
 
 @pytest.mark.skipif(_BITSANDBYTES_AVAILABLE, reason="bitsandbytes needs to be unavailable")
-@RunIf(mps=False)  # skip on MPS as Bitsandbytes is only supported on CUDA GPUs
+@pytest.mark.skipif(platform.system() == "Darwin")  # skip on Mac as Bitsandbytes is only supported on CUDA GPUs
 def test_bitsandbytes_plugin(monkeypatch):
     module = lightning.fabric.plugins.precision.bitsandbytes
     monkeypatch.setattr(module, "_BITSANDBYTES_AVAILABLE", lambda: True)