diff --git a/langtest/datahandler/datasource.py b/langtest/datahandler/datasource.py index 3cef616b1..c58202608 100644 --- a/langtest/datahandler/datasource.py +++ b/langtest/datahandler/datasource.py @@ -141,6 +141,10 @@ def __init_subclass__(cls, **kwargs): import pandas as pd dataset_cls = cls.__name__.replace("Dataset", "").lower() + + if dataset_cls in ["deltalivetables"]: + dataset_cls = "delta_live_tables" + if dataset_cls == "pandas": extensions = [ i.replace("read_", "") @@ -2014,7 +2018,7 @@ def export_data(self, data: List[Sample], output_path: str): raise NotImplementedError() -class DltDataset(BaseDataset): +class DeltaLiveTablesDataset(BaseDataset): """A class to handle datasets from the Delta Live Tables(DLT).""" supported_tasks = [