| from colab_train_llama32_remote import ( | |
| hf_space_repo_to_base_url, | |
| make_training_args, | |
| ) | |
| def test_hf_space_repo_to_base_url_formats_standard_hf_space_domain(): | |
| assert ( | |
| hf_space_repo_to_base_url("Ev3Dev/hackathon") | |
| == "https://ev3dev-hackathon.hf.space" | |
| ) | |
| def test_make_training_args_derives_base_url_when_missing(): | |
| args = make_training_args(space_repo_id="Ev3Dev/hackathon", base_url="") | |
| assert args.base_url == "https://ev3dev-hackathon.hf.space" | |
| assert args.model_id == "unsloth/Llama-3.2-3B-Instruct-bnb-4bit" | |