Minor change

SeldonIO · Jul 26, 2023 · a3519f3 · a3519f3
1 parent 2d05b32
commit a3519f3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/alibi_detect/saving/schemas.py b/alibi_detect/saving/schemas.py
@@ -93,7 +93,7 @@ def validate_device(cls, device: Any, values: dict) -> Any:
         backend = values['backend']
         if backend == Framework.TENSORFLOW or backend == Framework.SKLEARN:
             if device is not None:
-                raise TypeError('`device` should not be specified for TensorFlow or Sklearn models. Leave as `None`.')
+                raise TypeError('`device` should not be specified for TensorFlow or Sklearn backends. Leave as `None`.')
             else:
                 return device
         elif backend == Framework.PYTORCH or backend == Framework.KEOPS: