Skip to content

Commit 5a2c905

Browse files
committed
pr feedback
1 parent acf99ef commit 5a2c905

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

roboflow/core/workspace.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -287,7 +287,7 @@ def upload_dataset(
287287
project_type: str = "object-detection",
288288
batch_name=None,
289289
num_retries=0,
290-
are_predictions=False,
290+
is_prediction=False,
291291
):
292292
"""
293293
Upload a dataset to Roboflow.
@@ -301,7 +301,7 @@ def upload_dataset(
301301
project_type (str): type of the project (only `object-detection` is supported)
302302
batch_name (str, optional): name of the batch to upload the images to. Defaults to an automatically generated value.
303303
num_retries (int, optional): number of times to retry uploading an image if the upload fails. Defaults to 0.
304-
are_predictions (bool, optional): whether the annotations provided in the dataset are predictions and not ground truth. Defaults to False.
304+
is_prediction (bool, optional): whether the annotations provided in the dataset are predictions and not ground truth. Defaults to False.
305305
""" # noqa: E501 // docs
306306
if dataset_format != "NOT_USED":
307307
print("Warning: parameter 'dataset_format' is deprecated and will be removed in a future release")
@@ -356,7 +356,7 @@ def _upload_image(imagedesc):
356356
sequence_number=imagedesc.get("index"),
357357
sequence_size=len(images),
358358
num_retry_uploads=num_retries,
359-
is_prediction=are_predictions,
359+
is_prediction=is_prediction,
360360
)
361361

362362
return image, upload_time, upload_retry_attempts

tests/test_project.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -379,7 +379,7 @@ def test_project_upload_dataset(self):
379379
{"file": "pred1.jpg", "split": "train", "annotationfile": {"file": "pred1.xml"}},
380380
{"file": "pred2.jpg", "split": "valid", "annotationfile": {"file": "pred2.xml"}},
381381
],
382-
"params": {"are_predictions": True},
382+
"params": {"is_prediction": True},
383383
"assertions": {
384384
"upload": {"count": 2, "kwargs": {"is_prediction": True}},
385385
"save_annotation": {"count": 2},
@@ -390,7 +390,7 @@ def test_project_upload_dataset(self):
390390
"dataset": [
391391
{"file": "gt1.jpg", "split": "train", "annotationfile": {"file": "gt1.xml"}},
392392
],
393-
"params": {"are_predictions": False},
393+
"params": {"is_prediction": False},
394394
"assertions": {
395395
"upload": {"count": 1, "kwargs": {"is_prediction": False}},
396396
"save_annotation": {"count": 1},
@@ -402,7 +402,7 @@ def test_project_upload_dataset(self):
402402
{"file": "batch_pred.jpg", "split": "train", "annotationfile": {"file": "batch_pred.xml"}},
403403
],
404404
"params": {
405-
"are_predictions": True,
405+
"is_prediction": True,
406406
"batch_name": "prediction-batch",
407407
"num_retries": 2,
408408
},

0 commit comments

Comments
 (0)