setup

ask_download_dataset_statistics(statistics_dir: Path, repo_id: str = 'sciai-lab/structures25')[source]

Ask user if they want to download models.

ask_download_models(models_dir: Path, repo_id: str = 'sciai-lab/structures25')[source]

Ask user if they want to download models.

ask_path(prompt: str, default: Path) Path[source]

Prompt user for a directory path with a default.

download_dataset_statistics(repo_id: str, target_dir: Path)[source]

Download dataset statistics from Hugging Face Hub into target directory.

download_model(model_name: str, repo_id: str, target_dir: Path)[source]

Download a model from Hugging Face Hub into target directory.

main()[source]

Main setup function.

query_yes_no(question, default='yes')[source]

Ask a yes/no question via raw_input() and return their answer.

“question” is a string that is presented to the user. “default” is the presumed answer if the user just hits <Enter>. It must be “yes” (the default), “no” or None (meaning an answer is required of the user).

The “answer” return value is True for “yes” or False for “no”.