Skip to content

Conversation

@hiyuchang
Copy link
Collaborator

Description

  1. Remove @WORKFLOWS.register("math_boxed_workflow"), use the mapping scheme instead. Fix the import accordingly
  2. Add a registry test

Checklist

Please check the following items before code is ready to be reviewed.

  • Code has passed all tests
  • Docstrings have been added/updated in Google Style
  • Documentation has been updated
  • Code is ready for review

@gemini-code-assist
Copy link
Contributor

Summary of Changes

Hello @hiyuchang, I'm Gemini Code Assist1! I'm currently reviewing this pull request and will post my feedback shortly. In the meantime, here's a summary to help you and other reviewers quickly get up to speed!

This pull request significantly refactors the module registration system, moving from a decorator-based approach to a centralized, explicit mapping within Registry objects. This change aims to streamline module discovery, improve code clarity by removing decorators from class definitions, and enhance the overall maintainability and extensibility of the framework. It also includes a new test suite to ensure the robustness of the updated registry system.

Highlights

  • Registry Mechanism Overhaul: The core change involves refactoring the registry mechanism across the codebase. Instead of using @register_module decorators on individual classes, registries are now initialized with a default_mapping dictionary that explicitly links module names to their full import paths. This centralizes the registration process and improves maintainability.
  • Removal of Decorators: Numerous @register_module decorators have been removed from classes throughout the trinity.algorithm, trinity.buffer, trinity.common.rewards, trinity.common.workflows, and trinity.utils.monitor modules. This simplifies class definitions and delegates registration logic to the __init__.py files of their respective packages.
  • Configuration File Updates: Configuration files (.yaml) that previously used short aliases for workflow types (e.g., learn2ask_workflow) have been updated to use the full import path (e.g., examples.learn_to_ask.workflow.workflow_learn2ask.Learn2AskWorkflow), aligning with the new explicit mapping scheme.
  • New Registry Test Suite: A comprehensive test file (tests/common/registry_test.py) has been added to validate the new registry mapping functionality. This ensures that all registered modules can be correctly retrieved and are instances of their expected base classes.
  • Registry Class Enhancement: The Registry class in trinity/utils/registry.py has been enhanced to accept a default_mapping dictionary during initialization, which it now uses to dynamically import modules when requested.
Using Gemini Code Assist

The full guide for Gemini Code Assist can be found on our documentation page, here are some quick tips.

Invoking Gemini

You can request assistance from Gemini at any point by creating a comment using either /gemini <command> or @gemini-code-assist <command>. Below is a summary of the supported commands on the current page.

Feature Command Description
Code Review /gemini review Performs a code review for the current pull request in its current state.
Pull Request Summary /gemini summary Provides a summary of the current pull request in its current state.
Comment @gemini-code-assist Responds in comments when explicitly tagged, both in pull request comments and review comments.
Help /gemini help Displays a list of available commands.

Customization

To customize Gemini Code Assist for GitHub experience, repository maintainers can create a configuration file and/or provide a custom code review style guide (such as PEP-8 for Python) by creating and adding files to a .gemini/ folder in the base of the repository. Detailed instructions can be found here.

Limitations & Feedback

Gemini Code Assist may make mistakes. Please leave feedback on any instances where its feedback is incorrect or counter productive. You can react with 👍 and 👎 on @gemini-code-assist comments. If you're interested in giving your feedback about your experience with Gemini Code Assist for Github and other Google products, sign up here.

You can also get AI-powered code generation, chat, as well as code reviews directly in the IDE at no cost with the Gemini Code Assist IDE Extension.

Footnotes

  1. Review the Privacy Notices, Generative AI Prohibited Use Policy, Terms of Service, and learn how to configure Gemini Code Assist in GitHub here. Gemini can make mistakes, so double check it and use code with caution.

Copy link
Contributor

@gemini-code-assist gemini-code-assist bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request introduces a significant and beneficial refactoring of the registry system, moving from a decorator-based approach to a centralized mapping scheme. This greatly improves maintainability and helps prevent circular import issues. The addition of comprehensive registry tests is also a valuable contribution. I've found one critical issue related to a circular import in the trinity/buffer/storage module, which I've detailed in the comments. Once that is addressed, this PR will be in great shape.

@pan-x-c
Copy link
Collaborator

pan-x-c commented Dec 16, 2025

/unittest-all

@github-actions
Copy link

Summary

Tests 📝 Passed ✅ Failed ❌ Skipped ⏭️ Other ❓ Flaky 🍂 Duration ⏱️
0 0 0 0 0 0 18.3s

Tests

Test Name Status Flaky Duration

Github Test Reporter by CTRF 💚

@hiyuchang
Copy link
Collaborator Author

/unittest-all

@github-actions
Copy link

Summary

Tests 📝 Passed ✅ Failed ❌ Skipped ⏭️ Other ❓ Flaky 🍂 Duration ⏱️
215 212 0 3 0 0 1h 23m

Skipped

Tests Status
tests/explorer/workflow_test.py::TestAgentScopeWorkflowAdapter::test_adapter skipped ⏭️
tests/trainer/trainer_test.py::TestMultiModalGRPO::test_trainer skipped ⏭️
tests/trainer/trainer_test.py::TestMultiModalSFT::test_trainer skipped ⏭️

Tests

