INNER CODE UNIT · Python

get_amset

SMTG-Bham/ThermoParser · tp/cli/cli.py:298

def get_amset(amset_file, quantity, dtype, doping, direction, temperature, spin,
              scattering, band, kpoint, poscar):
    """Prints AMSET values at given conditions.

    Requires an AMSET transport or mesh file.
    """
    # I'm making a few of these for different inputs so it doesn't get
    # too unwieldly checking which source they're from, but they could
    # also be combined.

    if quantity == 'scattering_rates':
        quantity = 'weighted_rates'

    try:
        data = tp.data.load.amset(amset_file, quantity, doping=dtype)
    except UnicodeDecodeError:
        data = tp.data.load.amset_mesh(amset_file, quantity, spin=spin,
                                       doping=dtype)

View source record →