Skip to content

Commit 52de468

Browse files
committed
Add clean priority
1 parent 88cb53c commit 52de468

File tree

2 files changed

+23
-1
lines changed

2 files changed

+23
-1
lines changed

src/superannotate/lib/core/usecases/annotations.py

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -616,13 +616,26 @@ def __init__(
616616
self._client = backend_service_provider
617617
self._project_folder_name = project_folder_name
618618

619+
@staticmethod
620+
def get_clean_priority(priority):
621+
if len(str(priority)) > 8:
622+
priority = float(str(priority)[:8])
623+
if priority > 1000000:
624+
priority = 1000000
625+
if priority < 0:
626+
priority = 0
627+
if str(float(priority)).split('.')[1:2]:
628+
if len(str(float(priority)).split('.')[1]) > 5:
629+
priority = float(str(float(priority)).split('.')[0] + '.' + str(float(priority)).split('.')[1][:5])
630+
return priority
631+
619632
def execute(self):
620633
priorities = []
621634
to_send = []
622635
for i in self._scores:
623636
priorities.append({
624637
"name": i.name,
625-
"entropy_value": i.priority
638+
"entropy_value": self.get_clean_priority(i.priority)
626639
})
627640
to_send.append(i.name)
628641

tests/integration/test_upload_priority_scores.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,3 +32,12 @@ def test_upload_priority_scores(self):
3232
}])
3333
self.assertEqual(len(uploaded), 0)
3434
self.assertEqual(len(skipped), 2)
35+
uploaded, skipped = sa.upload_priority_scores(self.PROJECT_NAME, scores=[{
36+
"name": "example_image_3.jpg",
37+
"priority": 1.1234567890
38+
}, {
39+
"name": "example_image_4.jpg",
40+
"priority": 100000000
41+
}])
42+
self.assertEqual(sa.get_image_metadata(self.PROJECT_NAME,"example_image_4.jpg")['entropy_value'], 1000000)
43+
self.assertEqual(sa.get_image_metadata(self.PROJECT_NAME,"example_image_3.jpg")['entropy_value'], 1.12345)

0 commit comments

Comments
 (0)