From 72c944892003cde5c40c8515e31feeed900e285f Mon Sep 17 00:00:00 2001 From: HELSON Date: Wed, 9 Nov 2022 10:46:10 +0800 Subject: [PATCH] [NFC] polish colossalai/auto_parallel/tensor_shard/deprecated/op_handler/operator_handler.py code style (#1845) --- .../deprecated/op_handler/operator_handler.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/colossalai/auto_parallel/tensor_shard/deprecated/op_handler/operator_handler.py b/colossalai/auto_parallel/tensor_shard/deprecated/op_handler/operator_handler.py index 79f72d8d5..b120cc16b 100644 --- a/colossalai/auto_parallel/tensor_shard/deprecated/op_handler/operator_handler.py +++ b/colossalai/auto_parallel/tensor_shard/deprecated/op_handler/operator_handler.py @@ -1,14 +1,16 @@ +from abc import ABC, abstractmethod +from typing import Dict, List from webbrowser import Opera + import torch import torch.nn as nn -from abc import ABC, abstractmethod from torch.fx.node import Node -from typing import Dict, List + +from colossalai.auto_parallel.tensor_shard.deprecated.constants import * from colossalai.device.device_mesh import DeviceMesh from colossalai.tensor.sharding_spec import ShardingSpec -from .._utils import generate_resharding_costs, generate_sharding_spec -from colossalai.auto_parallel.tensor_shard.deprecated.constants import * +from .._utils import generate_resharding_costs, generate_sharding_spec from ..sharding_strategy import StrategiesVector __all__ = ['OperatorHandler'] @@ -60,7 +62,7 @@ class OperatorHandler(ABC): @abstractmethod def register_strategy(self) -> StrategiesVector: """ - Register + Register """ pass