File tree Expand file tree Collapse file tree 1 file changed +9
-7
lines changed Expand file tree Collapse file tree 1 file changed +9
-7
lines changed Original file line number Diff line number Diff line change @@ -2124,13 +2124,15 @@ def set_data_dtype(self, datatype):
2124
2124
>>> img.get_data_dtype() == np.dtype('int64')
2125
2125
True
2126
2126
"""
2127
- # Static aliases
2128
- if datatype == 'mask' :
2129
- datatype = 'u1'
2130
- # Dynamic aliases
2131
- elif datatype in ('compat' , 'smallest' ):
2132
- self ._dtype_alias = datatype
2133
- return
2127
+ # Numpy dtype comparison can fail in odd ways, check for aliases only if str
2128
+ if isinstance (datatype , str ):
2129
+ # Static aliases
2130
+ if datatype == 'mask' :
2131
+ datatype = 'u1'
2132
+ # Dynamic aliases
2133
+ elif datatype in ('compat' , 'smallest' ):
2134
+ self ._dtype_alias = datatype
2135
+ return
2134
2136
2135
2137
self ._dtype_alias = None
2136
2138
super ().set_data_dtype (datatype )
You can’t perform that action at this time.
0 commit comments