Finally fixed; extended test to check model loads correctly.
@@ -510,8 +510,8 @@ class ParametricUMAP(UMAP):
print("Keras full model saved to {}".format(parametric_model_output))
# # save model.pkl (ignoring unpickleable warnings)
- # with catch_warnings():
- # filterwarnings("ignore")
+ with catch_warnings():
+ filterwarnings("ignore")
# work around optimizers not pickling anymore (since tf 2.4)
self._optimizer_dict = self.optimizer.get_config()
model_output = os.path.join(save_location, "model.pkl")
