Unverified Commit 6a8f8cc0 authored by YuliangLiu0306's avatar YuliangLiu0306 Committed by GitHub
Browse files

[hotfix] got sliced types (#1614)

parent d397842f
......@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment