fix test 2

Author: R-Palazzo

deepecho/models/basic_gan.py

@@ -186,7 +186,7 @@ def __init__(
 
         if enable_gpu:
             if sys.platform == 'darwin':  # macOS
-                if torch.backends.mps.is_available():
+                if getattr(torch.backends, 'mps', None) and torch.backends.mps.is_available():
                     device = torch.device('mps')
                 else:
                     device = torch.device('cpu')

tests/integration/test_basic_gan.py

@@ -33,7 +33,11 @@ def test__init___enable_gpu(self):
         model = BasicGANModel(epochs=10, enable_gpu=True)
 
         # Assert
-        if sys.platform == 'darwin' and torch.backends.mps.is_available():
+        if (
+            sys.platform == 'darwin'
+            and getattr(torch.backends, 'mps', None)
+            and torch.backends.mps.is_available()
+        ):
             expected_device = torch.device('mps')
         elif torch.cuda.is_available():
             expected_device = torch.device('cuda')