Error of python setup.py install

[hhapk]

Thank you for your fascinating codes.
I wonder, there may be a mistake in init.py (in torch-radon/build_tools file direcotory). When I run python setup.py install in linux, bash raises a mistake "UnboundLocalError: local variable 'compute_capabilities' referenced before assignment"

I found "compute_capabilities" is wrong in the format of "compute_capabilites" in two different lines n init.py (in torch-radon/build_tools file direcotory). After I fix it, the code runs smoothly.
Looking forward to checking in your free time!