admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-27 19:56:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(rl): add multiprocessing_context option + check to rl training pipeline

Browse Source
This commit is contained in:
Steven Palma
2026-07-24 13:34:58 +02:00
parent e66da375c8
commit 21ba0ffd8e
5 changed files with 40 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/test_gaussian_actor_config.py
+2
View File
@@ -113,6 +113,7 @@ def test_gaussian_actor_config_default_initialization():
# Concurrency configuration
assert config.concurrency.actor == "threads"
assert config.concurrency.learner == "threads"
assert config.concurrency.multiprocessing_context == "spawn"
assert isinstance(config.actor_network_kwargs, ActorNetworkConfig)
assert isinstance(config.policy_kwargs, PolicyConfig)
@@ -152,6 +153,7 @@ def test_concurrency_config():
config = ConcurrencyConfig()
assert config.actor == "threads"
assert config.learner == "threads"
assert config.multiprocessing_context == "spawn"
def test_gaussian_actor_config_custom_initialization():