Missing code to save the fitted primitive parameters

[lx1037]

hello, thanks for sharing the code! I find that in both test.py and generate_predictions.py, there is a line from src.primitives import SaveParameters, but neither of them actually uses this moudule and to save the fitted primitive parameters in the subsequent code. Could you please supplement the relevant code to do so?