Skip to content

Commit d519b12

Browse files
committed
polish flake8
1 parent 68db31f commit d519b12

File tree

3 files changed

+170
-137
lines changed

3 files changed

+170
-137
lines changed

ding/reward_model/math_reward_model.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,9 @@
1-
from typing import Tuple, Optional, List, Dict
1+
from typing import List, Dict
22
from easydict import EasyDict
33
from torch.utils.tensorboard import SummaryWriter
44
from transformers import AutoTokenizer, AutoModel
55
import torch
66
import torch.nn.functional as F
7-
import re
8-
97
from ding.utils import REWARD_MODEL_REGISTRY
108
from .base_reward_model import BaseRewardModel
119

0 commit comments

Comments
 (0)