From 1c29b173c9baa12a2c3c81b66041e95007128b0b Mon Sep 17 00:00:00 2001 From: Zirui Zhu Date: Tue, 3 Jan 2023 18:17:06 +0800 Subject: [PATCH] [NFC] polish colossalai/auto_parallel/tensor_shard/node_handler/getitem_handler.py code style (#2289) --- .../auto_parallel/tensor_shard/node_handler/getitem_handler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/colossalai/auto_parallel/tensor_shard/node_handler/getitem_handler.py b/colossalai/auto_parallel/tensor_shard/node_handler/getitem_handler.py index 25baa7766..3466e9dd9 100644 --- a/colossalai/auto_parallel/tensor_shard/node_handler/getitem_handler.py +++ b/colossalai/auto_parallel/tensor_shard/node_handler/getitem_handler.py @@ -6,7 +6,7 @@ import torch from ..sharding_strategy import OperationData, OperationDataType from .node_handler import NodeHandler from .registry import operator_registry -from .strategy import (StrategyGenerator, TensorStrategyGenerator, TensorTupleStrategyGenerator) +from .strategy import StrategyGenerator, TensorStrategyGenerator, TensorTupleStrategyGenerator __all__ = ['GetItemHandler']