Exposed return types for testing

explabox/digestibles/__init__.py

@@ -13,6 +13,6 @@
 
 """Ingestibles are turned into digestibles, containing information to explore/examine/explain/expose your model."""
 
-from .digestibles import Dataset, Descriptives, Performance, WronglyClassified
+from .digestibles import Dataset, Descriptives, Instances, Performance, WronglyClassified
 
-__all__ = ["Dataset", "Descriptives", "Performance", "WronglyClassified"]
+__all__ = ["Dataset", "Descriptives", "Instances", "Performance", "WronglyClassified"]

explabox/examine/__init__.py

@@ -13,6 +13,6 @@
 
 """Calculate quantitative metrics on how the model performs, and examine where the model went wrong."""
 
-from .examiner import Examiner
+from .examiner import Examiner, Performance, WronglyClassified
 
-__all__ = ["Examiner"]
+__all__ = ["Examiner", "Performance", "WronglyClassified"]

explabox/explain/__init__.py

@@ -13,6 +13,6 @@
 
 """Add explainability to your model/dataset with the Explainer class."""
 
-from .text import Explainer
+from .text import Explainer, FeatureList, Instances
 
-__all__ = ["Explainer"]
+__all__ = ["Explainer", "FeatureList", "Instances"]

explabox/explain/text/__init__.py

@@ -13,6 +13,6 @@
 
 """Add explainability to your text model/dataset."""
 
-from .explainer import Explainer
+from .explainer import Explainer, FeatureList, Instances
 
-__all__ = ["Explainer"]
+__all__ = ["Explainer", "FeatureList", "Instances"]

explabox/explore/__init__.py

@@ -13,6 +13,6 @@
 
 """Functions/classes for exploring your data (dataset descriptives)."""
 
-from .explorer import Explorer
+from .explorer import Explorer, Dataset, Descriptives
 
-__all__ = ["Explorer"]
+__all__ = ["Explorer", "Dataset", "Descriptives"]

explabox/expose/__init__.py

@@ -13,6 +13,6 @@
 
 """Functions/classes for sensitivity testing (fairness and robustness)."""
 
-from .text import Exposer
+from .text import Exposer, LabelMetrics, MeanScore, SuccessTest
 
-__all__ = ["Exposer"]
+__all__ = ["Exposer", "LabelMetrics", "MeanScore", "SuccessTest"]

explabox/expose/text/__init__.py

@@ -25,6 +25,8 @@
 from ...ui.notebook import restyle
 from .exposer import (
     Exposer,
+    LabelMetrics,
+    MeanScore,
     RandomAscii,
     RandomCyrillic,
     RandomDigits,
@@ -35,6 +37,7 @@
     RandomString,
     RandomUpper,
     RandomWhitespace,
+    SuccessTest,
     compare_metric,
 )
 
@@ -44,6 +47,8 @@
 
 __all__ = [
     "Exposer",
+    "LabelMetrics",
+    "MeanScore",
     "OneToManyPerturbation",
     "OneToOnePerturbation",
     "RandomAscii",
@@ -55,6 +60,7 @@
     "RandomString",
     "RandomUpper",
     "RandomWhitespace",
+    "SuccessTest",
     "compare_accuracy",
     "compare_metric",
     "compare_precision",