Source code for matchms.importing.load_from_pickle

import pickle
from typing import List
from matchms.Spectrum import Spectrum




def load_from_pickle(filename: str, metadata_harmonization: bool) -> List[Spectrum]:
    """Load spectra stored in pickle

    Args:
        filename (str): Pickled file with spectra.

    Returns:
        Any: Unpickled object. Should be a list of Spectra.
    """
    with open(filename, "rb") as file:
        loaded_object = pickle.load(file)

    if not isinstance(loaded_object, list):
        raise TypeError("Expected list of spectra")
    for spectrum in loaded_object:
        if not isinstance(spectrum, Spectrum):
            raise TypeError("Expected list of spectra")

    if metadata_harmonization:
        loaded_object = [
            Spectrum(spectrum.mz, spectrum.intensities, spectrum.metadata, metadata_harmonization)
            for spectrum in loaded_object
        ]
    return loaded_object