    def push(self, idx: typing.Union[int, str], cmd: str):
        """Push a task into runner

        Parameters
        ----------
        idx : Union[int, str]
            id of the task
        cmd : str
            bash command line for the task

        """
        raise NotImplementedError
Push a task into runner

Parameters
----------
idx : Union[int, str]
id of the task
cmd : str
bash command line for the task
python/aitemplate/backend/task_runner.py