Skip to content

Commit

Permalink
feat: rename DltDataset to DeltaLiveTablesDataset and update dataset …
Browse files Browse the repository at this point in the history
…class naming convention
  • Loading branch information
chakravarthik27 committed Dec 2, 2024
1 parent 5589255 commit 3bd5ca0
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion langtest/datahandler/datasource.py
Original file line number Diff line number Diff line change
Expand Up @@ -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_", "")
Expand Down Expand Up @@ -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 = [
Expand Down

0 comments on commit 3bd5ca0

Please sign in to comment.