Discard Aim run when exception is raised

[luispsantos]

Discard Aim run when exception is raised

[mihran113]

Hey @luispsantos! Thanks a lot for opening the issue. The resource cleanup is a kind of a messy area currently, as we have done some magic tricks to close the run resources implicitly as well. We'll revisit that in the upcoming releases as we have some other issues there as well. But for now the below script should do the job:

from aim import Run, Repo

repo = Repo.from_path('.')
run = Run(repo=repo)
exception_occurred = False
try:
    ...  # complex code for ML experiments
    1 / 0  # simulate exception being raised
except:
    exception_occurred = True
finally:
    run.finalize()
    run.close()
    run_hash = run.hash
    run = None
    if exception_occurred:
        repo.delete_run(run_hash)
        print(f"Run {run_hash} discarded")

[luispsantos]

Hi @mihran113. Thanks a lot for your fix, this indeed fixed my issue! The current way of discarding runs is not very straightforward in my opinion, but I guess there'll be a rework of this on future releases. We can close this ticket 👍