revert script
This commit is contained in:
parent
00cefb44bc
commit
20203d554f
|
@ -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()
|
||||
|
|
Loading…
Reference in New Issue