Skip to content

Commit e4ff4f0

Browse files
authored
[Validate/Bugfix] Fix parameter passing for all configs (#284)
1 parent d282040 commit e4ff4f0

File tree

2 files changed

+24
-2
lines changed

2 files changed

+24
-2
lines changed

nucleus/validate/eval_functions/available_eval_functions.py

Lines changed: 23 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,9 @@
1010
from ..data_transfer_objects.eval_function import EvalFunctionEntry
1111
from ..errors import EvalFunctionNotAvailableError
1212

13+
# TODO(gunnar) split up into modules
14+
# pylint: disable=too-many-lines
15+
1316

1417
class PolygonIOUConfig(EvalFunctionConfig):
1518
def __call__(
@@ -66,6 +69,8 @@ def __call__(
6669
enforce_label_match=enforce_label_match,
6770
iou_threshold=iou_threshold,
6871
confidence_threshold=confidence_threshold,
72+
annotation_filters=annotation_filters,
73+
prediction_filters=prediction_filters,
6974
**kwargs,
7075
)
7176

@@ -123,6 +128,8 @@ def __call__(
123128
"""
124129
return super().__call__(
125130
iou_threshold=iou_threshold,
131+
annotation_filters=annotation_filters,
132+
prediction_filters=prediction_filters,
126133
**kwargs,
127134
)
128135

@@ -186,6 +193,8 @@ def __call__(
186193
enforce_label_match=enforce_label_match,
187194
iou_threshold=iou_threshold,
188195
confidence_threshold=confidence_threshold,
196+
annotation_filters=annotation_filters,
197+
prediction_filters=prediction_filters,
189198
**kwargs,
190199
)
191200

@@ -249,6 +258,8 @@ def __call__(
249258
enforce_label_match=enforce_label_match,
250259
iou_threshold=iou_threshold,
251260
confidence_threshold=confidence_threshold,
261+
annotation_filters=annotation_filters,
262+
prediction_filters=prediction_filters,
252263
**kwargs,
253264
)
254265

@@ -312,6 +323,8 @@ def __call__(
312323
enforce_label_match=enforce_label_match,
313324
iou_threshold=iou_threshold,
314325
confidence_threshold=confidence_threshold,
326+
annotation_filters=annotation_filters,
327+
prediction_filters=prediction_filters,
315328
**kwargs,
316329
)
317330

@@ -369,6 +382,8 @@ def __call__(
369382
"""
370383
return super().__call__(
371384
iou_threshold=iou_threshold,
385+
annotation_filters=annotation_filters,
386+
prediction_filters=prediction_filters,
372387
**kwargs,
373388
)
374389

@@ -432,6 +447,8 @@ def __call__(
432447
enforce_label_match=enforce_label_match,
433448
iou_threshold=iou_threshold,
434449
confidence_threshold=confidence_threshold,
450+
annotation_filters=annotation_filters,
451+
prediction_filters=prediction_filters,
435452
**kwargs,
436453
)
437454

@@ -495,6 +512,8 @@ def __call__(
495512
enforce_label_match=enforce_label_match,
496513
iou_threshold=iou_threshold,
497514
confidence_threshold=confidence_threshold,
515+
annotation_filters=annotation_filters,
516+
prediction_filters=prediction_filters,
498517
**kwargs,
499518
)
500519

@@ -794,7 +813,10 @@ def __call__(
794813
Finally, the most outer list combines these filters as a disjunction (OR).
795814
"""
796815
return super().__call__(
797-
confidence_threshold=confidence_threshold, f1_method=f1_method
816+
confidence_threshold=confidence_threshold,
817+
f1_method=f1_method,
818+
annotation_filters=annotation_filters,
819+
prediction_filters=prediction_filters,
798820
)
799821

800822
@classmethod

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ exclude = '''
2121

2222
[tool.poetry]
2323
name = "scale-nucleus"
24-
version = "0.10.2"
24+
version = "0.10.3"
2525
description = "The official Python client library for Nucleus, the Data Platform for AI"
2626
license = "MIT"
2727
authors = ["Scale AI Nucleus Team <nucleusapi@scaleapi.com>"]

0 commit comments

Comments
 (0)