#!/usr/bin/env python3
"""
Python wrapper to launch dflash training script via northjob/torchrun
"""
import subprocess
import sys
import os

if __name__ == "__main__":
    bash_script = "/workspace/hanrui/syxin_old/run_train_multinode_dflash.sh"
    args = sys.argv[1:]

    cmd = ["bash", bash_script] + args

    result = subprocess.run(cmd, env=os.environ.copy())

    sys.exit(result.returncode)