"""
This repo is forked from [Boyuan Chen](https://boyuan.space/)'s research 
template [repo](https://github.com/buoyancy99/research-template). 
By its MIT license, you must keep the above sentence in `README.md` 
and the `LICENSE` file to credit the author.
"""

import torch
from pathlib import Path


def safe_torch_save(obj, path: Path):
    """
    Safely save a torch object to disk.
    If the path does not exist, create folders as needed.
    """
    path.parent.mkdir(parents=True, exist_ok=True)
    torch.save(obj, path)