fix: add fallback for missing cpu_count

superlinear-ai · Sep 23, 2024 · fa8d39d · fa8d39d
1 parent e11d880
commit fa8d39d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_rag.py b/tests/test_rag.py
@@ -11,7 +11,7 @@
 
 def is_accelerator_available() -> bool:
     """Check if an accelerator is available."""
-    return llama_supports_gpu_offload() or os.cpu_count() >= 8  # noqa: PLR2004
+    return llama_supports_gpu_offload() or (os.cpu_count() or 1) >= 8  # noqa: PLR2004
 
 
 def test_insert_index_search(raglite_test_config: RAGLiteConfig) -> None: