From e871de693cf36936be05a02ed67b5ace3522e43a Mon Sep 17 00:00:00 2001 From: Alexander Goryunov Date: Sun, 1 Jan 2023 21:52:38 +0200 Subject: [PATCH] A typo in import --- model/reward/instructor/cls_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model/reward/instructor/cls_dataset.py b/model/reward/instructor/cls_dataset.py index 09aa821b..7992c37c 100644 --- a/model/reward/instructor/cls_dataset.py +++ b/model/reward/instructor/cls_dataset.py @@ -8,7 +8,7 @@ import json import os import random -from dataset import load_dataset +from datasets import load_dataset from torch.utils.data import Dataset from .utils import webgpt_return_format