Skip to content

Commit 4b21b1c

Browse files
committed
Completed at log and updated classification threshold default
1 parent f56d7f3 commit 4b21b1c

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

inference/models/inference.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -170,6 +170,7 @@ def initiate(self, inference_api_url=settings.INFERENCE_API_URL) -> None:
170170
self.log_error_and_set_status_failed("No external jobs created")
171171
self.status = InferenceJobStatus.FAILED
172172
self.updated_at = timezone.now()
173+
self.completed_at = timezone.now()
173174
self.save()
174175
return
175176

inference/utils/classification_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ def map_classification_to_tdamm_tags(classification_results, threshold=None):
1616
list: List of TDAMM tag values that exceed the threshold
1717
"""
1818
if threshold is None:
19-
threshold = float(getattr(settings, "TDAMM_CLASSIFICATION_THRESHOLD", 0.5))
19+
threshold = float(getattr(settings, "TDAMM_CLASSIFICATION_THRESHOLD"))
2020

2121
selected_tags = []
2222

0 commit comments

Comments
 (0)