tf_agents.system.default.multiprocessing_core.handle_test_main

Function that wraps the test main in a multiprocessing-friendly way.

tf_agents.system.default.multiprocessing_core.handle_test_main(
    parent_main_fn, *args, **kwargs
)

This function additionally accepts an extra_state_savers kwarg; users can provide a list of tf_agents.multiprocessing.StateSaver instances, where a StateSaver tells multiprocessing how to store some global state and how to restore it in the subprocess.

parent_main_fn A callable.
*args rgs for parent_main_fn.
**kwargs kwargs for parent_main_fn. This may also include extra_state_savers kwarg.

Output of parent_main_fn.