Update Base5ActionRLEnv.py

Fix init
This commit is contained in:
Shane 2024-05-24 22:05:28 +10:00 committed by GitHub
parent ffd828b6ad
commit 1d5abe5b75
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -22,7 +22,7 @@ class Base5ActionRLEnv(BaseEnvironment):
Base class for a 5 action environment
"""
def __init__(self, **kwargs):
def __init__(self, *args, action_space_type: str = "Discrete", **kwargs):
super().__init__(**kwargs)
self.action_space_type = action_space_type
self.actions = Actions