diff --git a/test/test_speed_v_torch.py b/test/test_speed_v_torch.py index 5742c93040..dd16454434 100644 --- a/test/test_speed_v_torch.py +++ b/test/test_speed_v_torch.py @@ -7,10 +7,10 @@ import time import numpy as np np.set_printoptions(linewidth=160) from functools import partial -from tinygrad.ops import GlobalCounters, DEBUG +from tinygrad.ops import GlobalCounters from tinygrad.tensor import Tensor from tinygrad.nn import Conv2d -from tinygrad.helpers import colored, getenv +from tinygrad.helpers import colored, getenv, DEBUG from extra.jit import TinyJit METAL = getenv("METAL") try: