Model Upload & Validation

[PaulCCCCCCH]

Part of Robustar v0.3 Main Features (#175)

When the user uploads a model with the UploadModel API, they will pass in a model definition (Python code as string), model weight file, and some other metadata. We need to import the code dynamically, validate the code, push data into the database if its valid, load in the weight file (if provided) and set the model as the current model. Specifically, we need to:

• Create a file (if not exist) back-end/utils/model_utils.py
• Write a function that takes a string of Python code, dump it into a file, and import it dynamically. Return the error if there is import fails.
• Load the model weight. Return the error if load fails.
• Validate the model by running the model against a small portion of the validation data. Return the error if anything fails.
• Use generic database queries defined in back-end/utils/db_ops.py to save the model metadata into the database.
• Set the model as the current model.