third_party/InternVideo/InternVideo2/multi_modality/tools/utils.py

import os
import shutil
import socket


def has_slurm():
    """determine the system has slurm or not
    Returns: True if has else False.

    """
    return shutil.which("sbatch") is not None

def random_port():
    """random a unused port
    Returns: str

    """
    with socket.socket() as s:
        s.bind(("", 0))
        return s.getsockname()[1]

def runcmd(cmd):
    """run command

    Args:
        cmd (str): The command to run

    """
    os.system(cmd)