From d672f88809bf67d14bb2f613a1a73f1a061121ae Mon Sep 17 00:00:00 2001 From: bxdd Date: Mon, 4 Dec 2023 19:36:42 +0800 Subject: [PATCH] [FIX] fix pickle model bugs --- learnware/tests/templates/pickle_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/learnware/tests/templates/pickle_model.py b/learnware/tests/templates/pickle_model.py index b4cb095..2039aa6 100644 --- a/learnware/tests/templates/pickle_model.py +++ b/learnware/tests/templates/pickle_model.py @@ -16,8 +16,8 @@ class PickleLoadedModel(BaseModel): ): super(PickleLoadedModel, self).__init__(input_shape=input_shape, output_shape=output_shape) dir_path = os.path.dirname(os.path.abspath(__file__)) - self.pickle_filepath = os.path.join(pickle_filepath, dir_path) - with open(pickle_filepath, "rb") as fd: + self.pickle_filepath = os.path.join(pickle_filename, dir_path) + with open(pickle_filename, "rb") as fd: self.model = pickle.load(fd) self.predict_method = predict_method self.fit_method = fit_method