Source code for mhcflurry.testing_utils

"""
Utilities used in MHCflurry unit tests.
"""
from . import Class1NeuralNetwork
from .common import configure_tensorflow


[docs]def startup():
    """
    Configure Keras backend for running unit tests.
    """
    configure_tensorflow("tensorflow-cpu", num_threads=2)


[docs]def cleanup():
    """
    Clear tensorflow session and other process-wide resources.
    """
    import tensorflow.keras.backend as K
    Class1NeuralNetwork.clear_model_cache()
    K.clear_session()