Skip to content

Commit

Permalink
Edit error message
Browse files Browse the repository at this point in the history
  • Loading branch information
wanliAlex committed Apr 23, 2024
1 parent 45061b1 commit 6144315
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion src/marqo/core/monitoring/monitoring.py
Original file line number Diff line number Diff line change
Expand Up @@ -183,4 +183,4 @@ def get_cuda_info(self) -> MarqoCudaInfoResponse:
cuda_devices.append(marqo_cuda_device_info)
return MarqoCudaInfoResponse(cuda_devices=cuda_devices)
else:
raise HardwareCompatabilityError("CUDA is not available on this machine")
raise HardwareCompatabilityError("CUDA is not available on this instance")
2 changes: 1 addition & 1 deletion tests/core/monitoring/test_monitoring.py
Original file line number Diff line number Diff line change
Expand Up @@ -371,7 +371,7 @@ def test_getCudaInfo_fail_on_cpu_instance(self):
"""
with self.assertRaises(HardwareCompatabilityError) as cm:
self.monitoring.get_cuda_info()
self.assertIn("CUDA is not available on this machine", cm.exception.message)
self.assertIn("CUDA is not available on this instance", cm.exception.message)

@patch('marqo.core.monitoring.monitoring.torch.cuda.is_available', return_value=True)
@patch('marqo.core.monitoring.monitoring.torch.cuda.device_count', return_value=1)
Expand Down

0 comments on commit 6144315

Please sign in to comment.