From 56d90d0e6c1838287af045f929a18771a72e796d Mon Sep 17 00:00:00 2001
From: Kashif Rasul <kashif.rasul@zalando.de>
Date: Mon, 15 Jul 2019 13:08:22 +0200
Subject: [PATCH] np dtype

---
 pts/feature/transform.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pts/feature/transform.py b/pts/feature/transform.py
index fa0cf5f..99e1a32 100644
--- a/pts/feature/transform.py
+++ b/pts/feature/transform.py
@@ -219,7 +219,7 @@ class AsNumpyArray(SimpleTransformation):
     """
 
     def __init__(
-        self, field: str, expected_ndim: int, dtype: DType = np.float32
+        self, field: str, expected_ndim: int, dtype: np.dtype = np.float32
     ) -> None:
         self.field = field
         self.expected_ndim = expected_ndim
@@ -501,7 +501,7 @@ class AddConstFeature(MapTransformation):
         target_field: str,
         pred_length: int,
         const: float = 1.0,
-        dtype: DType = np.float32,
+        dtype: np.dtype = np.float32,
     ) -> None:
         self.pred_length = pred_length
         self.const = const