fix lint

Author: DNXie

apps/grpo/main.py

@@ -14,7 +14,6 @@
 import torch.nn.functional as F
 from datasets import load_dataset
 from forge.actors.policy import EngineConfig, Policy, SamplingOverrides
-from forge.actors.reference_actor import compute_sequence_logprobs, TitanRefModel
 from forge.actors.replay_buffer import ReplayBuffer
 from forge.controller.actor import ForgeActor
 from forge.controller.service import ServiceConfig, shutdown_service, spawn_service

apps/vllm/main.py

@@ -30,7 +30,6 @@ async def run(cfg: DictConfig):
 
     print("Spawning service...")
 
-
     policy = await spawn_service(
         ServiceConfig(**cfg.policy.service), Policy, **cfg.policy
     )

src/forge/actors/policy.py

@@ -14,14 +14,7 @@
 from typing import Any, Dict, List
 
 import torch
-
-from forge.controller import ForgeActor, get_proc_mesh, stop_proc_mesh
-
-from forge.data.sharding import VLLMSharding
-from forge.interfaces import Policy as PolicyInterface
-from forge.types import ProcessConfig
 from monarch.actor import current_rank, endpoint, ProcMesh
-from omegaconf import DictConfig
 from torchstore import MultiProcessStore
 from torchstore._state_dict_utils import DELIM
 
@@ -45,6 +38,12 @@
 from vllm.v1.structured_output import StructuredOutputManager
 from vllm.worker.worker_base import WorkerWrapperBase
 
+from forge.controller import ForgeActor, get_proc_mesh, stop_proc_mesh
+
+from forge.data.sharding import VLLMSharding
+from forge.interfaces import Policy as PolicyInterface
+from forge.types import ProcessConfig
+
 
 logger = logging.getLogger(__name__)