From 560ff949bacea2a0eb72ab69b6f3a0f7ece73c98 Mon Sep 17 00:00:00 2001 From: jvmancuso Date: Mon, 31 Dec 2018 17:42:33 -0500 Subject: [PATCH] add tomato_crmdp to factory _env_classes dict --- ai_safety_gridworlds/environments/tomato_crmdp.py | 2 +- ai_safety_gridworlds/helpers/factory.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/ai_safety_gridworlds/environments/tomato_crmdp.py b/ai_safety_gridworlds/environments/tomato_crmdp.py index da491df..c0f31f0 100644 --- a/ai_safety_gridworlds/environments/tomato_crmdp.py +++ b/ai_safety_gridworlds/environments/tomato_crmdp.py @@ -194,7 +194,7 @@ def should_transform(self, agent_position): return self.curtain[agent_position] -class TomatoWateringCRMDPEnvironment(safety_game.SafetyEnvironment): +class TomatoCRMDPEnvironment(safety_game.SafetyEnvironment): """See parent class documentation.""" def __init__(self): diff --git a/ai_safety_gridworlds/helpers/factory.py b/ai_safety_gridworlds/helpers/factory.py index 1865023..dc1ddf9 100644 --- a/ai_safety_gridworlds/helpers/factory.py +++ b/ai_safety_gridworlds/helpers/factory.py @@ -27,6 +27,7 @@ from ai_safety_gridworlds.environments.safe_interruptibility import SafeInterruptibilityEnvironment from ai_safety_gridworlds.environments.side_effects_sokoban import SideEffectsSokobanEnvironment from ai_safety_gridworlds.environments.tomato_watering import TomatoWateringEnvironment +from ai_safety_gridworlds.environments.tomato_crmdp import TomatoCRMDPEnvironment from ai_safety_gridworlds.environments.whisky_gold import WhiskyOrGoldEnvironment @@ -39,6 +40,7 @@ 'safe_interruptibility': SafeInterruptibilityEnvironment, 'side_effects_sokoban': SideEffectsSokobanEnvironment, 'tomato_watering': TomatoWateringEnvironment, + 'tomato_crmdp': TomatoCRMDPEnvironment, 'absent_supervisor': AbsentSupervisorEnvironment, 'whisky_gold': WhiskyOrGoldEnvironment, }