From 20203d554fd29b56ba2002a6f91e6d73022757b5 Mon Sep 17 00:00:00 2001 From: Alexander Soare Date: Thu, 30 May 2024 17:18:59 +0100 Subject: [PATCH] revert script --- tests/test_datasets.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/tests/test_datasets.py b/tests/test_datasets.py index bd63e568..73e9d7bb 100644 --- a/tests/test_datasets.py +++ b/tests/test_datasets.py @@ -372,7 +372,3 @@ def test_aggregate_stats(): for agg_fn in ["mean", "min", "max"]: assert torch.allclose(stats[data_key][agg_fn], einops.reduce(data, "n -> 1", agg_fn)) assert torch.allclose(stats[data_key]["std"], torch.std(data, correction=0)) - - -if __name__ == "__main__": - test_multilerobotdataset_frames()