Skip to content

Commit 22f9af5

Browse files
yoblinclaude
andcommitted
nemotron: add draccus main() entry point for zephyr CLI compatibility
The zephyr CLI expects a main() function in the script. Without it, the job fails with AttributeError on startup. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1 parent 881e1a6 commit 22f9af5

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

lib/marin/src/marin/download/nemotron_cc/download_nemotron_cc.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
from collections.abc import Iterator
1919
from dataclasses import dataclass
2020

21+
import draccus
2122
import requests
2223
import zstandard
2324
from fray.cluster import ResourceConfig
@@ -147,3 +148,9 @@ def download_nemotron_cc(cfg: NemotronIngressConfig):
147148
ctx.execute(pipeline)
148149

149150
logger.info(f"Downloaded Nemotron CC files to {cfg.output_path}")
151+
152+
153+
@draccus.wrap()
154+
def main(cfg: NemotronIngressConfig) -> None:
155+
logging.basicConfig(level=logging.INFO)
156+
download_nemotron_cc(cfg)

0 commit comments

Comments
 (0)