Skip to content

Commit 2a129cb

Browse files
committed
update
1 parent 65affa2 commit 2a129cb

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

verl/utils/transferqueue_utils.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,9 +20,12 @@
2020
import numpy as np
2121
import torch
2222
from tensordict import NonTensorData, NonTensorStack, TensorDict
23-
from transfer_queue import AsyncTransferQueueClient, BatchMeta
2423

25-
from verl.experimental.transfer_queue import ZMQServerInfo
24+
from verl.experimental.transfer_queue import (
25+
AsyncTransferQueueClient,
26+
BatchMeta,
27+
ZMQServerInfo,
28+
)
2629
from verl.protocol import DataProto
2730

2831
# _TRANSFER_QUEUE_CONTROLLER_INFOS = None

0 commit comments

Comments
 (0)