PrivacyConfig is defined in src/agentunit/privacy/wrappers.py but is not actually used by PrivateDatasetWrapper.
Task:
- Update
PrivateDatasetWrapper.__init__ to accept an optional PrivacyConfig object.
- Use values from the config (like
enable_pii_masking, epsilon, etc.) instead of or in addition to individual arguments.
- Ensure consistency between the config dataclass and the wrapper's capabilities.
PrivacyConfigis defined insrc/agentunit/privacy/wrappers.pybut is not actually used byPrivateDatasetWrapper.Task:
PrivateDatasetWrapper.__init__to accept an optionalPrivacyConfigobject.enable_pii_masking,epsilon, etc.) instead of or in addition to individual arguments.