mypy fix

Author: SeBorgey

autointent/configs/_transformers.py

@@ -123,7 +123,7 @@ class CrossEncoderConfig(HFModelConfig):
 
 class RNNConfig(BaseModel):
     model_config = ConfigDict(extra="forbid")
-    device: str = Field(None, description="Torch notation for CPU or CUDA.")
+    device: str | None = Field(None, description="Torch notation for CPU or CUDA.")
     max_seq_length: int = Field(128, description="Maximum sequence length.")
     padding_idx: int = Field(0, description="Index used for padding.")
     batch_size: PositiveInt = Field(32, description="Batch size for model inference.")

autointent/modules/scoring/_rnn.py

@@ -198,6 +198,10 @@ def clear_cache(self) -> None:
         if hasattr(self, "_model"):
             del self._model
 
+    @property
+    def device(self) -> str:
+        """Get device used for model computations."""
+        return self._device
 
 class SupervisedRNNClassifier(nn.Module):
     def __init__(