diff --git a/yt_experiments/tiled_grid/tiled_grid.py b/yt_experiments/tiled_grid/tiled_grid.py index c6d2eef..d5d3031 100644 --- a/yt_experiments/tiled_grid/tiled_grid.py +++ b/yt_experiments/tiled_grid/tiled_grid.py @@ -11,9 +11,10 @@ def _validate_edge(edge: npt.ArrayLike, ds: Dataset): if not isinstance(edge, unyt.unyt_array): - return ds.arr(edge, 'code_length') + return ds.arr(edge, "code_length") return edge + def _validate_nd_int(nd: int, x: int | npt.ArrayLike) -> npt.NDArray: if isinstance(x, int): x = (x,) * nd @@ -22,6 +23,7 @@ def _validate_nd_int(nd: int, x: int | npt.ArrayLike) -> npt.NDArray: raise ValueError("Variable must have a length of 3") return x + class YTTiledArbitraryGrid: _ndim = 3