Switch to native traceback

rapidsai · Sep 23, 2024 · c529a1f · c529a1f
1 parent 7de8831
commit c529a1f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/python/cuml/cuml/benchmark/automated/pytest.ini b/python/cuml/cuml/benchmark/automated/pytest.ini
@@ -4,7 +4,7 @@ addopts =
           --benchmark-warmup-iterations=1
           --benchmark-min-rounds=3
           --benchmark-columns="min, max, mean, stddev, outliers, gpu_mem, rounds"
-
+          --tb=native
 markers =
           managedmem_on: RMM managed memory enabled
           managedmem_off: RMM managed memory disabled

diff --git a/python/cuml/cuml/tests/pytest.ini b/python/cuml/cuml/tests/pytest.ini
@@ -0,0 +1,5 @@
+# Copyright (c) 2024, NVIDIA CORPORATION.
+
+[pytest]
+addopts = --tb=native
+