Skip to content

Commit 7314bff

Browse files
committed
polish flake8
1 parent 68db31f commit 7314bff

File tree

3 files changed

+190
-167
lines changed

3 files changed

+190
-167
lines changed

ding/reward_model/math_reward_model.py

+1-3
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)