# file(GLOB TEST_OPS RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}" "test_*.py")
# string(REPLACE ".py" "" TEST_OPS "${TEST_OPS}")
if(WITH_DISTRIBUTE AND WITH_GPU)
    py_test_modules(test_auto_parallel_relaunch MODULES test_auto_parallel_relaunch ENVS ${dist_ENVS})
    set_tests_properties(test_auto_parallel_relaunch PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE" TIMEOUT 120)
    py_test_modules(test_relaunch_with_planner MODULES test_relaunch_with_planner ENVS ${dist_ENVS})
    set_tests_properties(test_relaunch_with_planner PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE" TIMEOUT 120)
    py_test_modules(test_relaunch_with_gpt_planner MODULES test_relaunch_with_gpt_planner ENVS ${dist_ENVS})
    set_tests_properties(test_relaunch_with_gpt_planner PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE" TIMEOUT 240)
    py_test_modules(test_engine_api MODULES test_engine_api ENVS ${dist_ENVS})
    set_tests_properties(test_engine_api PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE" TIMEOUT 80)
    py_test_modules(test_while_op_completion MODULES test_while_op_completion ENVS ${dist_ENVS})
    py_test_modules(test_converter MODULES test_converter ENVS ${dist_ENVS})
    set_tests_properties(test_converter PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE" TIMEOUT 50)

    py_test_modules(test_tunable_variable MODULES test_tunable_variable ENVS ${dist_ENVS})
    py_test_modules(test_tunable_space MODULES test_tunable_space ENVS ${dist_ENVS})
    py_test_modules(test_recorder MODULES test_recorder ENVS ${dist_ENVS})
    py_test_modules(test_trial MODULES test_trial ENVS ${dist_ENVS})
endif()
