fix hpf

lenskit · Jan 11, 2025 · 09e05f7 · 09e05f7
1 parent d7316c9
commit 09e05f7
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/lenskit-hpf/lenskit/hpf.py b/lenskit-hpf/lenskit/hpf.py
@@ -8,7 +8,7 @@
 
 import hpfrec
 import numpy as np
-from pydantic import BaseModel
+from pydantic import BaseModel, JsonValue
 from typing_extensions import override
 
 from lenskit.data import Dataset, ItemList, QueryInput, RecQuery, Vocabulary
@@ -18,6 +18,7 @@
 
 
 class HPFConfig(BaseModel, extra="allow"):
+    __pydantic_extra__: dict[str, JsonValue]
     features: int = 50
 
 
@@ -61,9 +62,9 @@ def train(self, data: Dataset):
             }
         )
 
-        hpf = hpfrec.HPF(self.config.features, reindex=False, **self.config.__pydantic_extra__)
+        hpf = hpfrec.HPF(self.config.features, reindex=False, **self.config.__pydantic_extra__)  # type: ignore
 
-        _logger.info("fitting HPF model with %d features", self.features)
+        _logger.info("fitting HPF model with %d features", self.config.features)
         hpf.fit(log)
 
         self.users_ = data.users