Traceback (most recent call last):
  File "/projects/bfqt/users/yurenh2/ml-projects/snn-training/files/experiments/scaled_reg_grid_search.py", line 45, in <module>
    from depth_scaling_benchmark import (
ImportError: cannot import name 'compute_gradient_stats' from 'depth_scaling_benchmark' (/projects/bfqt/users/yurenh2/ml-projects/snn-training/files/experiments/depth_scaling_benchmark.py)