Update getFermiMomentum_tb3d.py