fix test

BiomedSciAI · Jun 18, 2024 · aaf8499 · aaf8499
1 parent e16af1d
commit aaf8499
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fuse/eval/tests/test_eval.py b/fuse/eval/tests/test_eval.py
@@ -245,7 +245,7 @@ def test_eval_example_seq_gen_2(self) -> None:
 
  def test_pearson_correlation(self) -> None:
  res = example_pearson_correlation()
- self.assertAlmostEqual(res["metrics.pearsonr"], 1.0, places=2)
+ self.assertAlmostEqual(res["metrics.pearsonr.statistic"], 1.0, places=2)
 
 
 if __name__ == "__main__":