From 46359d6b5108ab7d1e6fed02ea61a90222b9f665 Mon Sep 17 00:00:00 2001 From: Peter Scherpelz Date: Fri, 16 Jul 2021 13:23:45 -0700 Subject: [PATCH] Fix ignoring arguments passed to save_model. Before this change, defaults were passed to _save_model, rather than the correct arguments the user supplied to save_model. --- GPy/models/gp_classification.py | 2 +- GPy/models/gp_regression.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/GPy/models/gp_classification.py b/GPy/models/gp_classification.py index 8bd01a28..a116b456 100644 --- a/GPy/models/gp_classification.py +++ b/GPy/models/gp_classification.py @@ -56,7 +56,7 @@ class GPClassification(GP): return GPClassification.from_gp(m) def save_model(self, output_filename, compress=True, save_data=True): - self._save_model(output_filename, compress=True, save_data=True) + self._save_model(output_filename, compress=compress, save_data=save_data) @staticmethod def _build_from_input_dict(input_dict, data=None): diff --git a/GPy/models/gp_regression.py b/GPy/models/gp_regression.py index 10b3ba61..3da596ef 100644 --- a/GPy/models/gp_regression.py +++ b/GPy/models/gp_regression.py @@ -54,4 +54,4 @@ class GPRegression(GP): return GPRegression.from_gp(m) def save_model(self, output_filename, compress=True, save_data=True): - self._save_model(output_filename, compress=True, save_data=True) + self._save_model(output_filename, compress=compress, save_data=save_data)