actually fixed

pmcurtin · pmcurtin · commit 45fe57c38138 · 2025-02-07T16:07:26.000-05:00
diff --git a/src/torchrunx/ext/lightning.py b/src/torchrunx/ext/lightning.py
@@ -1,8 +1,8 @@
 """Pytorch Lightning extension utilities."""
 
 import torch
-from lightning.fabric.plugins.environments.torchelastic import (
-    TorchElasticEnvironment,  # pyright: ignore [reportMissingImports]
+from lightning.fabric.plugins.environments.torchelastic import (  # pyright: ignore [reportMissingImports]
+    TorchElasticEnvironment,
 )
 
 

Original file line number	Diff line number	Diff line change
`@@ -1,8 +1,8 @@`
`1`	`1`	`"""Pytorch Lightning extension utilities."""`
`2`	`2`
`3`	`3`	`import torch`
`4`		`-from lightning.fabric.plugins.environments.torchelastic import (`
`5`		`- TorchElasticEnvironment, # pyright: ignore [reportMissingImports]`
	`4`	`+from lightning.fabric.plugins.environments.torchelastic import ( # pyright: ignore [reportMissingImports]`
	`5`	`+ TorchElasticEnvironment,`
`6`	`6`	`)`
`7`	`7`
`8`	`8`