Skip to content

Commit 8e4cb26

Browse files
committed
fix tests exept dumpload
1 parent 0cc749d commit 8e4cb26

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

autointent/modules/scoring/_rnn.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -203,6 +203,15 @@ def device(self) -> str:
203203
"""Get device used for model computations."""
204204
return self._device
205205

206+
@device.setter
207+
def device(self, value: str) -> None:
208+
"""Set device for model computations."""
209+
self._device = value
210+
211+
def get_implicit_initialization_params(self) -> dict[str, Any]:
212+
"""Return default params used in ``__init__`` method."""
213+
return {"rnn_config": self.rnn_config.model_dump()}
214+
206215
class SupervisedRNNClassifier(nn.Module):
207216
def __init__(
208217
self,

0 commit comments

Comments
 (0)