Use ChoiceSet.values() for access to raw values

This commit is contained in:
Jeremy Stretch 2020-01-09 14:56:33 -05:00
parent fe490d144a
commit 1d3651e255

View File

@ -168,7 +168,7 @@ def to_meters(length, unit):
if length < 0:
raise ValueError("Length must be a positive integer")
valid_units = [u[0] for u in CableLengthUnitChoices]
valid_units = CableLengthUnitChoices.values()
if unit not in valid_units:
raise ValueError(
"Unknown unit {}. Must be one of the following: {}".format(unit, ', '.join(valid_units))