From 1467d83edfd2ac5311e021eb6649413f3deae62e Mon Sep 17 00:00:00 2001
From: Frank Lee <somerlee.9@gmail.com>
Date: Wed, 18 May 2022 23:27:18 +0800
Subject: [PATCH] [cli] remove unused imports (#1001)

---
 colossalai/cli/benchmark/__init__.py        | 1 -
 colossalai/cli/benchmark/utils.py           | 2 --
 colossalai/cli/launcher/multinode_runner.py | 1 -
 3 files changed, 4 deletions(-)

diff --git a/colossalai/cli/benchmark/__init__.py b/colossalai/cli/benchmark/__init__.py
index 3653f1ffa..c020d33b6 100644
--- a/colossalai/cli/benchmark/__init__.py
+++ b/colossalai/cli/benchmark/__init__.py
@@ -1,4 +1,3 @@
-from random import choices
 import click
 
 from .utils import *
diff --git a/colossalai/cli/benchmark/utils.py b/colossalai/cli/benchmark/utils.py
index 0317f2a41..41ac38b82 100644
--- a/colossalai/cli/benchmark/utils.py
+++ b/colossalai/cli/benchmark/utils.py
@@ -1,10 +1,8 @@
 import math
 import time
-from grpc import Call
 import torch
 
 from colossalai.utils import MultiTimer
-from colossalai.core import global_context as gpc
 from colossalai.context import ParallelMode, Config
 from typing import List, Dict, Tuple, Callable
 
diff --git a/colossalai/cli/launcher/multinode_runner.py b/colossalai/cli/launcher/multinode_runner.py
index b3cabcbf2..c45ad5e5a 100644
--- a/colossalai/cli/launcher/multinode_runner.py
+++ b/colossalai/cli/launcher/multinode_runner.py
@@ -1,5 +1,4 @@
 import fabric
-from fabric import Connection
 from .hostinfo import HostInfo, HostInfoList
 from multiprocessing import Pipe, Process
 from multiprocessing import connection as mp_connection