Add HAP/detoxify module to Python TrustyAI (#197)

* Add TMaRCo to detoxify module * Revert changes to fairness/__init__.py * Fix linting
trustyai-explainability · Jan 22, 2024 · 8695afe · 8695afe
1 parent 3cf3fa8
commit 8695afe
Show file tree

Hide file tree

Showing 4 changed files with 789 additions and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -53,6 +53,16 @@ extras = [
     "aix360[default,tsice,tslime,tssaliency]==0.3.0"
 ]
 
+detoxify = [
+    "transformers~=4.36.2",
+    "datasets",
+    "scipy",
+    "torch",
+    "iter-tools",
+    "evaluate",
+    "trl"
+]
+
 [project.urls]
 homepage = "https://github.com/trustyai-explainability/trustyai-explainability-python"
 documentation = "https://trustyai-explainability-python.readthedocs.io/en/latest/"

diff --git a/src/trustyai/language/__init__.py b/src/trustyai/language/__init__.py
diff --git a/src/trustyai/language/detoxify/__init__.py b/src/trustyai/language/detoxify/__init__.py
@@ -0,0 +1,2 @@
+"""Language detoxification module."""
+from trustyai.language.detoxify.tmarco import TMaRCo
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		"""Language detoxification module."""
		from trustyai.language.detoxify.tmarco import TMaRCo