wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-09-11 12:42:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

np dtype

Browse Source
This commit is contained in:
Kashif Rasul
2019-07-15 13:08:22 +02:00
parent f59a7e1506
commit 56d90d0e6c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/feature/transform.py
+2 -2
View File
@@ -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