code: def set_input_partition_id(self, partition_id: int):
      self._input_partition_id = partition_id
  
imports: from dataclasses import dataclass
  from typing import Dict, List
label_desc: Use image node_id 2 for calling the Topo obj's underlying member method code with example usage: obj.set_input_partition_id(partition_id) without return types