diff --git a/tile_engine/ops/reduce/reduce_parameter.py b/tile_engine/ops/reduce/reduce_parameter.py index 4558dfe065..bdb0b9cfa6 100644 --- a/tile_engine/ops/reduce/reduce_parameter.py +++ b/tile_engine/ops/reduce/reduce_parameter.py @@ -4,7 +4,7 @@ from dataclasses import dataclass from itertools import product -from pyparsing import List +from typing import List TYPE_MAP = {"fp16": "ck_tile::half_t", "float": "float"}