diff options
Diffstat (limited to 'pint/facets/dask/__init__.py')
| -rw-r--r-- | pint/facets/dask/__init__.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/pint/facets/dask/__init__.py b/pint/facets/dask/__init__.py index e0821be..90c8972 100644 --- a/pint/facets/dask/__init__.py +++ b/pint/facets/dask/__init__.py @@ -14,7 +14,7 @@ from __future__ import annotations import functools from ...compat import compute, dask_array, persist, visualize -from ..plain import PlainRegistry +from ..plain import PlainRegistry, PlainQuantity def check_dask_array(f): @@ -31,7 +31,7 @@ def check_dask_array(f): return wrapper -class DaskQuantity: +class DaskQuantity(PlainQuantity): # Dask.array.Array ducking def __dask_graph__(self): if isinstance(self._magnitude, dask_array.Array): @@ -120,4 +120,4 @@ class DaskQuantity: class DaskRegistry(PlainRegistry): - _quantity_class = DaskQuantity + Quantity = DaskQuantity |