From 362ac74c28fafe60d1a7fb3b2cad7a1c976a2ecb Mon Sep 17 00:00:00 2001
From: Rita Kurban <ritakurban@Ritas-Mac.local>
Date: Fri, 27 Dec 2024 13:28:20 +0000
Subject: [PATCH] Rename TextClassification

---
 src/eva/language/data/datasets/classification/base.py     | 2 +-
 src/eva/language/data/datasets/classification/pubmedqa.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/eva/language/data/datasets/classification/base.py b/src/eva/language/data/datasets/classification/base.py
index ca081ff8..7adaa4db 100644
--- 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
index 7f185c30..9c07c956 100644
--- 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)"