[bug fixed] modify the wrong dependency (#277)

TUDB-Labs · Dec 9, 2024 · 6611ae3 · 6611ae3
1 parent b3dc8af
commit 6611ae3
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/pipeline_example.py b/tests/pipeline_example.py
@@ -1,7 +1,7 @@
-from mlora.pipeline.transport import RpcTransport
-from mlora.pipeline.function import SendOperator, RecvOperator
-from mlora.pipeline.messages import PipeMessageType
-from mlora.pipeline.stream import CudaStream
+from mlora.executor.pipeline.rpc_transport import RpcTransport
+from mlora.executor.pipeline.function import SendOperator, RecvOperator
+from mlora.executor.pipeline.messages import PipeMessageType
+from mlora.executor.pipeline.stream import CudaStream
 from mlora.utils import setup_seed
 
 import os