From 359d86b937f17cec069887f5c2c7472889cb455b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Fabian=20Fr=C3=B6hlich?= Date: Mon, 3 Feb 2025 14:33:31 +0000 Subject: [PATCH] fixup petab 0006 jax --- tests/petab_test_suite/test_petab_suite.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/petab_test_suite/test_petab_suite.py b/tests/petab_test_suite/test_petab_suite.py index 88c914b43a..d79140cd54 100755 --- a/tests/petab_test_suite/test_petab_suite.py +++ b/tests/petab_test_suite/test_petab_suite.py @@ -115,7 +115,7 @@ def _test_case(case, model_type, version, jax): gt_chi2 = solution[petabtests.CHI2] gt_llh = solution[petabtests.LLH] gt_simulation_dfs = solution[petabtests.SIMULATION_DFS] - if case.startswith("0006"): + if case.startswith("0006") and not jax: # account for flattening gt_simulation_dfs[0].loc[:, petab.OBSERVABLE_ID] = ( "obs_a__10__c0",