i think i did the converstion correct
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user