Add ICL base class to __all__ (#3079)

* add to all * fix
mosaicml · Mar 2, 2024 · 548842c · 548842c
1 parent 6ce38ad
commit 548842c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/composer/metrics/nlp.py b/composer/metrics/nlp.py
@@ -22,6 +22,7 @@
 log = logging.getLogger(__name__)
 
 __all__ = [
+    'InContextLearningMetric',
     'InContextLearningLMAccuracy',
     'InContextLearningMultipleChoiceAccuracy',
     'InContextLearningQAAccuracy',
@@ -196,6 +197,7 @@ def compute(self) -> Tensor:
 
 
 class InContextLearningMetric(Metric):
+    """Base class for In-context learning (ICL) metrics."""
 
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)