Setting the JAX env to only use CPUs

BQSKit · Jul 2, 2024 · 759d13d · 759d13d
1 parent f5c4a00
commit 759d13d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/setup.cfg b/setup.cfg
@@ -1,6 +1,6 @@
 [metadata]
 name = bqskit-qfactor-jax
-version = 0.0.1
+version = 1.0.0
 description = QFactor GPU implementation in BQSKit using JAX
 long_description = file: README.md
 long_description_content_type = text/markdown
@@ -59,6 +59,9 @@ testpaths = tests/
 python_files = test*.py
 norecursedirs = .env
 
+[pytestenv]
+JAX_PLATFORMS = cpu
+
 [mypy]
 check_untyped_defs = true
 disallow_any_generics = true