From 6a8f8cc05e9ca919fff15d6cba0496adcc3f5853 Mon Sep 17 00:00:00 2001
From: YuliangLiu0306 <72588413+YuliangLiu0306@users.noreply.github.com>
Date: Tue, 20 Sep 2022 14:32:42 +0800
Subject: [PATCH] [hotfix] got sliced types (#1614)

---
 colossalai/auto_parallel/solver/sharding_strategy.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/colossalai/auto_parallel/solver/sharding_strategy.py b/colossalai/auto_parallel/solver/sharding_strategy.py
index 342f6ff37..e7429d386 100644
--- a/colossalai/auto_parallel/solver/sharding_strategy.py
+++ b/colossalai/auto_parallel/solver/sharding_strategy.py
@@ -105,7 +105,7 @@ class StrategyGenerator_V2(ABC):
         self.device_mesh = device_mesh
 
     @abstractmethod
-    def generate(self, operand_mapping: Dict[str:Operand]) -> List[ShardingStrategy_V2]:
+    def generate(self, operand_mapping: Dict[str, Operand]) -> List[ShardingStrategy_V2]:
         """
         """
         pass