Browse Source

[FIX] fix a minor bug in save

pull/3/head
Gao Enhao 2 years ago
parent
commit
e92fc2498d
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      abl/learning/abl_model.py

+ 1
- 1
abl/learning/abl_model.py View File

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


Loading…
Cancel
Save