import torch def test_timer(benchmark_timer, seed): us = benchmark_timer(lambda: torch.ones(8).sum()) assert us > 0 and isinstance(seed, int)