From c7155ba68213463d0d4db7835e524c018234467e Mon Sep 17 00:00:00 2001 From: Sukhil Patel Date: Tue, 26 Mar 2024 14:24:28 +0000 Subject: [PATCH 1/2] Force using CPU to get test working locally with macs --- tests/test_end2end.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_end2end.py b/tests/test_end2end.py index c2c18482..523ec695 100644 --- 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) From 1e8ab1d639f9a2063964c66858badb51ae0d2c6a Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 26 Mar 2024 14:35:31 +0000 Subject: [PATCH 2/2] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- tests/test_end2end.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_end2end.py b/tests/test_end2end.py index 523ec695..7a7161d5 100644 --- 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, accelerator='cpu') + trainer = lightning.pytorch.trainer.trainer.Trainer(fast_dev_run=True, accelerator="cpu") trainer.fit(model=multimodal_model, datamodule=sample_train_val_datamodule)