i think i did the converstion correct

This commit is contained in:
2023-01-12 18:33:06 +01:00
parent 03c6341b19
commit ee444cb56c
9 changed files with 131 additions and 60 deletions

View File

@@ -1,7 +1,7 @@
from ai_economist.foundation.base import base_env
from threading import Event, Lock, Thread
from queue import Queue
class BaseEconVecEnv():
class BaseEconWrapper():
"""Base class for connecting reciever wrapper to a multi threaded econ simulation and training session"""
base_notification=Event() #Notification for Base
@@ -33,11 +33,12 @@ class BaseEconVecEnv():
def register_vote(self):
"""Register reciever on base. Base now knows"""
self.n_voters+=1
def run(self):
"""Start the base wrapper"""
thr=Thread(target=self._run,daemon=True)
thr.run()
thr.start()
return thr
def _run(self):
@@ -46,16 +47,20 @@ class BaseEconVecEnv():
self.reset_notification.clear()
self.step_notification.clear()
self.stop_edit_lock.release()
if self.stop_edit_lock.locked():
self.stop_edit_lock.release()
self.stop=False
self.action_edit_lock.release()
if self.action_edit_lock.locked():
self.action_edit_lock.release()
self.actor_actions={}
self.vote_lock.release()
if self.vote_lock.locked():
self.vote_lock.release()
self.reset_notification.clear()
self.n_votes_reset=0
self.n_votes_step=0
self.env_data_lock.release()
if self.env_data_lock.locked():
self.env_data_lock.release()
self.obs=None
self.rew=None
self.done=None