"""Log a short training loop to LaunchHelm.

Set LAUNCHHELM_URL, LAUNCHHELM_API_KEY, LAUNCHHELM_ORGANIZATION and
LAUNCHHELM_PROJECT in the environment or a .env file, then:

    python log_run.py
"""

import launchhelm

with launchhelm.init("quickstart", name="first-run", config={"lr": 0.0003}) as run:
    for step in range(8):
        loss = 2.0 / (step + 1)
        run.log({"loss": loss, "lr": 0.0003}, step=step)

print("run finished:", run.run_id)