Rename TextClassification

kaiko-ai · Dec 27, 2024 · 362ac74 · 362ac74
1 parent cf878b9
commit 362ac74
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/eva/language/data/datasets/classification/base.py b/src/eva/language/data/datasets/classification/base.py
@@ -8,7 +8,7 @@
 
 from eva.language.data.datasets.language import LanguageDataset
 
-class LanguageClassification(LanguageDataset[Tuple[str, torch.Tensor]], abc.ABC):
+class TextClassification(LanguageDataset[Tuple[str, torch.Tensor]], abc.ABC):
     """Text classification abstract dataset."""
 
     def __init__(self) -> None:

diff --git a/src/eva/language/data/datasets/classification/pubmedqa.py b/src/eva/language/data/datasets/classification/pubmedqa.py
@@ -8,7 +8,7 @@
 from eva.language.data.datasets.classification import base
 
 
-class PubMedQA(base.LanguageClassification):
+class PubMedQA(base.TextClassification):
     """Dataset class for PubMedQA question answering task."""
 
     _license: str = "MIT License (https://github.com/pubmedqa/pubmedqa/blob/master/LICENSE)"