summaryrefslogtreecommitdiff
path: root/pint/facets/dask/__init__.py
diff options
context:
space:
mode:
authorHernan Grecco <hgrecco@gmail.com>2023-04-30 13:30:07 -0300
committerHernan Grecco <hgrecco@gmail.com>2023-04-30 13:30:07 -0300
commitf536abcfbb823ba55a35922da94b26ecc5f6d7b8 (patch)
tree5be425e63950527c5b0c53cd73512c3975cd4943 /pint/facets/dask/__init__.py
parent5d3d6299c9cfef5b62366482590edd3c241d9944 (diff)
downloadpint-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__.py6
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