<reponame>sergeivolodin/causality-disentanglement-rl<filename>sparse_causal_model_learner_rl/loss/__init__.py<gh_stars>1-10
losses = {}

def loss(f):
    """Register a function as a loss."""
    losses[f.__name__] = f