utils.py

from __future__ import annotations

import pathlib


def find_exp_dirs(ignore_repo: bool = False) -> list[str]:
    repo_dir = pathlib.Path(__file__).parent
    exp_root_dir = repo_dir / 'experiments'
    if not exp_root_dir.exists():
        return []
    exp_dirs = sorted(exp_root_dir.glob('*'))
    exp_dirs = [
        exp_dir for exp_dir in exp_dirs
        if (exp_dir / 'pytorch_lora_weights.bin').exists()
    ]
    if ignore_repo:
        exp_dirs = [
            exp_dir for exp_dir in exp_dirs if not (exp_dir / '.git').exists()
        ]
    return [path.relative_to(repo_dir).as_posix() for path in exp_dirs]