diff --git a/spatialmath/base/argcheck.py b/spatialmath/base/argcheck.py index 40f94336..1206d676 100644 --- a/spatialmath/base/argcheck.py +++ b/spatialmath/base/argcheck.py @@ -559,7 +559,7 @@ def getunit(v: ArrayLike, unit: str = "rad", dim=None) -> Union[float, NDArray]: :seealso: :func:`getvector` """ - if not isinstance(v, Iterable) and dim == 0: + if not isinstance(v, Iterable) and dim in (0, None): # scalar in, scalar out if unit == "rad": return v