Skip to content

Commit 2e82afe

Browse files
dshabinVaghinakDev
authored andcommitted
Add clean priority
1 parent e0cad32 commit 2e82afe

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
@@ -645,13 +645,26 @@ def __init__(
645645
self._client = backend_service_provider
646646
self._project_folder_name = project_folder_name
647647

648+
@staticmethod
649+
def get_clean_priority(priority):
650+
if len(str(priority)) > 8:
651+
priority = float(str(priority)[:8])
652+
if priority > 1000000:
653+
priority = 1000000
654+
if priority < 0:
655+
priority = 0
656+
if str(float(priority)).split('.')[1:2]:
657+
if len(str(float(priority)).split('.')[1]) > 5:
658+
priority = float(str(float(priority)).split('.')[0] + '.' + str(float(priority)).split('.')[1][:5])
659+
return priority
660+
648661
def execute(self):
649662
priorities = []
650663
to_send = []
651664
for i in self._scores:
652665
priorities.append({
653666
"name": i.name,
654-
"entropy_value": i.priority
667+
"entropy_value": self.get_clean_priority(i.priority)
655668
})
656669
to_send.append(i.name)
657670

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)