remove None from CPU_DEVICES

Lightning-AI · Jan 3, 2022 · 6ca1018 · 6ca1018
1 parent e677d76
commit 6ca1018
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/utilities/apply_func.py b/pytorch_lightning/utilities/apply_func.py
@@ -35,7 +35,7 @@
     Batch = type(None)
 
 
-CPU_DEVICES = (None, "cpu", torch.device("cpu"))
+CPU_DEVICES = ("cpu", torch.device("cpu"))
 
 
 def to_dtype_tensor(