From 41159cdf16b47305e3b63cd20bc8b4976457f784 Mon Sep 17 00:00:00 2001 From: 0oshowero0 Date: Fri, 30 Jan 2026 20:04:06 +0800 Subject: [PATCH 1/2] add wechat Signed-off-by: 0oshowero0 --- README.md | 1 + 1 file changed, 1 insertion(+) diff --git a/README.md b/README.md index f87ccc0..7f46346 100644 --- a/README.md +++ b/README.md @@ -4,6 +4,7 @@ Paper | Zhihu +| WeChat

From 49fce537223345bad2e6bcb72fa0bf7c89d2c365 Mon Sep 17 00:00:00 2001 From: 0oshowero0 Date: Fri, 30 Jan 2026 20:16:16 +0800 Subject: [PATCH 2/2] fix CI Signed-off-by: 0oshowero0 --- .github/workflows/python-package.yml | 4 ++-- scripts/performance_test.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml index 9e4c6f6..f12c64e 100644 --- a/.github/workflows/python-package.yml +++ b/.github/workflows/python-package.yml @@ -45,9 +45,9 @@ jobs: pip install dist/*.whl --force-reinstall - name: Test with pytest (TQ_ZERO_COPY_SERIALIZATION=False) run: | - pytest + pytest tests - name: Test with pytest (TQ_ZERO_COPY_SERIALIZATION=True) run: | ray stop --force export TQ_ZERO_COPY_SERIALIZATION=True - pytest \ No newline at end of file + pytest tests \ No newline at end of file diff --git a/scripts/performance_test.py b/scripts/performance_test.py index 9586b40..396ddb2 100644 --- a/scripts/performance_test.py +++ b/scripts/performance_test.py @@ -32,8 +32,8 @@ from transfer_queue import ( # noqa: E402 - AsyncTransferQueueClient, SimpleStorageUnit, + TransferQueueClient, TransferQueueController, process_zmq_server_info, ) @@ -247,7 +247,7 @@ def _initialize_data_system(self): tq_config.storage_unit_infos = self.data_system_storage_unit_infos self.config = OmegaConf.merge(tq_config, self.config) - self.data_system_client = AsyncTransferQueueClient( + self.data_system_client = TransferQueueClient( client_id="Trainer", controller_info=self.data_system_controller_info ) self.data_system_client.initialize_storage_manager(manager_type="AsyncSimpleStorageManager", config=self.config)