remove reset_task from subproc vec env (#45)
This commit is contained in:
committed by
Peter Zhokhov
parent
34af61a132
commit
337d913a8f
@@ -68,11 +68,6 @@ class SubprocVecEnv(VecEnv):
|
|||||||
remote.send(('reset', None))
|
remote.send(('reset', None))
|
||||||
return np.stack([remote.recv() for remote in self.remotes])
|
return np.stack([remote.recv() for remote in self.remotes])
|
||||||
|
|
||||||
def reset_task(self):
|
|
||||||
for remote in self.remotes:
|
|
||||||
remote.send(('reset_task', None))
|
|
||||||
return np.stack([remote.recv() for remote in self.remotes])
|
|
||||||
|
|
||||||
def close_extras(self):
|
def close_extras(self):
|
||||||
if self.waiting:
|
if self.waiting:
|
||||||
for remote in self.remotes:
|
for remote in self.remotes:
|
||||||
|
Reference in New Issue
Block a user