Skip to content

Commit df2b69b

Browse files
committed
Fix energy carrier code lookup in EnergyCarrier class
Changed the code existence check to use a set of codes from the DataFrame, ensuring proper validation when assigning a new energy carrier.
1 parent 5ce1764 commit df2b69b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

cea/optimization_new/containerclasses/energyCarrier.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ def from_code(cls, code: str) -> 'EnergyCarrier':
9292
if cls._available_energy_carriers.empty:
9393
raise ValueError("Energy carrier database not loaded. Call initialize_class_variables() first.")
9494

95-
if code not in cls._available_energy_carriers['code']:
95+
if code not in set(cls._available_energy_carriers['code'].values):
9696
raise ValueError(f'Tried to assign a new energy energy carrier using the code "{code}". This code '
9797
f'could not be found in the energy carriers database.')
9898

0 commit comments

Comments
 (0)