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.
This commit is contained in:
Peter Scherpelz 2021-07-16 13:23:45 -07:00
parent 3e19a85575
commit 46359d6b51
2 changed files with 2 additions and 2 deletions

View file

@ -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):

View file

@ -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)