[fix] linter fix

This commit is contained in:
theblackcat102
2023-01-20 07:23:02 +00:00
parent c255148dc6
commit 22e3ab1a89
5 changed files with 70 additions and 79 deletions
@@ -2,7 +2,7 @@
High level functions for model training
"""
from custom_datasets.prompt_dialogue import InstructionTuning, PromptGeneratedDataset
from custom_datasets.qa_datasets import SODA, JokeExplaination, QADataset, WebGPT, SODADialogue
from custom_datasets.qa_datasets import SODA, JokeExplaination, QADataset, SODADialogue, WebGPT
from custom_datasets.summarization import SummarizationDataset
from custom_datasets.toxic_conversation import ProsocialDialogue, ProsocialDialogueExplaination
from custom_datasets.translation import WMT2019, DiveMT, TEDTalk