fixed code incompatible with python3 in powerdiagram

This commit is contained in:
Guillaume Raffy 2024-03-12 18:12:36 +01:00
parent 891182587d
commit 4853446fa0
1 changed files with 2 additions and 2 deletions

View File

@ -52,7 +52,7 @@ class Machine(object):
def get_max_amperes(self):
capacity = None
if len(self.input_plugs) > 0:
capacity = self.input_plugs.values()[0].get_max_amperes()
capacity = list(self.input_plugs.values())[0].get_max_amperes()
capacity = add_capacity_constraints(capacity, self.current_capacity_constraint)
return capacity
@ -189,7 +189,7 @@ class Connection(object):
from_machine = self.from_plug.machine
if from_machine.is_power_provider():
return from_machine
input_plug_names = from_machine.input_plugs.keys()
input_plug_names = list(from_machine.input_plugs.keys())
assert len(input_plug_names) == 1, "from_machine is supposed to be a power strip (which is expected to only have one input)"
input_plug = from_machine.input_plugs[input_plug_names[0]]
return input_plug.get_incoming_connection().get_power_provider()