fix comments

Azure · Jan 10, 2024 · 5ea7679 · 5ea7679
1 parent 6ffc5c1
commit 5ea7679
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/msamp/fsdp/flat_param.py b/msamp/fsdp/flat_param.py
@@ -7,7 +7,6 @@
 
 import torch
 import torch.nn as nn
-import torch.distributed as dist
 from torch.distributed.fsdp.flat_param import FlatParamHandle
 
 

diff --git a/tests/fsdp/test_fsdp_replacer.py b/tests/fsdp/test_fsdp_replacer.py
@@ -9,8 +9,8 @@
 
 import torch
 import torch.nn as nn
-from tests.helper import decorator
 
+from tests.helper import decorator
 from msamp.fsdp import FsdpReplacer