Fix VALIDATION_MODE ref

VinF · Mar 22, 2016 · 5f14f3e · 5f14f3e
1 parent f22185f
commit 5f14f3e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/General_Deep_Q_RL/environments/MG_two_storages_env.py b/General_Deep_Q_RL/environments/MG_two_storages_env.py
@@ -112,7 +112,7 @@ def reset(self, mode):
             self.production=self.production_train
             self.consumption_norm=self.consumption_train_norm
             self.consumption=self.consumption_train
-        elif mode == VALIDATION_MODE:
+        elif mode == MyEnv.VALIDATION_MODE:
             self.production_norm=self.production_valid_norm
             self.production=self.production_valid
             self.consumption_norm=self.consumption_valid_norm