Test Name Status Flaky Duration
tests/algorithm/advantage_fn_test.py::TestGroupedAdvantageFn::test_batch_level_std_grpo 41ms
tests/algorithm/advantage_fn_test.py::TestGroupedAdvantageFn::test_batch_level_step_wise_grpo_advantage 3ms
tests/algorithm/advantage_fn_test.py::TestGroupedAdvantageFn::test_duplicate_grpo 5ms
tests/algorithm/advantage_fn_test.py::TestGroupedAdvantageFn::test_grpo_advantage 3ms
tests/algorithm/advantage_fn_test.py::TestGroupedAdvantageFn::test_grpo_correct_bias 2ms
tests/algorithm/advantage_fn_test.py::TestGroupedAdvantageFn::test_grpo_reward_std 1ms
tests/algorithm/advantage_fn_test.py::TestGroupedAdvantageFn::test_step_wise_grpo_advantage 2ms
tests/algorithm/advantage_fn_test.py::TestGroupedAdvantageFn::test_step_wise_grpo_with_std_threshold 2ms
tests/algorithm/kl_fn_test.py::KLFnTest::test_abs_kl_fn 1ms
tests/algorithm/kl_fn_test.py::KLFnTest::test_corrected_k3_fallback 1ms
tests/algorithm/kl_fn_test.py::KLFnTest::test_corrected_k3_loss 1ms
tests/algorithm/kl_fn_test.py::KLFnTest::test_corrected_k3_same_policy 1ms
tests/algorithm/kl_fn_test.py::KLFnTest::test_corrected_k3_with_old_logprob 1ms
tests/algorithm/kl_fn_test.py::KLFnTest::test_dummy_kl_fn 1ms
tests/algorithm/kl_fn_test.py::KLFnTest::test_k1_kl_fn 1ms
tests/algorithm/kl_fn_test.py::KLFnTest::test_k2_kl_fn 1ms
tests/algorithm/kl_fn_test.py::KLFnTest::test_k3_kl_fn 1ms
tests/algorithm/kl_fn_test.py::KLFnTest::test_kl_loss_aggregation_modes 1ms
tests/algorithm/kl_fn_test.py::KLFnTest::test_low_var_kl_fn 1ms
tests/algorithm/policy_loss_test.py::VerlPolicyLossTest::test_dpo_policy_loss 1ms
tests/algorithm/policy_loss_test.py::VerlPolicyLossTest::test_gspo_policy_loss 1ms
tests/algorithm/policy_loss_test.py::VerlPolicyLossTest::test_mix_policy_loss 3ms
tests/algorithm/policy_loss_test.py::VerlPolicyLossTest::test_opmd_policy_loss 1ms
tests/algorithm/policy_loss_test.py::VerlPolicyLossTest::test_ppo_policy_loss 1ms
tests/algorithm/policy_loss_test.py::VerlPolicyLossTest::test_ppo_policy_loss_with_sequence_masking 1ms
tests/algorithm/policy_loss_test.py::VerlPolicyLossTest::test_sapo_policy_loss 1ms
tests/algorithm/policy_loss_test.py::VerlPolicyLossTest::test_sft_policy_loss 1ms
tests/buffer/experience_pipeline_test.py::TestExperiencePipeline::test_experience_pipeline 24.6s
tests/buffer/experience_pipeline_test.py::TestExperiencePipeline::test_pass_rate_calculation 15.6s
tests/buffer/experience_storage_test.py::ExperienceStorageTest::test_sql_experience_buffer 4.7s
tests/buffer/experience_storage_test.py::ExperienceStorageTest::test_sql_storage_0_sft 5.9s
tests/buffer/experience_storage_test.py::ExperienceStorageTest::test_sql_storage_1_dpo 6.6s
tests/buffer/file_test.py::TestFileBuffer::test_file_reader 460ms
tests/buffer/file_test.py::TestFileBuffer::test_file_writer 4.2s
tests/buffer/formatter_test.py::TestFormatter::test_dpo_messages_formatter 550ms
tests/buffer/formatter_test.py::TestFormatter::test_dpo_plaintext_formatter 497ms
tests/buffer/formatter_test.py::TestFormatter::test_multi_modal_sft_formatter 912ms
tests/buffer/formatter_test.py::TestFormatter::test_sft_messages_formatter 995ms
tests/buffer/formatter_test.py::TestFormatter::test_sft_plaintext_formatter 733ms
tests/buffer/formatter_test.py::TestFormatter::test_task_formatter 225ms
tests/buffer/queue_test.py::TestQueueBuffer::test_priority_queue_buffer_reuse 8.9s
tests/buffer/queue_test.py::TestQueueBuffer::test_priority_queue_capacity 4.8s
tests/buffer/queue_test.py::TestQueueBuffer::test_priority_queue_reuse_count_control 7.0s
tests/buffer/queue_test.py::TestQueueBuffer::test_queue_buffer_0_queue 5.7s
tests/buffer/queue_test.py::TestQueueBuffer::test_queue_buffer_1_priority_queue 5.9s
tests/buffer/queue_test.py::TestQueueBuffer::test_queue_buffer_capacity 6.4s
tests/buffer/reader_test.py::TestBufferReader::test_buffer_reader_registration 614ms
tests/buffer/reward_shaping_mapper_test.py::TestRewardShapingMapper::test_basic_usage 7ms
tests/buffer/sql_test.py::TestSQLBuffer::test_sql_exp_buffer_read_write 4.1s
tests/buffer/sql_test.py::TestSQLBuffer::test_sql_task_buffer_read_write 4.6s
tests/buffer/task_scheduler_test.py::TestTaskScheduler::test_task_scheduler_0 91ms
tests/buffer/task_scheduler_test.py::TestTaskScheduler::test_task_scheduler_1 72ms
tests/buffer/task_scheduler_test.py::TestTaskScheduler::test_task_scheduler_2 112ms
tests/buffer/task_scheduler_test.py::TestTaskScheduler::test_task_scheduler_3 113ms
tests/buffer/task_scheduler_test.py::TestTaskScheduler::test_task_scheduler_4 112ms
tests/buffer/task_scheduler_test.py::TestTaskScheduler::test_task_scheduler_5 117ms
tests/buffer/task_scheduler_test.py::TestTaskScheduler::test_task_scheduler_6 131ms
tests/buffer/task_scheduler_test.py::TestTaskScheduler::test_task_scheduler_simple 59ms
tests/buffer/task_storage_test.py::TaskStorageTest::test_read_task_0_file 72ms
tests/buffer/task_storage_test.py::TaskStorageTest::test_read_task_1_sql 3.8s
tests/buffer/task_storage_test.py::TaskStorageTest::test_read_task_2_file 53ms
tests/buffer/task_storage_test.py::TaskStorageTest::test_read_task_3_sql 4.1s
tests/buffer/task_storage_test.py::TaskStorageTest::test_read_task_4_file 51ms
tests/buffer/task_storage_test.py::TaskStorageTest::test_read_task_5_sql 4.4s
tests/cli/launcher_test.py::TestLauncherMain::test_debug_mode 49.0s
tests/cli/launcher_test.py::TestLauncherMain::test_main_run_command 6.6s
tests/cli/launcher_test.py::TestLauncherMain::test_main_run_in_dlc 1.9s
tests/cli/launcher_test.py::TestLauncherMain::test_main_studio_command 327ms
tests/cli/launcher_test.py::TestLauncherMain::test_multi_stage_run 1.7s
tests/common/config_test.py::TestConfig::test_all_examples_are_valid 34.1s
tests/common/config_test.py::TestConfig::test_chat_template_path 95ms
tests/common/config_test.py::TestConfig::test_config_flatten 42ms
tests/common/config_test.py::TestConfig::test_continue_from_checkpoint_is_valid 196ms
tests/common/config_test.py::TestConfig::test_default_workflow 93ms
tests/common/config_test.py::TestConfig::test_load_default_config 3.4s
tests/common/config_test.py::TestConfig::test_max_token_len_per_gpu_set_correctly 95ms
tests/common/config_test.py::TestConfig::test_optimizer_config_propagation 93ms
tests/common/config_test.py::TestConfig::test_update_config_from_ray_cluster 357ms
tests/common/experience_test.py::TestEID::test_eid_properties 1ms
tests/common/experience_test.py::TestExperience::test_action_mask_and_logprobs_type 1ms
tests/common/experience_test.py::TestExperience::test_assertions 1ms
tests/common/experience_test.py::TestExperience::test_dpo_experience 1ms
tests/common/experience_test.py::TestExperience::test_gather 1ms
tests/common/experience_test.py::TestExperience::test_gather_with_token_level_reward 1ms
tests/common/experience_test.py::TestExperience::test_hf_datasets_conversion 16ms
tests/common/experience_test.py::TestExperience::test_multi_turn_experience 1ms
tests/common/experience_test.py::TestExperience::test_serialize_deserialize 1ms
tests/common/experience_test.py::TestExperience::test_single_turn_experience 1ms
tests/common/experience_test.py::TestExperience::test_to_dict 1ms
tests/common/experience_test.py::TestExperienceConversion::test_batch_conversion 1ms
tests/common/experience_test.py::TestExperienceConversion::test_dpo_experience_batch_conversion 1ms
tests/common/experience_test.py::TestExperienceConversion::test_experience_model_experience_conversion 1ms
tests/common/experience_test.py::TestExperienceConversion::test_gather_experiences_with_custom_fields 1ms
tests/common/experience_test.py::TestExperienceConversion::test_multiturn_experience_batch_converstion 1ms
tests/common/vllm_test.py::ModelWrapperTest_0::test_generate 53.2s
tests/common/vllm_test.py::ModelWrapperTest_1::test_generate 32.1s
tests/common/vllm_test.py::ModelWrapperTest_2::test_generate 43.2s
tests/common/vllm_test.py::TestModelLen_0::test_model_len 15.9s
tests/common/vllm_test.py::TestModelLen_1::test_model_len 16.0s
tests/common/vllm_test.py::TestModelLen_2::test_model_len 15.4s
tests/common/vllm_test.py::TestModelLenWithoutPromptTruncation::test_model_len 15.4s
tests/common/vllm_test.py::TestAPIServer::test_api 21.1s
tests/common/vllm_test.py::TestLogprobs::test_logprobs_api 16.2s
tests/common/vllm_test.py::TestAsyncAPIServer::test_api_async 21.0s
tests/common/vllm_test.py::TestTokenizer::test_action_mask 251ms
tests/common/vllm_test.py::TestTokenizer::test_action_mask_with_tools 240ms
tests/common/vllm_test.py::TestAPIServerToolCall_0_deepseek_r1::test_api_tool_calls 18.3s
tests/common/vllm_test.py::TestAPIServerToolCall_1::test_api_tool_calls 16.0s
tests/common/vllm_test.py::TestSuperLongGeneration::test_generate 3m 37s
tests/explorer/explorer_test.py::TestExplorerCountdownEval::test_explorer 1m 13s
tests/explorer/explorer_test.py::TestExplorerGSM8KRULERNoEval::test_explorer 1m 44s
tests/explorer/explorer_test.py::TestExplorerGSM8k::test_explorer 3m 37s
tests/explorer/explorer_test.py::ServeTest::test_serve 1m 21s
tests/explorer/scheduler_test.py::SchedulerTest::test_async_workflow 12.7s
tests/explorer/scheduler_test.py::SchedulerTest::test_concurrent_operations 12.3s
tests/explorer/scheduler_test.py::SchedulerTest::test_dynamic_timeout 20.4s
tests/explorer/scheduler_test.py::SchedulerTest::test_get_results 27.5s
tests/explorer/scheduler_test.py::SchedulerTest::test_metric_calculation_with_non_repeatable_workflow_0 12.2s
tests/explorer/scheduler_test.py::SchedulerTest::test_metric_calculation_with_non_repeatable_workflow_1 12.8s
tests/explorer/scheduler_test.py::SchedulerTest::test_metric_calculation_with_repeatable_workflow_0 12.5s
tests/explorer/scheduler_test.py::SchedulerTest::test_metric_calculation_with_repeatable_workflow_1 12.6s
tests/explorer/scheduler_test.py::SchedulerTest::test_multi_step_execution 12.7s
tests/explorer/scheduler_test.py::SchedulerTest::test_non_repeatable_workflow 12.7s
tests/explorer/scheduler_test.py::SchedulerTest::test_over_rollout_min_wait 16.3s
tests/explorer/scheduler_test.py::SchedulerTest::test_scheduler_all_methods 22.2s
tests/explorer/scheduler_test.py::SchedulerTest::test_scheduler_restart_after_stop 23.4s
tests/explorer/scheduler_test.py::SchedulerTest::test_split_tasks 15.5s
tests/explorer/scheduler_test.py::SchedulerTest::test_stepwise_experience_eid 32.4s
tests/explorer/scheduler_test.py::SchedulerTest::test_wait_all 15.5s
tests/explorer/scheduler_test.py::SchedulerTest::test_wait_all_timeout_with_multi_batch 21.2s
tests/explorer/scheduler_test.py::TestRunnerStateCollection::test_runner_state_collection 17.4s
tests/explorer/step_wise_workflow_test.py::WorkflowTest::test_reward_propagation_workflow_0 2ms
tests/explorer/step_wise_workflow_test.py::WorkflowTest::test_reward_propagation_workflow_1 603ms
tests/explorer/step_wise_workflow_test.py::WorkflowTest::test_step_wise_reward_workflow_0 1ms
tests/explorer/step_wise_workflow_test.py::WorkflowTest::test_step_wise_reward_workflow_1 1.0s
tests/explorer/step_wise_workflow_test.py::WorkflowTest::test_workflows_raise_error 1ms
tests/explorer/step_wise_workflow_test.py::WorkflowTest::test_workflows_stop_at_max_env_steps 1.0s
tests/explorer/workflow_test.py::WorkflowTest::test_gsm8k_workflow 16ms
tests/explorer/workflow_test.py::WorkflowTest::test_math_boxed_workflow 24ms
tests/explorer/workflow_test.py::WorkflowTest::test_math_complex_workflow 266ms
tests/explorer/workflow_test.py::WorkflowTest::test_math_eval_workflow 4ms
tests/explorer/workflow_test.py::WorkflowTest::test_math_fraction_workflow 16ms
tests/explorer/workflow_test.py::WorkflowTest::test_math_workflow 10ms
tests/explorer/workflow_test.py::WorkflowTest::test_rm_gallery_workflow 114ms
tests/explorer/workflow_test.py::WorkflowTest::test_workflow_repeatable_0 1ms
tests/explorer/workflow_test.py::WorkflowTest::test_workflow_repeatable_1 101ms
tests/explorer/workflow_test.py::WorkflowTest::test_workflow_resettable_0 1ms
tests/explorer/workflow_test.py::WorkflowTest::test_workflow_resettable_1 201ms
tests/explorer/workflow_test.py::MultiTurnWorkflowTest_0::test_multi_turn_workflow 14.4s
tests/explorer/workflow_test.py::MultiTurnWorkflowTest_1::test_multi_turn_workflow 14.7s
tests/explorer/workflow_test.py::TestWorkflowStateRecording::test_workflow_state_recording 4.0s
tests/explorer/workflow_test.py::TestAgentScopeWorkflowAdapter::test_adapter ⏭️ 2ms
tests/explorer/workflow_test.py::TestWorkflowRunner::test_workflow_runner 305ms
tests/explorer/workflow_test.py::TestWorkflowRunner::test_workflow_runner_get_state 8.1s
tests/explorer/workflow_test.py::TestWorkflowRunner::test_workflow_with_openai 17.3s
tests/manager/synchronizer_test.py::TestSynchronizerExit::test_synchronizer 1m
tests/manager/synchronizer_test.py::TestStateDictBasedSynchronizer_0::test_synchronizer 1m 44s
tests/manager/synchronizer_test.py::TestStateDictBasedSynchronizer_1::test_synchronizer 1m 43s
tests/manager/synchronizer_test.py::TestStateDictBasedSynchronizer_2::test_synchronizer 2m 28s
tests/manager/synchronizer_test.py::TestStateDictBasedSynchronizer_3::test_synchronizer 2m 31s
tests/manager/synchronizer_test.py::TestNCCLBasedSynchronizer_0::test_synchronizer 1m 40s
tests/manager/synchronizer_test.py::TestNCCLBasedSynchronizer_1::test_synchronizer 1m 39s
tests/service/data_juicer_test.py::TestDataJuicer::test_config 2.2s
tests/service/data_juicer_test.py::TestDataJuicer::test_server_start 21.6s
tests/service/data_juicer_test.py::TestDataJuicerExperiencePipeline::test_data_juicer_operators 31.3s
tests/service/data_juicer_test.py::TestDataJuicerTaskPipeline::test_data_juicer_task_pipeline 14.3s
tests/trainer/trainer_test.py::TestTrainerCountdown_0_fsdp::test_trainer 3m 11s
tests/trainer/trainer_test.py::TestTrainerCountdown_1_megatron::test_trainer 4m 39s
tests/trainer/trainer_test.py::TestStepAheadAsyncRL::test_trainer 1m 19s
tests/trainer/trainer_test.py::TestTrainerGSM8K_0_fsdp::test_trainer 1m 16s
tests/trainer/trainer_test.py::TestTrainerGSM8K_1_fsdp2::test_trainer 1m 16s
tests/trainer/trainer_test.py::TestTrainerGSM8K_2_fsdp::test_trainer 1m 15s
tests/trainer/trainer_test.py::TestTrainerGSM8K_3_fsdp2::test_trainer 1m 28s
tests/trainer/trainer_test.py::TestTrainerSFTWarmupGSM8K::test_trainer 2m 22s
tests/trainer/trainer_test.py::TestTrainerDPO::test_trainer 58.1s
tests/trainer/trainer_test.py::TestTrainerSFT::test_trainer 54.7s
tests/trainer/trainer_test.py::TestTrainerToolsSFT::test_trainer_tools 54.9s
tests/trainer/trainer_test.py::TestFullyAsyncMode_0_fsdp::test_fully_async_mode 1m 50s
tests/trainer/trainer_test.py::TestFullyAsyncMode_1_fsdp::test_fully_async_mode 1m 56s
tests/trainer/trainer_test.py::TestFullyAsyncMode_2_megatron::test_fully_async_mode 2m 39s
tests/trainer/trainer_test.py::TestTrainerCheckpointSave_0_fsdp::test_trainer 2m 24s
tests/trainer/trainer_test.py::TestTrainerCheckpointSave_1_megatron::test_trainer 4m 17s
tests/trainer/trainer_test.py::TestTrainerMIX::test_trainer 2m 22s
tests/trainer/trainer_test.py::TestMultiModalGRPO::test_trainer ⏭️ 860ms
tests/trainer/trainer_test.py::TestMultiModalSFT::test_trainer ⏭️ 808ms
tests/trainer/trainer_test.py::TestTrainerLoRA::test_trainer 3m 45s
tests/trainer/trainer_test.py::TestOverRollout::test_trainer 1m 15s
tests/trainer/trainer_test.py::TestTrainerPromptTruncation::test_trainer 1m 5s
tests/utils/eval_utils_test.py::TestComputeScore::test_both_boxed_and_equivalent 15ms
tests/utils/eval_utils_test.py::TestComputeScore::test_both_boxed_and_not_equivalent 1ms
tests/utils/eval_utils_test.py::TestComputeScore::test_empty_ground_truth 2ms
tests/utils/eval_utils_test.py::TestComputeScore::test_empty_solution_string 1ms
tests/utils/eval_utils_test.py::TestComputeScore::test_multiple_boxed_answers_in_solution 2ms
tests/utils/eval_utils_test.py::TestComputeScore::test_solution_boxed_truth_raw_and_equivalent 1ms
tests/utils/eval_utils_test.py::TestComputeScore::test_solution_boxed_truth_raw_and_not_equivalent 1ms
tests/utils/eval_utils_test.py::TestComputeScore::test_solution_not_boxed 1ms
tests/utils/eval_utils_test.py::TestComputeScore::test_solution_raw_and_ground_truth_boxed_equivalent 1ms
tests/utils/eval_utils_test.py::TestMathEvalUtils::test_extract_answer 4ms
tests/utils/eval_utils_test.py::TestMathEvalUtils::test_verify_math_answer 74ms
tests/utils/eval_utils_test.py::TestEvalUtils::test_is_equiv 6ms
tests/utils/log_test.py::LogTest::test_actor_log 5.1s
tests/utils/log_test.py::LogTest::test_group_by_node 4.9s
tests/utils/log_test.py::LogTest::test_no_actor_log 909ms
tests/utils/plugin_test.py::TestPluginLoader::test_load_plugins_local_0__workspace_tests_utils_plugins 100ms
tests/utils/plugin_test.py::TestPluginLoader::test_load_plugins_local_1_tests_utils_plugins 96ms
tests/utils/plugin_test.py::TestPluginLoader::test_load_plugins_remote_0__workspace_tests_utils_plugins 21.8s
tests/utils/plugin_test.py::TestPluginLoader::test_load_plugins_remote_1_tests_utils_plugins 21.9s
tests/utils/plugin_test.py::TestPluginLoader::test_passing_custom_class_0__workspace_tests_utils_plugins 11.9s
tests/utils/plugin_test.py::TestPluginLoader::test_passing_custom_class_1_tests_utils_plugins 11.5s
tests/utils/registry_test.py::TestRegistryWithRay::test_dynamic_import 11.0s
tests/utils/registry_test.py::TestRegistry::test_algorithm_registry_mapping 2ms
tests/utils/registry_test.py::TestRegistry::test_buffer_module_registry_mapping 1ms
tests/utils/registry_test.py::TestRegistry::test_common_module_registry_mapping 52ms
tests/utils/registry_test.py::TestRegistry::test_register_module 1ms
tests/utils/registry_test.py::TestRegistry::test_utils_module_registry_mapping 1ms

