fix errors on specifying the voltage barrier type

Future-Power-Networks · Jan 30, 2022 · 556231c · 556231c
1 parent 81aa2a3
commit 556231c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/args/env_args/var_voltage_control.yaml b/args/env_args/var_voltage_control.yaml
@@ -1,7 +1,7 @@
 env: "voltage_control"
 
 env_args:
-    "voltage_loss_type": "l1" # "l1", "l2", "bowl", "courant_beltrami", "bump"
+    "voltage_barrier_type": "l1" # "l1", "l2", "bowl", "courant_beltrami", "bump"
     "voltage_weight": 1.0
     "q_weight": 0.1
     "line_weight": null

diff --git a/environments/var_voltage_control/voltage_control_env.py b/environments/var_voltage_control/voltage_control_env.py
@@ -55,7 +55,7 @@ def __init__(self, kwargs):
 
         # define episode and rewards
         self.episode_limit = args.episode_limit
-        self.voltage_loss_type = getattr(args, "voltage_loss", "l1")
+        self.voltage_barrier_type = getattr(args, "voltage_barrier_type", "l1")
         self.voltage_weight = getattr(args, "voltage_weight", 1.0)
         self.q_weight = getattr(args, "q_weight", 0.1)
         self.line_weight = getattr(args, "line_weight", None)
@@ -87,7 +87,7 @@ def __init__(self, kwargs):
         self.last_q = self.powergrid.sgen["q_mvar"].to_numpy(copy=True)
 
         # initialise voltage barrier function
-        self.voltage_barrier = VoltageBarrier(self.voltage_loss_type)
+        self.voltage_barrier = VoltageBarrier(self.voltage_barrier_type)
         self._rendering_initialized = False
 
     def reset(self, reset_time=True):