clean unused package . (#239)

Signed-off-by: zhanluxianshen <[email protected]>

tests/e2e/arithmetic_sequence/rl/main_trainer.py

@@ -14,22 +14,16 @@
 """
 Using FSDPTrainer
 """
-import re
 import os
 import hydra
-import numpy as np
 import ray
 import torch
-from torch.utils.data import DataLoader
 from transformers import PreTrainedTokenizer, AutoTokenizer
 
-import verl.utils.torch_functional as verl_F
 from verl import DataProto
 from verl.trainer.ppo.ray_trainer import RayPPOTrainer
 from verl.utils.fs import copy_local_path_from_hdfs
-from verl.utils.model import compute_position_id_with_mask
 from tests.e2e.envs.digit_completion import CharTokenizer
-import pandas as pd
 
 
 def make_reward_function(tokenizer, num_examine):

verl/models/transformers/llama.py

@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import os
 import torch
 from typing import Optional, List, Union, Tuple, Unpack, Callable
 

verl/models/transformers/qwen2.py

@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import os
 import torch
 from typing import Optional, Tuple
 

verl/trainer/main_ppo.py

@@ -109,8 +109,6 @@ def run_ppo(config, compute_score=None):
 @ray.remote
 def main_task(config, compute_score=None):
     from verl.utils.fs import copy_local_path_from_hdfs
-    from transformers import AutoTokenizer
-
     # print initial config
     from pprint import pprint
     from omegaconf import OmegaConf

verl/utils/torch_functional.py

@@ -17,7 +17,6 @@
 
 from typing import Dict, Union, List, Optional
 
-import os
 import torch
 import torch.distributed
 import torch.nn.functional as F