Skip to content

Commit b8fb273

Browse files
committed
define get_implicit_initialization_params in child classes
1 parent f3f4446 commit b8fb273

File tree

3 files changed

+11
-0
lines changed

3 files changed

+11
-0
lines changed

autointent/modules/base/_decision.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,9 @@
1818
class BaseDecision(BaseModule, ABC):
1919
"""Base class for decision modules."""
2020

21+
def get_implicit_initialization_params(self) -> dict[str, Any]:
22+
return {}
23+
2124
@abstractmethod
2225
def fit(
2326
self,

autointent/modules/base/_embedding.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
"""Base class for embedding modules."""
22

33
from abc import ABC
4+
from typing import Any
45

56
from autointent import Context
67
from autointent.custom_types import ListOfLabels
@@ -10,6 +11,9 @@
1011
class BaseEmbedding(BaseModule, ABC):
1112
"""Base class for embedding modules."""
1213

14+
def get_implicit_initialization_params(self) -> dict[str, Any]:
15+
return {}
16+
1317
def get_train_data(self, context: Context) -> tuple[list[str], ListOfLabels]:
1418
"""Get train data.
1519

autointent/modules/base/_regex.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,13 @@
11
"""Base class for embedding modules."""
22

33
from abc import ABC
4+
from typing import Any
45

56
from autointent.modules.base import BaseModule
67

78

89
class BaseRegex(BaseModule, ABC):
910
"""Base class for rule-based modules."""
11+
12+
def get_implicit_initialization_params(self) -> dict[str, Any]:
13+
return {}

0 commit comments

Comments
 (0)