# Add the root path of the pygra library
import os ; import sys 
sys.path.append(os.path.dirname(os.path.realpath(__file__))+"/../../../src")





from pygra import geometry
g = geometry.triangular_lattice()
#g = geometry.honeycomb_lattice()
h = g.get_hamiltonian(has_spin=True)
h.turn_dense()
#h.shift_fermi(1.5)
h.set_filling(0.5,nk=20)
from pygra import magneticresponse as response
from pygra import parallel
parallel.cores = 6
response.magnetic_response_map(h,nk=20,nq=40,j=[0.5,0.,0.])
#h.get_bands()
#dos.dos(h,nk=100,use_kpm=True)






