Made CPU the only device, GPU is not supported on AWS lambda

Author: m-messer

evaluation_function/models/basic_nn.py

@@ -72,8 +72,11 @@ def train_model(device):
     return model
 
 def run(response, answer, params: Params) -> Result:
-    print("GPU") if torch.backends.mps.is_available() else print("CPU")
-    device = torch.device("mps" if torch.backends.mps.is_available() else "cpu")
+    # print("GPU") if torch.backends.mps.is_available() else print("CPU")
+    # device = torch.device("mps" if torch.backends.mps.is_available() else "cpu")
+
+    device = "cpu"
+
     refresh = params.get("refresh", False)
     if refresh:
         model = train_model(device)