Skip to content

[BugFix] Fix hidden multiprocessing import in coding_ppo tutorial #3779

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
vmoens merged 3 commits into from
Jun 3, 2026
Merged

[BugFix] Fix hidden multiprocessing import in coding_ppo tutorial #3779

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
22e9873
Fix hidden multiprocessing import in coding_ppo tutorial
coder-jayp May 19, 2026
bc51024
Merge remote-tracking branch 'origin/main' into fix/coding-ppo-multip…
vmoens Jun 3, 2026
d198aae
[BugFix] Expose PPO tutorial multiprocessing import
vmoens Jun 3, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 5 additions & 4 deletions tutorials/sphinx-tutorials/coding_ppo.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -108,13 +108,14 @@
# sphinx_gallery_start_ignore
import warnings

from torch import multiprocessing as _multiprocessing

warnings.filterwarnings("ignore")
# Set multiprocessing start method to fork if not already set
# This allows the tutorial to run as a script without if __name__ == "__main__"
from torch import multiprocessing

if multiprocessing.get_start_method(allow_none=True) is None:
multiprocessing.set_start_method("fork")
if _multiprocessing.get_start_method(allow_none=True) is None:
_multiprocessing.set_start_method("fork")

# sphinx_gallery_end_ignore

Expand All @@ -124,7 +125,7 @@
import torch
from tensordict.nn import TensorDictModule
from tensordict.nn.distributions import NormalParamExtractor
from torch import nn
from torch import multiprocessing, nn

from torchrl.collectors import Collector
from torchrl.data.replay_buffers import ReplayBuffer
Expand Down
Loading