Reformat EL test.

This commit is contained in:
Raphael Mitsch 2022-12-12 14:05:41 +01:00
parent cb640abe81
commit df6e4ab055

View File

@ -1200,16 +1200,9 @@ def test_threshold(meet_threshold: bool, config: Dict[str, Any]):
"entity_linker", "entity_linker",
last=True, last=True,
config={ config={
**( "threshold": None if meet_threshold else 1.0,
{"threshold": None} # Prior for candidate may be 1.0, rendering the our test setting with threshold 1.0 useless otherwise.
if meet_threshold "incl_prior": meet_threshold,
else {
"threshold": 1.0,
# Prior for candidate may be 1.0, rendering the our test setting with threshold 1.0 useless
# otherwise.
"incl_prior": False,
}
),
"model": config, "model": config,
}, },
) )