[hotfix] got sliced types (#1614)

2 years ago · 6a8f8cc05e
parent d397842fa8
commit 6a8f8cc05e
1 changed files with 1 additions and 1 deletions
--- 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