@@ -188,7 +188,12 @@ def test_dense_int_numpy_to_sparse_label():
188
188
with tempfile .TemporaryFile () as f :
189
189
write_numpy_to_sparse_tensor (f , array , label_data )
190
190
f .seek (0 )
191
- for record_data , expected_data , expected_keys , label in zip (_read_recordio (f ), array_data , keys_data , label_data ):
191
+ for record_data , expected_data , expected_keys , label in zip (
192
+ _read_recordio (f ),
193
+ array_data ,
194
+ keys_data ,
195
+ label_data
196
+ ):
192
197
record = Record ()
193
198
record .ParseFromString (record_data )
194
199
assert record .features ["values" ].int32_tensor .values == expected_data
@@ -205,7 +210,12 @@ def test_dense_float32_numpy_to_sparse_label():
205
210
with tempfile .TemporaryFile () as f :
206
211
write_numpy_to_sparse_tensor (f , array , label_data )
207
212
f .seek (0 )
208
- for record_data , expected_data , expected_keys , label in zip (_read_recordio (f ), array_data , keys_data , label_data ):
213
+ for record_data , expected_data , expected_keys , label in zip (
214
+ _read_recordio (f ),
215
+ array_data ,
216
+ keys_data ,
217
+ label_data
218
+ ):
209
219
record = Record ()
210
220
record .ParseFromString (record_data )
211
221
assert record .features ["values" ].float32_tensor .values == expected_data
@@ -222,7 +232,12 @@ def test_dense_float64_numpy_to_sparse_label():
222
232
with tempfile .TemporaryFile () as f :
223
233
write_numpy_to_sparse_tensor (f , array , label_data )
224
234
f .seek (0 )
225
- for record_data , expected_data , expected_keys , label in zip (_read_recordio (f ), array_data , keys_data , label_data ):
235
+ for record_data , expected_data , expected_keys , label in zip (
236
+ _read_recordio (f ),
237
+ array_data ,
238
+ keys_data ,
239
+ label_data
240
+ ):
226
241
record = Record ()
227
242
record .ParseFromString (record_data )
228
243
assert record .features ["values" ].float64_tensor .values == expected_data
0 commit comments