
    	]j                         d dl Z d dlmZ d dlmZ e j
                  j                  d        Ze j
                  j                  d        Zy)    N)LabelStudio)LabelInterfacec           	      r   t        | |j                        }t        j                  dddddfdddd	d
dddidfdddidfffd      }|j                  j                  d|      }|j                  j                  |j                  dddiidddiidddiig       |j                  j                  |j                        D cg c]  }| }}t        |      dk(  sJ |j                  j                  d|j                  d       |j                  j                  |j                        }t        |      dk(  sJ |d   j                  dk(  sJ |d   j                  dk(  sJ |j                  j                  d|j                  d|d   j                  gd !       |j                  j                  |j                        }	t        |	      d"k(  sJ |	d   j                  d   d   d#   d   dk(  sJ |	d   j                   dk(  sJ |	d   j                  d   d   d#   d   dk(  sJ |	d   j                   dk(  sJ |j                  j                  d$|j                  d%|	d   j"                  |	d   j"                  |d   j                  gd&!       |j                  j                  |j                  d'(      D ]  }|j                  |d   j                  k(  r|j$                  rJ |j                  s:J t        |j$                        dk(  sJ |j$                  d   d)   d   d   d#   d   dk(  sJ t        |j                        dk(  sJ |j                  d   j                  d   d   d#   d   dk(  sJ |j                  d   j                   dk(  sJ |j                  d   j&                  d*k(  rJ  y c c}w )+Nbase_urlapi_keyTexttextz$text)namevalue Choiceslabelsingle)r   toNamechoiceChoicer   label_Alabel_B)r
   r   New Projecttitlelabel_configdatatest 1test 2test 3requestproject   z'http://test.ml.backend.for.sdk.com:9092ModelSingleurlr!   r      r   retrieve_tasks_predictionsT)allexcludedidr!   selected_items   choicespredictions_to_annotationsFr(   includedr(   r!   fieldsresultg?)r   r   r   createprojectsimport_tasksr+   taskslistlenmlr   r%   actionspredictionsr4   model_versiontaskannotationsscore)
django_live_urlbusiness_clientml_backend_for_test_predictlsr   pr?   r8   
ml_backendr=   s
             H/root/env/lib/python3.12/site-packages/label_studio/tests/sdk/test_ml.py1test_batch_predictions_single_prediction_per_taskrI      s   	o7N7N	OB!((fw?D FhG3R83R8
	
L 	! 	 	A KK	fh'(fh'(fh'(
   !hhmmADDm9:dT:E:u:?? EELL>TaLb ADD)Jz?aa=-///a= IIII JJ'#%(++?   ..%%add%3K {q    q>  #G,Y7:iGGGq>''=888 q>  #G,Y7:iGGGq>''=888 JJ'A##A##a	
   add59 477eAhkk!''''''''t''(A---##A&x03G<YGJiWWWt''(A---##A&--a09)DQG9TTT##A&44EEE##A&,,3334c ;s   	N4c                    t        | |j                        }t        d      }|j                  j	                  d|j
                        }|j                  j                  |j                  dddiidddiiddd	iig
       |j                  j                  |j                        D cg c]  }| }}t        |      dk(  sJ |j                  j	                  d|j                  d       |j                  j	                  d|j                  d|d   j                  |d   j                  gd       |j                  j                  |j                        }t        |      dk(  sJ |j                  j                  |j                  d      D ]  }|j                  |d   j                  k(  r|j                  s,J t        |j                        dk(  sJ t        |j                        D ]X  \  }	}
|
j                  d   d   d   d   dddg|	    k(  sJ |
j                   dd |	    k(  sJ |	dk(  r|
j"                  d!k(  rWJ Z  y c c}w )"Nr   a  
            <View>
              <Text name="text" value="$text"/>
              <Choices name="label" toName="text" choice="single">
                <Choice value="label_A"></Choice>
                <Choice value="label_B"></Choice>
              </Choices>
            </View>r   r   r   r
   r   r   r   r   r    r"   z'http://test.ml.backend.for.sdk.com:9093ModelMultipler$   r'   Fr   r-   r0   r*      r(   r2   r&   r   r.   label_ABModelABg?)r   r   r   r6   r5   _configr7   r+   r8   r9   r:   r;   r<   r=   	enumerater4   r>   rA   )rB   rC   rD   rE   lirF   r?   r8   r=   i
predictions              rH   4test_batch_predictions_multiple_predictions_per_taskrW   b   sZ    
o7N7N	OB			
B 	ZZ 	 	A KK	fh'(fh'(fh'(
   !hhmmADDm9:dT:E:u:?? EELL>TcLd JJ'$58;;a2LM   ..%%add%3K {q   add59 	B77eAhkk!''''t''(A---!*4+;+;!< B:!((+G4Y?BPSUXzZ[}oF^^^^!//U47)3DDDD23q&z''3.AAcB	B' ;s   (	I
)	pytestlabel_studio_sdk.clientr    label_studio_sdk.label_interfacer   mark	django_dbrI   rW   r       rH   <module>r^      sO     / ; X4 X4v 7B 7Br]   