| """Publish the full Beam Boot / Super Beam source as a public Hugging Face model repo. |
| |
| Uploads code only: beamboot/, tests/, scripts/, roms/, examples/, envs/, README.md, pyproject.toml, |
| SUBMISSION.md. Virtual environments, checkpoints (artifacts/), results/ and outputs/ are not uploaded. |
| The HF write token comes from the Modal secret and is held in memory only; it is never printed. |
| |
| .venv/bin/modal run scripts/publish_source.py |
| """ |
| import os |
| from pathlib import Path |
|
|
| import modal |
|
|
| app = modal.App('beamboot-publish-source') |
| ROOT = Path(__file__).resolve().parent.parent |
| INCLUDE = ['beamboot', 'tests', 'scripts', 'roms', 'examples', 'envs', |
| 'README.md', 'pyproject.toml', 'SUBMISSION.md', 'figures'] |
|
|
|
|
| @app.function(image=modal.Image.debian_slim(python_version='3.12'), |
| secrets=[modal.Secret.from_name('steerability-submission-write')]) |
| def get_write_token(): |
| token = os.environ.get('HUGGINGFACE_TOKEN') or os.environ.get('HF_TOKEN') |
| if not token: |
| raise RuntimeError('No HF write token in configured secret') |
| return token |
|
|
|
|
| @app.local_entrypoint() |
| def main(): |
| from huggingface_hub import HfApi |
| api = HfApi(token=get_write_token.remote()) |
| repo_id = api.whoami()['name'] + '/beam-boot' |
| api.create_repo(repo_id=repo_id, repo_type='model', private=False, exist_ok=True) |
| allow = [] |
| for item in INCLUDE: |
| allow.append(item + '/**' if (ROOT / item).is_dir() else item) |
| api.upload_folder(repo_id=repo_id, repo_type='model', folder_path=str(ROOT), |
| allow_patterns=allow, |
| ignore_patterns=['**/__pycache__/**', '**/*.pyc', '**/.pytest_cache/**'], |
| commit_message='Publish Beam Boot / Super Beam source') |
| print('Public source: https://huggingface.co/' + repo_id, flush=True) |
|
|
