Skip to content

Commit

Permalink
[FIX] darglint
Browse files Browse the repository at this point in the history
  • Loading branch information
f-dangel committed Nov 3, 2023
1 parent 25c422f commit 5edd54f
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions einconv/simplifications/symbolic.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,9 @@ def __init__(self, name: str, shape: Tuple[int, ...], indices: Tuple[str, ...]):
name: Tensor name.
shape: Tensor dimensions.
indices: Name of each axis.
Raises:
ValueError: If the number of dimensions does not match the number of axes.
"""
if len(shape) != len(indices):
raise ValueError(
Expand All @@ -31,9 +34,11 @@ def __init__(self, name: str, shape: Tuple[int, ...], indices: Tuple[str, ...]):
self.shape = shape
self.indices = indices

# human-readable record of transformations
self.history: List[Tuple[str, Tuple[int, ...], Tuple[str, ...]]] = [
("initial", shape, indices)
]
# record of functional transformations
self.transforms: List[Callable[[Tensor], Tensor]] = []

def instantiate(
Expand Down

0 comments on commit 5edd54f

Please sign in to comment.