fixed assert in utils test of reporter now that data is not transposed.

choderalab · Mar 23, 2024 · bb58931 · bb58931
1 parent 92c84cf
commit bb58931
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/chiron/tests/test_utils.py b/chiron/tests/test_utils.py
@@ -133,5 +133,5 @@ def test_reporter(prep_temp_dir, ho_multistate_sampler_multiple_ks):
         "step",
     ]
     u_kn = ho_sampler._reporter.get_property("u_kn")
-    assert u_kn.shape == (4, 4, 6)
+    assert u_kn.shape == (6, 4, 4)
     assert os.path.exists(ho_sampler._reporter.log_file_path)