fix custom_info

This commit is contained in:
robcaulk
2022-12-05 20:22:54 +01:00
parent d8565261e1
commit 62c69bf2b5
4 changed files with 4 additions and 6 deletions

View File

@@ -20,8 +20,8 @@ class Base4ActionRLEnv(BaseEnvironment):
"""
Base class for a 4 action environment
"""
def __init__(self, *args):
super().__init__(*args)
def __init__(self, **kwargs):
super().__init__(**kwargs)
self.actions = Actions
def set_action_space(self):