Skip to content

Commit

Permalink
use explicit multiprocessing context for creating Pipe in subproc.py (#…
Browse files Browse the repository at this point in the history
  • Loading branch information
maxhuettenrauch committed Apr 19, 2024
1 parent a043711 commit 9c0b3e7
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions tianshou/env/worker/subproc.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
import time
from collections import OrderedDict
from collections.abc import Callable
from multiprocessing import Pipe, connection
from multiprocessing import connection
from multiprocessing.context import BaseContext
from typing import Any, Literal

Expand Down Expand Up @@ -149,11 +149,11 @@ def __init__(
share_memory: bool = False,
context: BaseContext | Literal["fork", "spawn"] | None = None,
) -> None:
self.parent_remote, self.child_remote = Pipe()
self.share_memory = share_memory
self.buffer: dict | tuple | ShArray | None = None
if not isinstance(context, BaseContext):
context = multiprocessing.get_context(context)
self.parent_remote, self.child_remote = context.Pipe()
self.share_memory = share_memory
self.buffer: dict | tuple | ShArray | None = None
assert hasattr(context, "Process") # for mypy
if self.share_memory:
dummy = env_fn()
Expand Down

0 comments on commit 9c0b3e7

Please sign in to comment.