Skip to content

Commit 572b0f2

Browse files
author
Egor Malykh
committed
fixed long lines
1 parent a7aed45 commit 572b0f2

File tree

1 file changed

+18
-3
lines changed

1 file changed

+18
-3
lines changed

tests/unit/test_common.py

Lines changed: 18 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -188,7 +188,12 @@ def test_dense_int_numpy_to_sparse_label():
188188
with tempfile.TemporaryFile() as f:
189189
write_numpy_to_sparse_tensor(f, array, label_data)
190190
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+
):
192197
record = Record()
193198
record.ParseFromString(record_data)
194199
assert record.features["values"].int32_tensor.values == expected_data
@@ -205,7 +210,12 @@ def test_dense_float32_numpy_to_sparse_label():
205210
with tempfile.TemporaryFile() as f:
206211
write_numpy_to_sparse_tensor(f, array, label_data)
207212
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+
):
209219
record = Record()
210220
record.ParseFromString(record_data)
211221
assert record.features["values"].float32_tensor.values == expected_data
@@ -222,7 +232,12 @@ def test_dense_float64_numpy_to_sparse_label():
222232
with tempfile.TemporaryFile() as f:
223233
write_numpy_to_sparse_tensor(f, array, label_data)
224234
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+
):
226241
record = Record()
227242
record.ParseFromString(record_data)
228243
assert record.features["values"].float64_tensor.values == expected_data

0 commit comments

Comments
 (0)