From 099f07bb67df05e1112126857745322da2e4e3ff Mon Sep 17 00:00:00 2001
From: Xuehai Pan <XuehaiPan@pku.edu.cn>
Date: Fri, 10 May 2024 08:09:39 +0000
Subject: [PATCH] fix: fix test warnings

---
 pyproject.toml   | 1 +
 tests/helpers.py | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index a19e84f4..17d14b05 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -316,4 +316,5 @@ filterwarnings = [
     'ignore:jax\.numpy\.DeviceArray is deprecated\. Use jax\.Array\.:DeprecationWarning',
     'ignore:.*functorch.*deprecate.*:UserWarning',
     'ignore:.*Apple Paravirtual device.*:UserWarning',
+    'ignore:.*NVML.*:UserWarning',
 ]
diff --git a/tests/helpers.py b/tests/helpers.py
index 1b624ce9..0dc415d4 100644
--- a/tests/helpers.py
+++ b/tests/helpers.py
@@ -65,7 +65,7 @@ def parametrize(**argvalues) -> pytest.mark.parametrize:
         argvalues = list(itertools.product(*tuple(map(argvalues.get, arguments))))
 
     ids = tuple(
-        '-'.join(f'{arg}({val})' for arg, val in zip(arguments, values)) for values in argvalues
+        '-'.join(f'{arg}({val!r})' for arg, val in zip(arguments, values)) for values in argvalues
     )
 
     return pytest.mark.parametrize(arguments, argvalues, ids=ids)