diff options
author | Hernan Grecco <hgrecco@gmail.com> | 2023-04-30 13:30:07 -0300 |
---|---|---|
committer | Hernan Grecco <hgrecco@gmail.com> | 2023-04-30 13:30:07 -0300 |
commit | f536abcfbb823ba55a35922da94b26ecc5f6d7b8 (patch) | |
tree | 5be425e63950527c5b0c53cd73512c3975cd4943 /pint/facets/dask/__init__.py | |
parent | 5d3d6299c9cfef5b62366482590edd3c241d9944 (diff) | |
download | pint-f536abcfbb823ba55a35922da94b26ecc5f6d7b8.tar.gz |
Reworked the Quantity, Unit, Measurement, Group and System class to be static to ease with typing
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 |