Skip to content

Commit 8695afe

Browse files
authored
Add HAP/detoxify module to Python TrustyAI (#197)
* Add TMaRCo to detoxify module * Revert changes to fairness/__init__.py * Fix linting
1 parent 3cf3fa8 commit 8695afe

File tree

4 files changed

+789
-0
lines changed

4 files changed

+789
-0
lines changed

pyproject.toml

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,16 @@ extras = [
5353
"aix360[default,tsice,tslime,tssaliency]==0.3.0"
5454
]
5555

56+
detoxify = [
57+
"transformers~=4.36.2",
58+
"datasets",
59+
"scipy",
60+
"torch",
61+
"iter-tools",
62+
"evaluate",
63+
"trl"
64+
]
65+
5666
[project.urls]
5767
homepage = "https://github.com/trustyai-explainability/trustyai-explainability-python"
5868
documentation = "https://trustyai-explainability-python.readthedocs.io/en/latest/"

src/trustyai/language/__init__.py

Whitespace-only changes.
Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
"""Language detoxification module."""
2+
from trustyai.language.detoxify.tmarco import TMaRCo

0 commit comments

Comments
 (0)