INNER CODE UNIT · Python

main

operatorai/modelstore · workflows/main.py:37

def main(modelstore_in: str):
    """Executes all of the integration tests in a given storage type"""
    print(f"🆕  Running modelstore example with {modelstore_in} backend.")

    # Create a modelstore instance
    model_store = create_model_store(modelstore_in)

    # Run actions on the modelstore instance that are
    # not dependent on a trained model (e.g. model states)
    print("⏱  Running storage actions")
    actions.run_on_storage(model_store)

    # Run actions on the modelstore instance that
    # are dependent on a trained model (e.g. upload, download)
    print("⏱  Running storage + model actions")
    with tempfile.TemporaryDirectory() as tmp_dir:
        extra_files = extra.files(tmp_dir, num_files=2)
        for model in iter_models():

View source record →