Class TempGit

tests/python/ci/test_utils.py:32–55

A wrapper to run commands in a directory (specifically for use in CI tests)

class TempGit:
    """
    A wrapper to run commands in a directory (specifically for use in CI tests)
    """

    def __init__(self, cwd):
        self.cwd = cwd
        # Jenkins git is too old and doesn't have 'git init --initial-branch',
        # so init and checkout need to be separate steps
        self.run("init", stderr=subprocess.PIPE, stdout=subprocess.PIPE)
        self.run("checkout", "-b", "main", stderr=subprocess.PIPE)
        self.run("remote", "add", "origin", "https://github.com/apache/tvm.git")

    def run(self, *args, **kwargs):
        """
        Run a git command based on *args
        """
        proc = subprocess.run(
            ["git"] + list(args), encoding="utf-8", cwd=self.cwd, check=False, **kwargs
        )
        if proc.returncode != 0:
            raise RuntimeError(f"git command failed: '{args}'")

        return proc


def run_script(command: list[Any], check: bool = True, **kwargs):