From e92fc2498d4c9f05ef9ab09cc09478a8e6bbf2f2 Mon Sep 17 00:00:00 2001 From: Gao Enhao Date: Sun, 11 Jun 2023 22:38:05 +0800 Subject: [PATCH] [FIX] fix a minor bug in save --- abl/learning/abl_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/abl/learning/abl_model.py b/abl/learning/abl_model.py index df83238..700976a 100644 --- a/abl/learning/abl_model.py +++ b/abl/learning/abl_model.py @@ -105,7 +105,7 @@ class ABLModel: def save(self, *args, **kwargs) -> None: _model = self.classifier_list[0] if hasattr(_model, "save"): - self._model.save(*args, **kwargs) + _model.save(*args, **kwargs) else: raise NotImplementedError(f"{type(_model).__name__} object dosen't have the save method")