Fix Distribution Inference Module

[asim29]

A lot of work needs to be done in this module, specifically on a high-level:

• Need to design a strategy such that the attack module itself works on a single target model. However, the evaluation requires multiple modules. The metrics module might need to implement a full pipeline.
• Need to generalize the code such that it works with any number of sensitive attributes, using only their indexes rather than hardcoded values for the sensitive attributes.

[sebszyller]

types missing in the attack

[asim29]

I added types to a lot of the extra functions, but the main attack function (SuriSATML2023.attack()) I will likely change entirely. I'll do that in the same PR.

[sebszyller]

this is still WIP, right?

[asim29]

this is still WIP, right?

yes, slightly lower priority right now