diff --git a/model/reward/instructor/rank_datasets.py b/model/reward/instructor/rank_datasets.py index 0d710124..bd53eb02 100644 --- a/model/reward/instructor/rank_datasets.py +++ b/model/reward/instructor/rank_datasets.py @@ -18,9 +18,8 @@ """ -from typing import List, Dict from dataclasses import dataclass -from typing import Optional, Union +from typing import Dict, List, Optional, Union import numpy as np import torch