fix: fix test warnings

metaopt · May 10, 2024 · 099f07b · 099f07b
1 parent 945a3b7
commit 099f07b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -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)