fix import error

congchan · Nov 9, 2023 · b3c3d6b · b3c3d6b
1 parent 7a83b50
commit b3c3d6b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/data/utils.py b/data/utils.py
@@ -1,5 +1,7 @@
 """data utils"""
 
+import json
+
 def read_text(path, debug=None):
     " read in text data "
     data = []
@@ -14,6 +16,7 @@ def read_text(path, debug=None):
             data.append(line.strip())
     return data
 
+
 def read_json_line(file):
     tmp_ls = open(file, 'r', encoding='utf-8').readlines()
     tmp_ls = [json.loads(_.strip()) for _ in tmp_ls]

diff --git a/rlhf/reward_modeling/rm_utils.py b/rlhf/reward_modeling/rm_utils.py
@@ -1,6 +1,8 @@
-import datasets
+import logging
 import numpy as np
 import yaml
+
+import datasets
 import torch
 from torch.utils.data import DataLoader
 from torch import nn