From d5884a62bb58f238d74c9ae9553d66b667d43a68 Mon Sep 17 00:00:00 2001 From: MarcelRobeer Date: Fri, 3 Oct 2025 10:14:05 +0200 Subject: [PATCH] Exposed return types for testing --- explabox/digestibles/__init__.py | 4 ++-- explabox/examine/__init__.py | 4 ++-- explabox/explain/__init__.py | 4 ++-- explabox/explain/text/__init__.py | 4 ++-- explabox/explore/__init__.py | 4 ++-- explabox/expose/__init__.py | 4 ++-- explabox/expose/text/__init__.py | 6 ++++++ 7 files changed, 18 insertions(+), 12 deletions(-) diff --git a/explabox/digestibles/__init__.py b/explabox/digestibles/__init__.py index 83df44f..421ac30 100644 --- a/explabox/digestibles/__init__.py +++ b/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"] diff --git a/explabox/examine/__init__.py b/explabox/examine/__init__.py index 229a141..4a2d7c1 100644 --- a/explabox/examine/__init__.py +++ b/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"] diff --git a/explabox/explain/__init__.py b/explabox/explain/__init__.py index 19725ba..8798e08 100644 --- a/explabox/explain/__init__.py +++ b/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"] diff --git a/explabox/explain/text/__init__.py b/explabox/explain/text/__init__.py index 55e427c..f7c5000 100644 --- a/explabox/explain/text/__init__.py +++ b/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"] diff --git a/explabox/explore/__init__.py b/explabox/explore/__init__.py index e96a292..022ad94 100644 --- a/explabox/explore/__init__.py +++ b/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"] diff --git a/explabox/expose/__init__.py b/explabox/expose/__init__.py index 302d217..f8cd9a0 100644 --- a/explabox/expose/__init__.py +++ b/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"] diff --git a/explabox/expose/text/__init__.py b/explabox/expose/text/__init__.py index d571cca..0024537 100644 --- a/explabox/expose/text/__init__.py +++ b/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",