mirror of
https://git.intern.spaceteamaachen.de/ALPAKA/SPATZ.git
synced 2025-09-28 21:17:33 +00:00
Some cleanup
This commit is contained in:
@@ -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."
|
||||
|
||||
|
Reference in New Issue
Block a user