bug fix

Author: voorhs

autointent/_dump_tools.py

@@ -26,10 +26,10 @@
 
 ModuleSimpleAttributes = None | str | int | float | bool | list  # type: ignore[type-arg]
 
-ModuleAttributes: TypeAlias = (  # type: ignore[type-arg]
+ModuleAttributes: TypeAlias = (
     ModuleSimpleAttributes
     | TagsList
-    | np.ndarray
+    | np.ndarray  # type: ignore[type-arg]
     | Embedder
     | VectorIndex
     | BaseEstimator

autointent/modules/scoring/_torch/base_scorer.py

@@ -65,6 +65,7 @@ def predict(self, utterances: list[str]) -> npt.NDArray[Any]:
 
     def clear_cache(self) -> None:
         if hasattr(self, "_model"):
+            self._model.vocab_config.vocab = None
             del self._model
             torch.cuda.empty_cache()