Only create profiling trace

OpenSourceEconomics · Feb 5, 2024 · 1138dca · 1138dca
1 parent fb133e0
commit 1138dca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test_long.py b/test_long.py
@@ -33,7 +33,7 @@ def test_analytical_solution(model_name, model_config):
     }
 
 
-with jax.profiler.trace("/tmp/jax-trace", create_perfetto_link=True):
+with jax.profiler.trace("/tmp/jax-trace", create_perfetto_trace=True):
     test_analytical_solution(
         "iskhakov_2017_five_periods",
         get_model("iskhakov_2017_five_periods"),