Skip to content

Commit 7a4c343

Browse files
committed
Minor cleanup
1 parent c022837 commit 7a4c343

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

src/forge/actors/policy.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13,13 +13,10 @@
1313
from typing import Dict, List
1414

1515
import torch
16-
from forge.controller.service import ServiceConfig
17-
from forge.controller.spawn import spawn_service
1816

1917
from forge.data.sharding import VLLMSharding
2018
from forge.interfaces import Policy as PolicyInterface
2119
from monarch.actor import Actor, current_rank, endpoint, proc_mesh
22-
from omegaconf import DictConfig
2320
from torchstore import MultiProcessStore
2421
from torchstore._state_dict_utils import DELIM
2522

@@ -337,7 +334,6 @@ async def setup(self, store: MultiProcessStore = None):
337334
self.torchstore = store
338335
# TODO: remove ["gpus"] when monarch implements a flat rank
339336
self.rank = current_rank()["gpus"]
340-
# print(f" RAnk: {self.rank}")
341337
self.worker = self.setup_worker()
342338

343339
@endpoint

0 commit comments

Comments
 (0)