Update ipu.py

ananthsub · ananthsub · commit 561aabdeea10 · 2022-02-07T22:33:58.000-08:00
diff --git a/pytorch_lightning/accelerators/ipu.py b/pytorch_lightning/accelerators/ipu.py
@@ -23,7 +23,7 @@
 class IPUAccelerator(Accelerator):
     """Accelerator for IPUs."""
 
-    def setup_environment(self, root_device: torch.device) -> None:
+    def __init__(self) -> None:
         """
         Raises:
             MisconfigurationException:
