diff --git a/labelstower/getter.py b/labelstower/getter.py index b63370983f92a8e3b46bb28e802bb2052817f7e2..f6f411c064e8b64d2d3bd3b5df69489bfe4c4348 100644 --- a/labelstower/getter.py +++ b/labelstower/getter.py @@ -14,13 +14,13 @@ def get_selected_elements(id_mandatory_labels: int, id_forbiden_labels: str, num EXEMPLE ------- >>> print(get_selected_elements("","",0)) # si aucun labels n'est interdit ou obligatoire - [{'id': 4, 'name': 'peugeot, 2000'}, {'id': 3, 'name': 'renault, 2000'}, {'id': 2, 'name': 'renault, bleu, 2000'}, {'id': 1, 'name': 'renault, rouge, 2000'}] + [{'id': 4, 'element_name': 'peugeot, 2000'}, {'id': 3, 'element_name': 'renault, 2000'}, {'id': 2, 'element_name': 'renault, bleu, 2000'}, {'id': 1, 'element_name': 'renault, rouge, 2000'}] >>> print(get_selected_elements("1,2","5,9",2)) # si la voiture doit-être bleu - [{'id': 4, 'name': 'peugeot, 2000'}] + [{'id': 4, 'element_name': 'peugeot, 2000'}] >>> print(get_selected_elements("5","",1)) - [{'id': 3, 'name': 'renault, 2000'}, {'id': 2, 'name': 'renault, bleu, 2000'}, {'id': 1, 'name': 'renault, rouge, 2000'}] + [{'id': 3, 'element_name': 'renault, 2000'}, {'id': 2, 'element_name': 'renault, bleu, 2000'}, {'id': 1, 'element_name': 'renault, rouge, 2000'}] >>> print(get_selected_elements("","5",0)) - [{'id': 4, 'name': 'peugeot, 2000'}] + [{'id': 4, 'element_name': 'peugeot, 2000'}] """ #query slightly vary if number_of_mandatory_labels > 0 or not if(number_of_mandatory_labels): @@ -84,7 +84,7 @@ def get_discriminating_labels(selected_elements: List[Dict[int, str]]) -> List[D EXEMPLE ------- >>> get_discriminating_labels([{"id":2,"name":"renault, bleu, 2000"}, {"id":1,"name":"renault, rouge, 2000"}]) - [{'id': 8, 'name': 'rouge'}, {'id': 9, 'name': 'bleu'}] + [{'id': 8, 'label_name': 'rouge'}, {'id': 9, 'label_name': 'bleu'}] >>> get_discriminating_labels([]) [] """ @@ -122,7 +122,7 @@ def get_high_discriminating_labels(discriminating_labels: List[Dict[int, str]]) EXEMPLE ------- >>> get_high_discriminating_labels([{"id":7, "name":"couleur"}, {"id":8, "name":"rouge"}, {"id":9, "name":"bleu"}]) - [{'id': 7, 'name': 'couleur'}] + [{'id': 7, 'label_name': 'couleur'}] """ id_discriminating_labels = get_ids(discriminating_labels)