AttributeError: 'DECIModule' object has no attribute 'sem_module'

[leizhangtech]

Hi,

Thank you for the great work. I was going through the sample code in multi_investment_sales_attribution.ipynb and everything seemed to be running smoothly during training. However, I encountered an error when executing the torch.save(lightning_module.sem_module, "deci.pt").
AttributeError: 'DECIModule' object has no attribute 'sem_module'.
Would you please help troubleshooting the error ? thanks

I installed causica with pip, seems the latest release version is 0.3.4.

[pawni]

Thanks for raising the issue @leizhangtech!

Do you have a longer example of when you're running into the issue? The sem_module should be accessible once the module is properly set up but not before. Are you able to use the latest version of the code?