fix to pyproject.toml

priyakasimbeg · priyakasimbeg · commit e23e99acad1f · 2025-05-28T18:33:15.000Z
diff --git a/pyproject.toml b/pyproject.toml
@@ -114,7 +114,7 @@ jax_gpu = [
   "jax",
   "jaxlib",
   "jax-cuda12-plugin[with-cuda]",
-  "jax-cuda12-pjrt"
+  "jax-cuda12-pjrt",
   # "jax[cuda12]==0.6.0", 
   "algoperf[jax_core_deps]",
 ]
