A typo in model.py?

[XZ-X]

Hi,

Recently I'm following up DIRTY. The work is great, and the code repo is impressive. Thank the developers for the efforts!

Just want to mention that there might be a typo in the following file: (dirty/model/model.py:496), the variable in task_targets should be targets, instead of preds.

DIRTY/dirty/model/model.py

Line 496 in 7a5514a

f"{task}_targets": preds,