| 152 | |
| 153 | |
| 154 | class ReplaceVariableCommand(GraphCommand): |
| 155 | def __init__(self, var_name: str, replace_to: Variable) -> None: |
| 156 | super().__init__(command_type=GraphCommandType.REPLACE_VAR) |
| 157 | self.op_name = var_name |
| 158 | self.replace_to = replace_to |
| 159 | |
| 160 | |
| 161 | class TruncateGraphCommand(GraphCommand): |