[src/rl] Set imbalance costs through fleetenv

indyfree · indyfree · commit 408f4be3c902 · 2019-04-12T13:11:21.000+02:00
diff --git a/src/evsim/envs/fleet_env.py b/src/evsim/envs/fleet_env.py
@@ -47,6 +47,9 @@ def init_sim(self):
         )
         self.sim = Simulation(cfg, self.controller)
 
+    def imbalance_costs(self, cost):
+        self.controller.imbalance_costs = cost
+
     def seed(self, seed=None):
         self.np_random, seed = seeding.np_random(seed)
         return [seed]

Original file line number	Diff line number	Diff line change
`@@ -47,6 +47,9 @@ def init_sim(self):`
`47`	`47`	`)`
`48`	`48`	`self.sim = Simulation(cfg, self.controller)`
`49`	`49`
	`50`	`+ def imbalance_costs(self, cost):`
	`51`	`+ self.controller.imbalance_costs = cost`
	`52`	`+`
`50`	`53`	`def seed(self, seed=None):`
`51`	`54`	`self.np_random, seed = seeding.np_random(seed)`
`52`	`55`	`return [seed]`