from pint import UnitRegistry

ureg = UnitRegistry()


class MyClass:
    @ureg.check(None, '[electric_potential]')
    def set_voltage(self, voltage) -> None:
        pass
print('Hello')