Closed leogagnon closed 4 months ago
LGTM, pinging @frankaging: have you ever used modules which output dicts? Hasn't come up for me but this fix seems reasonable.
LGTM! Later, we might want to further remove the assumption of intervening on the first element by creating field in the config to provide the intervening keyword.
@leogagnon Thanks!
Description
Add simple (and naive) for dictionary output of modules. Similar to how
tuples
s were dealt with, I just select the first value of the dict. Only made changes toBaseModel
. I'm not yet super familiar with the code so there might be some other modification needed.Testing Done
It works with my model which has module dict outputs (for ZeroIntervention and LoreftIntervention). Didn't do extensive testing as this felt quite straightforward.
Checklist:
[Your Priority] Your Title