Modifying specific hoppings about pybinding HOT 4 CLOSED

import pybinding as pb
from pybinding.repository.graphene import a
import matplotlib.pyplot as plt
from numpy import sqrt, pi

phi0 = 1
a_cc = a/np.sqrt(3) # nm, carbon carbon bond length
gamma = 3.033 # eV hopping energy from full solution of tight binding nearest neighbor
t1 = gamma
t2 = gamma
t3 = gamma

lat = pb.Lattice(a1=a*np.array([1/2,np.sqrt(3)/2]), a2=a*np.array([1/2,-np.sqrt(3)/2]))
lat.add_sublattices(
        ('A',[0, -a_cc/2], 0),
        ('B', [0, a_cc/2], 0)
)
lat.register_hopping_energies({'t1':t1, 't2': t2, 't3': t3})
lat.add_hoppings(
        ([0,0],'A','B',"t1"),
        ([0,1],'A','B',"t2"),
        ([1,0],'B','A',"t3")
)
lat.plot()
plt.show()
plt.savefig("lat.pdf")
def vectorPotential(ab, x, y):
    b1Term = -(1/sqrt(3))*np.sin((2*pi*x/ab) - (2*pi*y/(ab*sqrt(3))))
    b2Term = (2/sqrt(3))*np.cos(4*pi*y/(sqrt(3)*ab))
    b3Term = (1/sqrt(3))*np.cos((2*pi*x/ab)+(2*pi*y/(ab*sqrt(3))))
    Ax = b1Term + b2Term + b3Term
    return Ax

shiftInEnergies = []
def pilarStrain(B,ab):
    @pb.hopping_energy_modifier
    def strained_hopping(energy, x1, y1, z1, x2, y2, z2, hop_id):
        Ax = vectorPotential(ab,x1,y1)
        t1_shifted = gamma*(1-(((3*pi*a_cc*B)/(2*phi0))*Ax))
        t2_shifted = gamma*(1+(((3*pi*a_cc*B)/(2*phi0))*Ax))
        t3_shifted = t2_shifted
        xVec = x2-x1
        yVec = y2-y1
        energy[hop_id == 't1'] *= 1.1
        if hop_id == 't1':
            energy = t1_shifted
        elif hop_id == 't2':
            energy = t2_shifted
        elif hop_id == 't3':
            energy = t3_shifted
        global shiftInEnergies
        shiftInEnergies = energy - gamma
        return energy
    return  strained_hopping

model = pb.Model(
        lat,
        pb.primitive(10, 10),
        pb.translational_symmetry(10*a,10*a),
        pilarStrain(1, 1)
)

plt.figure()
bz = lat.brillouin_zone()
bands = pb.solver.lapack(model).calc_wavefunction(bz[3]*0, bz[3]/100, (bz[3]+bz[4])/200, bz[3]*0, step=0.001).bands_disentangled
plt.plot(bands.k_path.as_1d(), bands.energy)
plt.gca().set_ylim([-.5, .5])
plt.show()
plt.savefig("band.pdf")

pip uninstall pybinding
pip install -i https://test.pypi.org/simple/ pybinding