made a bit of progress

wrapper/base_econ_wrapper.py

@@ -17,7 +17,7 @@ class BaseEconWrapper():
     info=None
     n_data_retrieved=0
 
-    def __init__(self, econ: base_env.BaseEnvironment):
+    def __init__(self, econ: base_env.BaseEnvironment,auto_reset):
         self.env=econ
         self.vote_lock=Lock()
           
@@ -26,6 +26,8 @@ class BaseEconWrapper():
         self.action_edit_lock=Lock()
         self.stop_edit_lock=Lock()
         self.env_data_lock=Lock()
+        self.auto_reset=auto_reset
+  
 
     def register_vote(self):
         """Register reciever on base. Returns ID of Voter to pass on during blocking"""
@@ -79,7 +81,7 @@ class BaseEconWrapper():
 
             #check for reset
             self.vote_lock.acquire() # we might edit votes
-            if self.n_voters==self.n_votes_reset:
+            if self.n_voters<=self.n_votes_reset:
                 ## perform reset
                 self.n_votes_reset=0
                 self._reset()
@@ -96,6 +98,7 @@ class BaseEconWrapper():
            # release actions
             # we are done
 
+
     def stop_env(self):
         """Stops the wrapper"""
         self.stop_edit_lock.acquire()
@@ -110,13 +113,17 @@ class BaseEconWrapper():
         self.n_votes_reset=0
         self.obs=self.env.reset() #Reset env
         self.rew=None
-        self.done=None
+        self.done={'__all__': False}
         self.info=None
         self.env_data_lock.release() #Release lock
         # Notify for reset
         self.reset_notification.set()
         for v in self.step_notifications:
             v.clear() # unlock stepping
+    def force_reset(self):
+        """Force a reset with no votes"""
+        self._reset()
+        #self.n_votes_reset=-self.n_voters
 
     def _step(self):
         """Steping interaly"""