Merge pull request #169 from openclimatefix/update-e2e-test

Force using CPU to get e2e test working locally with macs/mps
openclimatefix · Mar 26, 2024 · 5f5d154 · 5f5d154
2 parents f1534d8 + 1e8ab1d
commit 5f5d154
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_end2end.py b/tests/test_end2end.py
@@ -2,5 +2,5 @@
 
 
 def test_model_trainer_fit(multimodal_model, sample_train_val_datamodule):
-    trainer = lightning.pytorch.trainer.trainer.Trainer(fast_dev_run=True)
+    trainer = lightning.pytorch.trainer.trainer.Trainer(fast_dev_run=True, accelerator="cpu")
     trainer.fit(model=multimodal_model, datamodule=sample_train_val_datamodule)