From 1a1bbc4bac434b7d0c9f436b0a3bc4a9d6c85b02 Mon Sep 17 00:00:00 2001 From: Scott Sievert Date: Mon, 4 Sep 2017 16:41:00 -0500 Subject: [PATCH 1/2] Pass additional kwargs to pickle.load (in __load) --- pyradigm/pyradigm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pyradigm/pyradigm.py b/pyradigm/pyradigm.py index 2738bc2..fe53a81 100644 --- a/pyradigm/pyradigm.py +++ b/pyradigm/pyradigm.py @@ -585,7 +585,7 @@ def __copy(self, other): return self - def __load(self, path): + def __load(self, path, **kwargs): """Method to load the serialized dataset from disk.""" try: path = os.path.abspath(path) @@ -593,7 +593,7 @@ def __load(self, path): # loaded_dataset = pickle.load(df) self.__data, self.__classes, self.__labels, \ self.__dtype, self.__description, \ - self.__num_features, self.__feature_names = pickle.load(df) + self.__num_features, self.__feature_names = pickle.load(df, **kwargs) # ensure the loaded dataset is valid self.__validate(self.__data, self.__classes, self.__labels) From 1404d75da51a979b021acb780ac84edd0da81fae Mon Sep 17 00:00:00 2001 From: Scott Sievert Date: Tue, 17 Oct 2017 22:46:09 -0500 Subject: [PATCH 2/2] ENH: pass kwargs on in MLDataset.save --- pyradigm/pyradigm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pyradigm/pyradigm.py b/pyradigm/pyradigm.py index fe53a81..62c88dd 100644 --- a/pyradigm/pyradigm.py +++ b/pyradigm/pyradigm.py @@ -603,7 +603,7 @@ def __load(self, path, **kwargs): except: raise - def save(self, path): + def save(self, path, **kwargs): """Method to serialize the dataset to disk.""" try: path = os.path.abspath(path) @@ -612,7 +612,7 @@ def save(self, path): pickle.dump((self.__data, self.__classes, self.__labels, self.__dtype, self.__description, self.__num_features, self.__feature_names), - df) + df, **kwargs) return except IOError as ioe: raise IOError('Unable to save the dataset to file: {}', format(ioe))