Github Test Reporter by CTRF 💚

@pan-x-c
Copy link
Collaborator

pan-x-c commented Dec 17, 2025

/gemini review

Copy link
Contributor

@gemini-code-assist gemini-code-assist bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request introduces a significant and beneficial refactoring of the module registration system. It moves from a decorator-based approach to a centralized mapping within __init__.py files for each registry. This improves code organization, reduces import side-effects, and makes the registration process more explicit and easier to track. The changes are consistently applied across the entire codebase, including the core library, examples, and documentation. A comprehensive new test suite for the registry functionality has also been added, which is excellent. I've found a couple of minor typos in the documentation and test files, which I've commented on. Overall, this is a high-quality pull request that improves the maintainability of the framework.

@hiyuchang
Copy link
Collaborator Author

/unittest-all

@github-actions
Copy link

Summary

Tests 📝 Passed ✅ Failed ❌ Skipped ⏭️ Other ❓ Flaky 🍂 Duration ⏱️
215 212 0 3 0 0 1h 15m

Skipped

Tests Status
tests/explorer/workflow_test.py::TestAgentScopeWorkflowAdapter::test_adapter skipped ⏭️
tests/trainer/trainer_test.py::TestMultiModalGRPO::test_trainer skipped ⏭️
tests/trainer/trainer_test.py::TestMultiModalSFT::test_trainer skipped ⏭️

