From 5ea7679a59fed8d03e38b70b19490c02fe8ff2e9 Mon Sep 17 00:00:00 2001
From: tocean <yuxiangyang@microsoft.com>
Date: Wed, 10 Jan 2024 07:02:24 +0000
Subject: [PATCH] fix comments

---
 msamp/fsdp/flat_param.py         | 1 -
 tests/fsdp/test_fsdp_replacer.py | 2 +-
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/msamp/fsdp/flat_param.py b/msamp/fsdp/flat_param.py
index 56de6a3f..35807f7f 100644
--- 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
index 0e4405fb..d53b0514 100644
--- 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