fix: made the to method optional on the encoder

src/seb/interfaces/model.py

@@ -1,4 +1,5 @@
 import json
+import logging
 from dataclasses import dataclass
 from datetime import date
 from pathlib import Path
@@ -14,6 +15,9 @@
     from .task import Task
 
 
+logger = logging.getLogger(__name__)
+
+
 @runtime_checkable
 class Encoder(Protocol):
     """
@@ -42,10 +46,10 @@ def encode(
         """
         ...
 
-    def to(self, device: torch.device):
-        ...
-
     # The following methods are optional and can be implemented if the model supports them.
+    # def to(self, device: torch.device):
+    #     ...
+
     # def encode_queries(self, queries: list[str], **kwargs: Any) -> np.ndarray:
     #     ...
 
@@ -109,7 +113,10 @@ def load_model(self):
 
     def to(self, device: torch.device):
         self.load_model()
-        self._model = self._model.to(device)  # type: ignore
+        try:
+            self._model = self._model.to(device)  # type: ignore
+        except AttributeError:
+            logging.debug(f"Model {self._model} does not have a to method")
 
     @property
     def model(self) -> Encoder:

tests/cli/test_cli.py

@@ -4,7 +4,6 @@
 
 import numpy as np
 import pytest
-
 import seb
 from seb.cli import cli, run_benchmark_cli