Skip to content

Commit 6f1a7b8

Browse files
authored
Fix deprecation warnings about Traitlets tag (#50)
1 parent 8a59f59 commit 6f1a7b8

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

traittypes/traittypes.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -187,7 +187,8 @@ def __init__(self, default_value=Empty, allow_none=False, dtype=None, **kwargs):
187187
import pandas as pd
188188
kwargs['klass'] = pd.DataFrame
189189
super(DataFrame, self).__init__(
190-
default_value=default_value, allow_none=allow_none, dtype=dtype, **kwargs)
190+
default_value=default_value, allow_none=allow_none, **kwargs)
191+
self.tag(dtype=dtype)
191192

192193

193194
class Series(PandasType):
@@ -202,7 +203,8 @@ def __init__(self, default_value=Empty, allow_none=False, dtype=None, **kwargs):
202203
import pandas as pd
203204
kwargs['klass'] = pd.Series
204205
super(Series, self).__init__(
205-
default_value=default_value, allow_none=allow_none, dtype=dtype, **kwargs)
206+
default_value=default_value, allow_none=allow_none, **kwargs)
207+
self.tag(dtype=dtype)
206208
self.dtype = dtype
207209

208210

@@ -266,7 +268,8 @@ def __init__(self, default_value=Empty, allow_none=False, dtype=None, **kwargs):
266268
import xarray as xr
267269
kwargs['klass'] = xr.Dataset
268270
super(Dataset, self).__init__(
269-
default_value=default_value, allow_none=allow_none, dtype=dtype, **kwargs)
271+
default_value=default_value, allow_none=allow_none, **kwargs)
272+
self.tag(dtype=dtype)
270273

271274

272275
class DataArray(XarrayType):
@@ -281,5 +284,6 @@ def __init__(self, default_value=Empty, allow_none=False, dtype=None, **kwargs):
281284
import xarray as xr
282285
kwargs['klass'] = xr.DataArray
283286
super(DataArray, self).__init__(
284-
default_value=default_value, allow_none=allow_none, dtype=dtype, **kwargs)
287+
default_value=default_value, allow_none=allow_none, **kwargs)
288+
self.tag(dtype=dtype)
285289
self.dtype = dtype

0 commit comments

Comments
 (0)