create checkpoint with only 1 device

Author: ryan597

tests/tests_pytorch/core/test_saving.py

@@ -8,7 +8,7 @@
 
 def create_boring_checkpoint(tmpdir):
     model = BoringModel()
-    trainer = pl.Trainer(accelerator="auto", max_epochs=1, enable_model_summary=False, enable_progress_bar=False)
+    trainer = pl.Trainer(devices=1, max_epochs=1, enable_model_summary=False, enable_progress_bar=False)
     trainer.fit(model)
     trainer.save_checkpoint(f"{tmpdir}/boring.ckpt")