Skip to content

Commit 34717a9

Browse files
author
Val Brodsky
committed
Remove all appearances of as_list() method
1 parent a7e8005 commit 34717a9

File tree

4 files changed

+6
-6
lines changed

4 files changed

+6
-6
lines changed

tests/data/annotation_types/test_collection.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ def test_generator(list_of_labels):
6161

6262
def test_conversion(list_of_labels):
6363
generator = LabelGenerator(list_of_labels)
64-
label_collection = generator.as_list()
64+
label_collection = list(generator)
6565
assert len(label_collection) == len(list_of_labels)
6666
assert [x for x in label_collection] == list_of_labels
6767

tests/data/serialization/labelbox_v1/test_tiled_image.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ def test_image(file_path):
2020
payload = json.load(f)
2121

2222
collection = LBV1Converter.deserialize(payload)
23-
collection_as_list = collection.as_list()
23+
collection_as_list = list(collection)
2424

2525
assert len(collection_as_list) == 2
2626

tests/data/serialization/ndjson/test_classification.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ def test_classification():
99
with open('tests/data/assets/ndjson/classification_import.json',
1010
'r') as file:
1111
data = json.load(file)
12-
res = NDJsonConverter.deserialize(data).as_list()
12+
res = list(NDJsonConverter.deserialize(data))
1313
res = list(NDJsonConverter.serialize(res))
1414
assert res == data
1515

@@ -18,6 +18,6 @@ def test_classification_with_name():
1818
with open('tests/data/assets/ndjson/classification_import_name_only.json',
1919
'r') as file:
2020
data = json.load(file)
21-
res = NDJsonConverter.deserialize(data).as_list()
21+
res = list(NDJsonConverter.deserialize(data))
2222
res = list(NDJsonConverter.serialize(res))
2323
assert res == data

tests/data/serialization/ndjson/test_document.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ def test_pdf():
2222
"""
2323
with open('tests/data/assets/ndjson/pdf_import.json', 'r') as f:
2424
data = json.load(f)
25-
res = NDJsonConverter.deserialize(data).as_list()
25+
res = list(NDJsonConverter.deserialize(data))
2626
res = list(NDJsonConverter.serialize(res))
2727
assert [round_dict(x) for x in res] == [round_dict(x) for x in data]
2828
f.close()
@@ -34,7 +34,7 @@ def test_pdf_with_name_only():
3434
"""
3535
with open('tests/data/assets/ndjson/pdf_import_name_only.json', 'r') as f:
3636
data = json.load(f)
37-
res = NDJsonConverter.deserialize(data).as_list()
37+
res = list(NDJsonConverter.deserialize(data))
3838
res = list(NDJsonConverter.serialize(res))
3939
assert [round_dict(x) for x in res] == [round_dict(x) for x in data]
4040
f.close()

0 commit comments

Comments
 (0)