Skip to content

Commit a085d83

Browse files
committed
Fix tests
1 parent ced0897 commit a085d83

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

tests/integration/annotation_import/conftest.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -141,14 +141,17 @@ def configured_project(client, ontology, rand_gen, image_url):
141141

142142

143143
@pytest.fixture
144-
def configured_project_without_data_rows(client, ontology, rand_gen):
144+
def configured_project_pdf(client, ontology, rand_gen, pdf_url):
145145
project = client.create_project(name=rand_gen(str))
146146
dataset = client.create_dataset(name=rand_gen(str))
147147
editor = list(
148148
client.get_labeling_frontends(
149149
where=LabelingFrontend.name == "editor"))[0]
150150
project.setup(editor, ontology)
151-
project.update(queue_mode=project.QueueMode.Batch)
151+
data_row_ids = []
152+
data_row_ids.append(dataset.create_data_row(row_data=pdf_url).uid)
153+
project.datasets.connect(dataset)
154+
project.data_row_ids = data_row_ids
152155
yield project
153156
project.delete()
154157
dataset.delete()

0 commit comments

Comments
 (0)