From d2532390b6164e537b9dc41bea359860abac6c83 Mon Sep 17 00:00:00 2001 From: "Keaton J. Burns" Date: Mon, 15 Jan 2024 12:01:47 -0500 Subject: [PATCH] Fix remedy_scales to accept ndarray arguments --- dedalus/core/distributor.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/dedalus/core/distributor.py b/dedalus/core/distributor.py index 8c42d696..346080e1 100644 --- a/dedalus/core/distributor.py +++ b/dedalus/core/distributor.py @@ -8,6 +8,7 @@ import itertools from collections import OrderedDict from math import prod +import numbers from .coords import CoordinateSystem, DirectProduct from ..tools.array import reshape_vector @@ -185,11 +186,12 @@ def remedy_scales(self, scales): """Remedy different scale inputs.""" if scales is None: scales = 1 - if not isinstance(scales, (list, tuple)): - scales = [scales] * self.dim + if isinstance(scales, numbers.Number): + scales = (scales,) * self.dim + scales = tuple(scales) if 0 in scales: raise ValueError("Scales must be nonzero.") - return tuple(scales) + return scales def get_transform_object(self, axis): return self.transforms[axis]