From bae51d7f06799730d9d4472566e2d9ee07ea9b49 Mon Sep 17 00:00:00 2001
From: Henry <henry.webel@cpr.ku.dk>
Date: Sat, 24 Feb 2024 09:45:21 +0100
Subject: [PATCH] :bug: make CF Sklearn model run without MPS

- set fastai defaults to cpu
---
 vaep/sklearn/cf_transformer.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/vaep/sklearn/cf_transformer.py b/vaep/sklearn/cf_transformer.py
index 42f7bcc8c..cb1ce7b02 100644
--- a/vaep/sklearn/cf_transformer.py
+++ b/vaep/sklearn/cf_transformer.py
@@ -81,7 +81,8 @@ def fit(self, X: pd.Series, y: pd.Series = None,
             y_range=(int(X.squeeze().min()),
                      int(X.squeeze().max()) + 1)
         )
-
+        if not cuda:
+            default_device(use=False)  # set to cpu
         if y is not None:
             X, frac = collab.combine_data(X, y)
         else: