Skip to content

Commit

Permalink
Fusion: remove NoneType import due to CI Python version
Browse files Browse the repository at this point in the history
  • Loading branch information
NaderAlAwar committed Jan 3, 2024
1 parent 7e11d97 commit 50f136e
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions pykokkos/core/fusion/trace.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
import ast
from dataclasses import dataclass
from types import NoneType
from typing import Any, Callable, Dict, List, Optional, Set, Tuple, Union

from pykokkos.core.parsers import Parser, PyKokkosEntity
Expand Down Expand Up @@ -71,7 +70,7 @@ def __init__(self) -> None:
self.op_id: int = 0

# This functions as an ordered set
self.operations: Dict[TracerOperation, NoneType] = {}
self.operations: Dict[TracerOperation, None] = {}

# Map from each data object id (future or array) to the current version
self.data_version: Dict[int, int] = {}
Expand Down

0 comments on commit 50f136e

Please sign in to comment.