From c4a226b7294683805e3f375a903f3ca40c5fb119 Mon Sep 17 00:00:00 2001
From: Camille Zhong <44392324+Camille7777@users.noreply.github.com>
Date: Mon, 27 Mar 2023 22:03:46 +0800
Subject: [PATCH] [NFC] polish tensor_placement_policy.py code style (#3265)

---
 colossalai/gemini/tensor_placement_policy.py | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/colossalai/gemini/tensor_placement_policy.py b/colossalai/gemini/tensor_placement_policy.py
index cfcfb3856..0e575254c 100644
--- a/colossalai/gemini/tensor_placement_policy.py
+++ b/colossalai/gemini/tensor_placement_policy.py
@@ -1,16 +1,16 @@
+import functools
 from abc import ABC, abstractmethod
 from time import time
-from typing import List, Optional
+from typing import List, Optional, Type
+
 import torch
+
+from colossalai.gemini.memory_tracer import MemStatsCollector
+from colossalai.gemini.stateful_tensor import StatefulTensor
+from colossalai.gemini.tensor_utils import colo_model_data_tensor_move_inline, colo_tensor_mem_usage
 from colossalai.utils import get_current_device
 from colossalai.utils.memory import colo_device_memory_capacity
 
-from colossalai.gemini.tensor_utils import colo_model_data_tensor_move_inline, colo_tensor_mem_usage
-from colossalai.gemini.stateful_tensor import StatefulTensor
-from colossalai.gemini.memory_tracer import MemStatsCollector
-from typing import Type
-import functools
-
 
 class TensorPlacementPolicy(ABC):