diff --git a/tinygrad/codegen/opt/postrange.py b/tinygrad/codegen/opt/postrange.py index f7913b1007..cc2809efb4 100644 --- a/tinygrad/codegen/opt/postrange.py +++ b/tinygrad/codegen/opt/postrange.py @@ -1,4 +1,3 @@ -from __future__ import annotations import math, itertools from collections import defaultdict from typing import cast, Final diff --git a/tinygrad/engine/schedule.py b/tinygrad/engine/schedule.py index 201077be2b..16c6d8e20b 100644 --- a/tinygrad/engine/schedule.py +++ b/tinygrad/engine/schedule.py @@ -1,9 +1,7 @@ -from __future__ import annotations import time from typing import cast from collections import deque -from tinygrad.uop.ops import UOp, Ops, buffers, UOpMetaClass, track_rewrites -from tinygrad.uop.ops import PatternMatcher, UPat, graph_rewrite, graph_rewrite_map +from tinygrad.uop.ops import UOp, Ops, buffers, UOpMetaClass, track_rewrites, PatternMatcher, UPat, graph_rewrite, graph_rewrite_map from tinygrad.uop.spec import type_verify, tensor_spec from tinygrad.device import Buffer, MultiBuffer from tinygrad.helpers import DEBUG, cpu_profile, TracingKey, SPEC, flatten, pluralize