calc = TransportCalculator(h=h, h1=h1, h2=h2, energies=energies) transm = calc.get_transmission() plt.plot(energies, transm)