Tests

Test Name Status Flaky Duration
tests/algorithm/advantage_fn_test.py::TestGroupedAdvantageFn::test_batch_level_std_grpo 42ms
tests/algorithm/advantage_fn_test.py::TestGroupedAdvantageFn::test_batch_level_step_wise_grpo_advantage 3ms
tests/algorithm/advantage_fn_test.py::TestGroupedAdvantageFn::test_duplicate_grpo 5ms
tests/algorithm/advantage_fn_test.py::TestGroupedAdvantageFn::test_grpo_advantage 3ms
tests/algorithm/advantage_fn_test.py::TestGroupedAdvantageFn::test_grpo_correct_bias 2ms
tests/algorithm/advantage_fn_test.py::TestGroupedAdvantageFn::test_grpo_reward_std 1ms
tests/algorithm/advantage_fn_test.py::TestGroupedAdvantageFn::test_step_wise_grpo_advantage 2ms
tests/algorithm/advantage_fn_test.py::TestGroupedAdvantageFn::test_step_wise_grpo_with_std_threshold 2ms
tests/algorithm/kl_fn_test.py::KLFnTest::test_abs_kl_fn 1ms
tests/algorithm/kl_fn_test.py::KLFnTest::test_corrected_k3_fallback 1ms
tests/algorithm/kl_fn_test.py::KLFnTest::test_corrected_k3_loss 1ms
tests/algorithm/kl_fn_test.py::KLFnTest::test_corrected_k3_same_policy 1ms
tests/algorithm/kl_fn_test.py::KLFnTest::test_corrected_k3_with_old_logprob 1ms
tests/algorithm/kl_fn_test.py::KLFnTest::test_dummy_kl_fn 1ms
tests/algorithm/kl_fn_test.py::KLFnTest::test_k1_kl_fn 1ms
tests/algorithm/kl_fn_test.py::KLFnTest::test_k2_kl_fn 1ms
tests/algorithm/kl_fn_test.py::KLFnTest::test_k3_kl_fn 1ms
tests/algorithm/kl_fn_test.py::KLFnTest::test_kl_loss_aggregation_modes 1ms
tests/algorithm/kl_fn_test.py::KLFnTest::test_low_var_kl_fn 1ms
tests/algorithm/policy_loss_test.py::VerlPolicyLossTest::test_dpo_policy_loss 2ms
tests/algorithm/policy_loss_test.py::VerlPolicyLossTest::test_gspo_policy_loss 1ms
tests/algorithm/policy_loss_test.py::VerlPolicyLossTest::test_mix_policy_loss 3ms
tests/algorithm/policy_loss_test.py::VerlPolicyLossTest::test_opmd_policy_loss 1ms
tests/algorithm/policy_loss_test.py::VerlPolicyLossTest::test_ppo_policy_loss 1ms
tests/algorithm/policy_loss_test.py::VerlPolicyLossTest::test_ppo_policy_loss_with_sequence_masking 1ms
tests/algorithm/policy_loss_test.py::VerlPolicyLossTest::test_sapo_policy_loss 1ms
tests/algorithm/policy_loss_test.py::VerlPolicyLossTest::test_sft_policy_loss 1ms
tests/buffer/experience_pipeline_test.py::TestExperiencePipeline::test_experience_pipeline 13.3s
tests/buffer/experience_pipeline_test.py::TestExperiencePipeline::test_pass_rate_calculation 7.1s
tests/buffer/experience_storage_test.py::ExperienceStorageTest::test_sql_experience_buffer 5.1s
tests/buffer/experience_storage_test.py::ExperienceStorageTest::test_sql_storage_0_sft 5.0s
tests/buffer/experience_storage_test.py::ExperienceStorageTest::test_sql_storage_1_dpo 5.6s
tests/buffer/file_test.py::TestFileBuffer::test_file_reader 459ms
tests/buffer/file_test.py::TestFileBuffer::test_file_writer 1.9s
tests/buffer/formatter_test.py::TestFormatter::test_dpo_messages_formatter 531ms
tests/buffer/formatter_test.py::TestFormatter::test_dpo_plaintext_formatter 452ms
tests/buffer/formatter_test.py::TestFormatter::test_multi_modal_sft_formatter 1.4s
tests/buffer/formatter_test.py::TestFormatter::test_sft_messages_formatter 969ms
tests/buffer/formatter_test.py::TestFormatter::test_sft_plaintext_formatter 712ms
tests/buffer/formatter_test.py::TestFormatter::test_task_formatter 221ms
tests/buffer/queue_test.py::TestQueueBuffer::test_priority_queue_buffer_reuse 6.4s
tests/buffer/queue_test.py::TestQueueBuffer::test_priority_queue_capacity 2.4s
tests/buffer/queue_test.py::TestQueueBuffer::test_priority_queue_reuse_count_control 4.4s
tests/buffer/queue_test.py::TestQueueBuffer::test_queue_buffer_0_queue 3.6s
tests/buffer/queue_test.py::TestQueueBuffer::test_queue_buffer_1_priority_queue 3.4s
tests/buffer/queue_test.py::TestQueueBuffer::test_queue_buffer_capacity 4.4s
tests/buffer/reader_test.py::TestBufferReader::test_buffer_reader_registration 814ms
tests/buffer/reward_shaping_mapper_test.py::TestRewardShapingMapper::test_basic_usage 8ms
tests/buffer/sql_test.py::TestSQLBuffer::test_sql_exp_buffer_read_write 2.8s
tests/buffer/sql_test.py::TestSQLBuffer::test_sql_task_buffer_read_write 3.2s
tests/buffer/task_scheduler_test.py::TestTaskScheduler::test_task_scheduler_0 97ms
tests/buffer/task_scheduler_test.py::TestTaskScheduler::test_task_scheduler_1 72ms
tests/buffer/task_scheduler_test.py::TestTaskScheduler::test_task_scheduler_2 111ms
tests/buffer/task_scheduler_test.py::TestTaskScheduler::test_task_scheduler_3 111ms
tests/buffer/task_scheduler_test.py::TestTaskScheduler::test_task_scheduler_4 112ms
tests/buffer/task_scheduler_test.py::TestTaskScheduler::test_task_scheduler_5 117ms
tests/buffer/task_scheduler_test.py::TestTaskScheduler::test_task_scheduler_6 132ms
tests/buffer/task_scheduler_test.py::TestTaskScheduler::test_task_scheduler_simple 59ms
tests/buffer/task_storage_test.py::TaskStorageTest::test_read_task_0_file 73ms
tests/buffer/task_storage_test.py::TaskStorageTest::test_read_task_1_sql 3.0s
tests/buffer/task_storage_test.py::TaskStorageTest::test_read_task_2_file 52ms
tests/buffer/task_storage_test.py::TaskStorageTest::test_read_task_3_sql 2.9s
tests/buffer/task_storage_test.py::TaskStorageTest::test_read_task_4_file 52ms
tests/buffer/task_storage_test.py::TaskStorageTest::test_read_task_5_sql 3.6s
tests/cli/launcher_test.py::TestLauncherMain::test_debug_mode 48.3s
tests/cli/launcher_test.py::TestLauncherMain::test_main_run_command 6.7s
tests/cli/launcher_test.py::TestLauncherMain::test_main_run_in_dlc 1.4s
tests/cli/launcher_test.py::TestLauncherMain::test_main_studio_command 328ms
tests/cli/launcher_test.py::TestLauncherMain::test_multi_stage_run 1.7s
tests/common/config_test.py::TestConfig::test_all_examples_are_valid 35.2s
tests/common/config_test.py::TestConfig::test_chat_template_path 95ms
tests/common/config_test.py::TestConfig::test_config_flatten 42ms
tests/common/config_test.py::TestConfig::test_continue_from_checkpoint_is_valid 193ms
tests/common/config_test.py::TestConfig::test_default_workflow 92ms
tests/common/config_test.py::TestConfig::test_load_default_config 3.2s
tests/common/config_test.py::TestConfig::test_max_token_len_per_gpu_set_correctly 99ms
tests/common/config_test.py::TestConfig::test_optimizer_config_propagation 95ms
tests/common/config_test.py::TestConfig::test_update_config_from_ray_cluster 361ms
tests/common/experience_test.py::TestEID::test_eid_properties 1ms
tests/common/experience_test.py::TestExperience::test_action_mask_and_logprobs_type 1ms
tests/common/experience_test.py::TestExperience::test_assertions 1ms
tests/common/experience_test.py::TestExperience::test_dpo_experience 1ms
tests/common/experience_test.py::TestExperience::test_gather 1ms
tests/common/experience_test.py::TestExperience::test_gather_with_token_level_reward 1ms
tests/common/experience_test.py::TestExperience::test_hf_datasets_conversion 16ms
tests/common/experience_test.py::TestExperience::test_multi_turn_experience 1ms
tests/common/experience_test.py::TestExperience::test_serialize_deserialize 1ms
tests/common/experience_test.py::TestExperience::test_single_turn_experience 1ms
tests/common/experience_test.py::TestExperience::test_to_dict 1ms
tests/common/experience_test.py::TestExperienceConversion::test_batch_conversion 1ms
tests/common/experience_test.py::TestExperienceConversion::test_dpo_experience_batch_conversion 1ms
tests/common/experience_test.py::TestExperienceConversion::test_experience_model_experience_conversion 1ms
tests/common/experience_test.py::TestExperienceConversion::test_gather_experiences_with_custom_fields 1ms
tests/common/experience_test.py::TestExperienceConversion::test_multiturn_experience_batch_converstion 1ms
tests/common/vllm_test.py::ModelWrapperTest_0::test_generate 51.5s
tests/common/vllm_test.py::ModelWrapperTest_1::test_generate 32.2s
tests/common/vllm_test.py::ModelWrapperTest_2::test_generate 41.7s
tests/common/vllm_test.py::TestModelLen_0::test_model_len 15.2s
tests/common/vllm_test.py::TestModelLen_1::test_model_len 15.5s
tests/common/vllm_test.py::TestModelLen_2::test_model_len 15.4s
tests/common/vllm_test.py::TestModelLenWithoutPromptTruncation::test_model_len 15.6s
tests/common/vllm_test.py::TestAPIServer::test_api 21.1s
tests/common/vllm_test.py::TestLogprobs::test_logprobs_api 15.5s
tests/common/vllm_test.py::TestAsyncAPIServer::test_api_async 21.1s
tests/common/vllm_test.py::TestTokenizer::test_action_mask 236ms
tests/common/vllm_test.py::TestTokenizer::test_action_mask_with_tools 237ms
tests/common/vllm_test.py::TestAPIServerToolCall_0_deepseek_r1::test_api_tool_calls 17.2s
tests/common/vllm_test.py::TestAPIServerToolCall_1::test_api_tool_calls 15.2s
tests/common/vllm_test.py::TestSuperLongGeneration::test_generate 4m 27s
tests/explorer/explorer_test.py::TestExplorerCountdownEval::test_explorer 1m 15s
tests/explorer/explorer_test.py::TestExplorerGSM8KRULERNoEval::test_explorer 1m 30s
tests/explorer/explorer_test.py::TestExplorerGSM8k::test_explorer 3m 33s
tests/explorer/explorer_test.py::ServeTest::test_serve 1m 14s
tests/explorer/scheduler_test.py::SchedulerTest::test_async_workflow 9.2s
tests/explorer/scheduler_test.py::SchedulerTest::test_concurrent_operations 8.5s
tests/explorer/scheduler_test.py::SchedulerTest::test_dynamic_timeout 16.9s
tests/explorer/scheduler_test.py::SchedulerTest::test_get_results 24.3s
tests/explorer/scheduler_test.py::SchedulerTest::test_metric_calculation_with_non_repeatable_workflow_0 9.1s
tests/explorer/scheduler_test.py::SchedulerTest::test_metric_calculation_with_non_repeatable_workflow_1 8.9s
tests/explorer/scheduler_test.py::SchedulerTest::test_metric_calculation_with_repeatable_workflow_0 8.9s
tests/explorer/scheduler_test.py::SchedulerTest::test_metric_calculation_with_repeatable_workflow_1 8.9s
tests/explorer/scheduler_test.py::SchedulerTest::test_multi_step_execution 9.2s
tests/explorer/scheduler_test.py::SchedulerTest::test_non_repeatable_workflow 9.1s
tests/explorer/scheduler_test.py::SchedulerTest::test_over_rollout_min_wait 13.0s
tests/explorer/scheduler_test.py::SchedulerTest::test_scheduler_all_methods 18.9s
tests/explorer/scheduler_test.py::SchedulerTest::test_scheduler_restart_after_stop 17.1s
tests/explorer/scheduler_test.py::SchedulerTest::test_split_tasks 12.1s
tests/explorer/scheduler_test.py::SchedulerTest::test_stepwise_experience_eid 29.3s
tests/explorer/scheduler_test.py::SchedulerTest::test_wait_all 12.0s
tests/explorer/scheduler_test.py::SchedulerTest::test_wait_all_timeout_with_multi_batch 17.9s
tests/explorer/scheduler_test.py::TestRunnerStateCollection::test_runner_state_collection 13.9s
tests/explorer/step_wise_workflow_test.py::WorkflowTest::test_reward_propagation_workflow_0 2ms
tests/explorer/step_wise_workflow_test.py::WorkflowTest::test_reward_propagation_workflow_1 602ms
tests/explorer/step_wise_workflow_test.py::WorkflowTest::test_step_wise_reward_workflow_0 1ms
tests/explorer/step_wise_workflow_test.py::WorkflowTest::test_step_wise_reward_workflow_1 1.0s
tests/explorer/step_wise_workflow_test.py::WorkflowTest::test_workflows_raise_error 1ms
tests/explorer/step_wise_workflow_test.py::WorkflowTest::test_workflows_stop_at_max_env_steps 1.0s
tests/explorer/workflow_test.py::WorkflowTest::test_gsm8k_workflow 16ms
tests/explorer/workflow_test.py::WorkflowTest::test_math_boxed_workflow 26ms
tests/explorer/workflow_test.py::WorkflowTest::test_math_complex_workflow 784ms
tests/explorer/workflow_test.py::WorkflowTest::test_math_eval_workflow 4ms
tests/explorer/workflow_test.py::WorkflowTest::test_math_fraction_workflow 17ms
tests/explorer/workflow_test.py::WorkflowTest::test_math_workflow 10ms
tests/explorer/workflow_test.py::WorkflowTest::test_rm_gallery_workflow 112ms
tests/explorer/workflow_test.py::WorkflowTest::test_workflow_repeatable_0 1ms
tests/explorer/workflow_test.py::WorkflowTest::test_workflow_repeatable_1 101ms
tests/explorer/workflow_test.py::WorkflowTest::test_workflow_resettable_0 1ms
tests/explorer/workflow_test.py::WorkflowTest::test_workflow_resettable_1 202ms
tests/explorer/workflow_test.py::MultiTurnWorkflowTest_0::test_multi_turn_workflow 14.4s
tests/explorer/workflow_test.py::MultiTurnWorkflowTest_1::test_multi_turn_workflow 14.6s
tests/explorer/workflow_test.py::TestWorkflowStateRecording::test_workflow_state_recording 4.0s
tests/explorer/workflow_test.py::TestAgentScopeWorkflowAdapter::test_adapter ⏭️ 2ms
tests/explorer/workflow_test.py::TestWorkflowRunner::test_workflow_runner 301ms
tests/explorer/workflow_test.py::TestWorkflowRunner::test_workflow_runner_get_state 8.1s
tests/explorer/workflow_test.py::TestWorkflowRunner::test_workflow_with_openai 17.5s
tests/manager/synchronizer_test.py::TestSynchronizerExit::test_synchronizer 42.5s
tests/manager/synchronizer_test.py::TestStateDictBasedSynchronizer_0::test_synchronizer 1m 26s
tests/manager/synchronizer_test.py::TestStateDictBasedSynchronizer_1::test_synchronizer 1m 29s
tests/manager/synchronizer_test.py::TestStateDictBasedSynchronizer_2::test_synchronizer 2m 12s
tests/manager/synchronizer_test.py::TestStateDictBasedSynchronizer_3::test_synchronizer 2m 17s
tests/manager/synchronizer_test.py::TestNCCLBasedSynchronizer_0::test_synchronizer 1m 24s
tests/manager/synchronizer_test.py::TestNCCLBasedSynchronizer_1::test_synchronizer 1m 24s
tests/service/data_juicer_test.py::TestDataJuicer::test_config 1.7s
tests/service/data_juicer_test.py::TestDataJuicer::test_server_start 21.9s
tests/service/data_juicer_test.py::TestDataJuicerExperiencePipeline::test_data_juicer_operators 27.4s
tests/service/data_juicer_test.py::TestDataJuicerTaskPipeline::test_data_juicer_task_pipeline 14.3s
tests/trainer/trainer_test.py::TestTrainerCountdown_0_fsdp::test_trainer 2m 51s
tests/trainer/trainer_test.py::TestTrainerCountdown_1_megatron::test_trainer 4m 34s
tests/trainer/trainer_test.py::TestStepAheadAsyncRL::test_trainer 1m 19s
tests/trainer/trainer_test.py::TestTrainerGSM8K_0_fsdp::test_trainer 1m 4s
tests/trainer/trainer_test.py::TestTrainerGSM8K_1_fsdp2::test_trainer 1m 6s
tests/trainer/trainer_test.py::TestTrainerGSM8K_2_fsdp::test_trainer 1m 9s
tests/trainer/trainer_test.py::TestTrainerGSM8K_3_fsdp2::test_trainer 1m 18s
tests/trainer/trainer_test.py::TestTrainerSFTWarmupGSM8K::test_trainer 2m 6s
tests/trainer/trainer_test.py::TestTrainerDPO::test_trainer 48.7s
tests/trainer/trainer_test.py::TestTrainerSFT::test_trainer 46.4s
tests/trainer/trainer_test.py::TestTrainerToolsSFT::test_trainer_tools 47.3s
tests/trainer/trainer_test.py::TestFullyAsyncMode_0_fsdp::test_fully_async_mode 1m 36s
tests/trainer/trainer_test.py::TestFullyAsyncMode_1_fsdp::test_fully_async_mode 1m 39s
tests/trainer/trainer_test.py::TestFullyAsyncMode_2_megatron::test_fully_async_mode 2m 20s
tests/trainer/trainer_test.py::TestTrainerCheckpointSave_0_fsdp::test_trainer 2m 6s
tests/trainer/trainer_test.py::TestTrainerCheckpointSave_1_megatron::test_trainer 4m 1s
tests/trainer/trainer_test.py::TestTrainerMIX::test_trainer 2m 8s
tests/trainer/trainer_test.py::TestMultiModalGRPO::test_trainer ⏭️ 811ms
tests/trainer/trainer_test.py::TestMultiModalSFT::test_trainer ⏭️ 807ms
tests/trainer/trainer_test.py::TestTrainerLoRA::test_trainer 3m 15s
tests/trainer/trainer_test.py::TestOverRollout::test_trainer 1m 10s
tests/trainer/trainer_test.py::TestTrainerPromptTruncation::test_trainer 55.6s
tests/utils/eval_utils_test.py::TestComputeScore::test_both_boxed_and_equivalent 15ms
tests/utils/eval_utils_test.py::TestComputeScore::test_both_boxed_and_not_equivalent 1ms
tests/utils/eval_utils_test.py::TestComputeScore::test_empty_ground_truth 2ms
tests/utils/eval_utils_test.py::TestComputeScore::test_empty_solution_string 1ms
tests/utils/eval_utils_test.py::TestComputeScore::test_multiple_boxed_answers_in_solution 2ms
tests/utils/eval_utils_test.py::TestComputeScore::test_solution_boxed_truth_raw_and_equivalent 1ms
tests/utils/eval_utils_test.py::TestComputeScore::test_solution_boxed_truth_raw_and_not_equivalent 1ms
tests/utils/eval_utils_test.py::TestComputeScore::test_solution_not_boxed 1ms
tests/utils/eval_utils_test.py::TestComputeScore::test_solution_raw_and_ground_truth_boxed_equivalent 1ms
tests/utils/eval_utils_test.py::TestMathEvalUtils::test_extract_answer 4ms
tests/utils/eval_utils_test.py::TestMathEvalUtils::test_verify_math_answer 76ms
tests/utils/eval_utils_test.py::TestEvalUtils::test_is_equiv 6ms
tests/utils/log_test.py::LogTest::test_actor_log 2.2s
tests/utils/log_test.py::LogTest::test_group_by_node 2.1s
tests/utils/log_test.py::LogTest::test_no_actor_log 908ms
tests/utils/plugin_test.py::TestPluginLoader::test_load_plugins_local_0__workspace_tests_utils_plugins 102ms
tests/utils/plugin_test.py::TestPluginLoader::test_load_plugins_local_1_tests_utils_plugins 95ms
tests/utils/plugin_test.py::TestPluginLoader::test_load_plugins_remote_0__workspace_tests_utils_plugins 10.6s
tests/utils/plugin_test.py::TestPluginLoader::test_load_plugins_remote_1_tests_utils_plugins 10.7s
tests/utils/plugin_test.py::TestPluginLoader::test_passing_custom_class_0__workspace_tests_utils_plugins 6.1s
tests/utils/plugin_test.py::TestPluginLoader::test_passing_custom_class_1_tests_utils_plugins 6.0s
tests/utils/registry_test.py::TestRegistryWithRay::test_dynamic_import 5.4s
tests/utils/registry_test.py::TestRegistry::test_algorithm_registry_mapping 2ms
tests/utils/registry_test.py::TestRegistry::test_buffer_module_registry_mapping 1ms
tests/utils/registry_test.py::TestRegistry::test_common_module_registry_mapping 45ms
tests/utils/registry_test.py::TestRegistry::test_register_module 1ms
tests/utils/registry_test.py::TestRegistry::test_utils_module_registry_mapping 1ms

Github Test Reporter by CTRF 💚

@pan-x-c pan-x-c merged commit 85b8577 into modelscope:main Dec 18, 2025
2 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants