hotfix: remove useless slow assert from ShapeTracker

This commit is contained in:
George Hotz
2024-01-09 10:56:36 -08:00
parent 4b687af98f
commit bf6281f316

View File

@@ -54,8 +54,6 @@ def idxs_to_idx(shape:Tuple[int, ...], idxs:Tuple[Node, ...]) -> Node:
@dataclass(frozen=True)
class ShapeTracker:
views: Tuple[View, ...]
def __post_init__(self):
assert isinstance(self.views, tuple) and all(isinstance(v, View) for v in self.views), "ShapeTracker must be created with a tuple of Views"
def __add__(self, st:ShapeTracker) -> ShapeTracker:
base = ShapeTracker(self.views)