From a6ec4fbf58c221047172b358c650d169b5b5a544 Mon Sep 17 00:00:00 2001 From: Alexander Soare Date: Wed, 3 Apr 2024 09:53:15 +0100 Subject: [PATCH] remove try-catch --- tests/test_datasets.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/tests/test_datasets.py b/tests/test_datasets.py index 05c9e80d4..df41b03f6 100644 --- a/tests/test_datasets.py +++ b/tests/test_datasets.py @@ -58,12 +58,9 @@ def test_compute_stats(): for k, pattern in buffer.stats_patterns.items(): expected_mean = einops.reduce(all_data[k], pattern, "mean") assert torch.allclose(computed_stats[k]["mean"], expected_mean) - try: - assert torch.allclose( - computed_stats[k]["std"], - torch.sqrt(einops.reduce((all_data[k] - expected_mean) ** 2, pattern, "mean")) - ) - except: - breakpoint() + assert torch.allclose( + computed_stats[k]["std"], + torch.sqrt(einops.reduce((all_data[k] - expected_mean) ** 2, pattern, "mean")) + ) assert torch.allclose(computed_stats[k]["min"], einops.reduce(all_data[k], pattern, "min")) assert torch.allclose(computed_stats[k]["max"], einops.reduce(all_data[k], pattern, "max"))