Door Area Indicators

2022-01-18 11:39:19 +01:00
parent 3ce6302e8a
commit a16d7e709e
7 changed files with 34 additions and 9 deletions
--- a/environments/factory/factory_battery.py
+++ b/environments/factory/factory_battery.py
@ -155,7 +155,7 @@ class BatteryFactory(BaseFactory):
        if isinstance(btry_prop, dict):
            btry_prop = BatteryProperties(**btry_prop)
        if isinstance(rewards_dest, dict):
-            rewards_dest = RewardsBtry(**rewards_dest)
+            rewards_dest = BatteryProperties(**rewards_dest)
        self.btry_prop = btry_prop
        self.rewards_dest = rewards_dest
        super().__init__(*args, **kwargs)