specify CPU device in warm_up test (#7014)

deepset-ai · Feb 16, 2024 · 44b5ae2 · 44b5ae2
1 parent b552b0b
commit 44b5ae2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/components/audio/test_whisper_local.py b/test/components/audio/test_whisper_local.py
@@ -71,7 +71,7 @@ def test_from_dict(self):
 
  def test_warmup(self):
  with patch("haystack.components.audio.whisper_local.whisper") as mocked_whisper:
- transcriber = LocalWhisperTranscriber(model="large-v2")
+ transcriber = LocalWhisperTranscriber(model="large-v2", device=ComponentDevice.from_str("cpu"))
  mocked_whisper.load_model.assert_not_called()
  transcriber.warm_up()
  mocked_whisper.load_model.assert_called_once_with("large-v2", device=torch.device(type="cpu"))