From 4cadb25b96d3fef31e27a83e0fbede7f5bcb9153 Mon Sep 17 00:00:00 2001 From: CZYCW Date: Tue, 28 Mar 2023 10:31:02 +0800 Subject: [PATCH] [NFC] policy colossalai/fx/proxy.py code style (#3269) --- colossalai/fx/proxy.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/colossalai/fx/proxy.py b/colossalai/fx/proxy.py index 06272c48f..7317072c6 100644 --- a/colossalai/fx/proxy.py +++ b/colossalai/fx/proxy.py @@ -1,7 +1,9 @@ import operator +from typing import Any, List, Union + import torch -from torch.fx.proxy import Proxy, Attribute -from typing import List, Union, Any +from torch.fx.proxy import Attribute, Proxy + from colossalai.fx.tracer.meta_patch import meta_patched_function __all__ = ['ColoProxy']