Update argmax.py

deeppavlov · Nov 11, 2024 · 840cc77 · 840cc77
1 parent b16cbdf
commit 840cc77
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/autointent/modules/prediction/argmax.py b/autointent/modules/prediction/argmax.py
@@ -13,12 +13,14 @@
 
 class ArgmaxPredictor(PredictionModule):
     metadata = {}  # noqa: RUF012
+    name = "argmax"
 
     def __init__(self) -> None:
         pass
 
+
     @classmethod
-    def from_context(cls) -> Self:
+    def from_context(cls, context: Context) -> Self:
         return cls()
 
     def fit(