Add longdouble type

caniko · Dec 5, 2022 · febc2fe · febc2fe
1 parent 5ac44da
commit febc2fe
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/pydantic_numpy/dtype.py b/pydantic_numpy/dtype.py
@@ -26,7 +26,14 @@ def validate(cls, val: Any, field: ModelField):
         return val
 
 
-class double(np.float64, _BaseDType):
+class longdouble(np.longdouble, _BaseDType):
+    pass
+
+
+float128 = longdouble
+
+
+class double(np.double, _BaseDType):
     pass
 
 
@@ -105,6 +112,7 @@ class uint8(np.uint8, _BaseDType):
 
 # NDArray typings
 
+NDArrayFp128 = NDArray[float128]
 NDArrayFp64 = NDArray[float64]
 NDArrayFp32 = NDArray[float32]
 NDArrayFp16 = NDArray[float16]