Some cleanup

This commit is contained in:
dario
2023-12-14 23:30:35 +01:00
parent 4a93fd4ee6
commit 9bfe61d9c7
3 changed files with 7 additions and 2 deletions

View File

@@ -5,7 +5,7 @@ from typing import Any, Tuple
class Transform:
def apply(self, t: float, x: np.array):
def apply(self, t: float, x: ArrayLike):
y = self(t, x)
assert x.shape == y.shape, "Transform has to maintain the array's shape."