Missing logarithm in function _component_log_likelihood in mfa.py

[salwig]

Dear @eitanrich,

It seems there is a missing logarithm in the function _component_log_likelihood in mfa.py (line 103).

Currently, it is written as:

return PI.reshape(1, K) + log_prob_data_given_components.T

I believe it should be:

return torch.log(PI).reshape(1, K) + log_prob_data_given_components.T

Could you please review this and update it if necessary?

Thank you in advance!