From 1fea658561621d8835a085afa4b369ac4c62a8ef Mon Sep 17 00:00:00 2001
From: zigzagcai <caizheng@pjlab.org.cn>
Date: Fri, 15 Dec 2023 18:52:54 +0800
Subject: [PATCH] fix pylint

---
 internlm/utils/storage_manager.py | 41 ++++++++++++++++++++-----------
 1 file changed, 26 insertions(+), 15 deletions(-)

diff --git a/internlm/utils/storage_manager.py b/internlm/utils/storage_manager.py
index 8afcbd2..8319322 100644
--- a/internlm/utils/storage_manager.py
+++ b/internlm/utils/storage_manager.py
@@ -9,22 +9,33 @@ dill.Pickler.dumps, dill.Pickler.loads = dill.dumps, dill.loads
 multiprocessing.reduction.ForkingPickler = dill.Pickler
 multiprocessing.reduction.dump = dill.dump
 
-import asyncio
-import concurrent.futures
-import hashlib
-import io
-import os
-import pickle
-import re
-import socket
-import stat
-from asyncio import InvalidStateError
-from asyncio.tasks import ALL_COMPLETED
-from datetime import datetime
-from typing import Any, Awaitable, Callable, Dict, List, Union
+import asyncio  # noqa: E402  #pylint: disable=wrong-import-position
+import concurrent.futures  # noqa: E402  #pylint: disable=wrong-import-position
+import hashlib  # noqa: E402  #pylint: disable=wrong-import-position
+import io  # noqa: E402  #pylint: disable=wrong-import-position
+import os  # noqa: E402  #pylint: disable=wrong-import-position
+import pickle  # noqa: E402  #pylint: disable=wrong-import-position
+import re  # noqa: E402  #pylint: disable=wrong-import-position
+import socket  # noqa: E402  #pylint: disable=wrong-import-position
+import stat  # noqa: E402  #pylint: disable=wrong-import-position
+from asyncio import (  # noqa: E402  #pylint: disable=wrong-import-position
+    InvalidStateError,
+)
+from asyncio.tasks import (  # noqa: E402  #pylint: disable=wrong-import-position
+    ALL_COMPLETED,
+)
+from datetime import datetime  # noqa: E402  #pylint: disable=wrong-import-position
+from typing import (  # noqa: E402  #pylint: disable=wrong-import-position
+    Any,
+    Awaitable,
+    Callable,
+    Dict,
+    List,
+    Union,
+)
 
-import torch
-import torch.distributed as dist
+import torch  # noqa: E402  #pylint: disable=wrong-import-position
+import torch.distributed as dist  # noqa: E402  #pylint: disable=wrong-import-position
 
 try:
     import boto3