
    	]jB                         d dl Z d dlZd dlmZmZ ej
                  j                  d        Zej
                  j                  d        Zy)    N)make_project	make_taskc                    t        t        ddd      | j                  d      }t        ddd	ii|       | j	                  d
|j
                  ddd      }|j                  dk(  sJ | j                  d|j
                   d      }t        j                  |j                        }t        |d         dk(  sJ |d   d   d   d   d   d   d   dk(  sJ |d   d   d   dk(  sJ y )NT9  
                <View>
                  <Text name="text" value="$text"></Text>
                  <Choices name="label" choice="single" toName="text">
                    <Choice value="label_A"></Choice>
                    <Choice value="label_B"></Choice>
                  </Choices>
                </View>"test_get_single_prediction_on_taskis_publishedlabel_configtitleFconfiguseruse_ml_backenddatatexttest 1/api/ml/ModelSinglez'http://test.ml.backend.for.sdk.com:9092projectr   urlr      /api/projects//nextpredictions   r   resultvaluechoiceslabel_Amodel_versionr   dictr   r   postidstatus_codegetjsonloadscontentlenbusiness_clientml_backend_for_test_predictr   responsepayloads        L/root/env/lib/python3.12/site-packages/label_studio/tests/ml/test_predict.pyr   r      s"    7
 !!G" v)*G4 ##zz"<
 $ H 3&&& ""^GJJ<u#EFHjj))*G w}%&!+++=!!$X.q1':9EaHIUUU=!!$_5FFF    c                 "   t        t        ddd      | j                  d      }t        ddd	ii|       | j	                  d
|j
                  ddd      }|j                  dk(  sJ | j                  d|j
                   d      }t        j                  |j                        }t        |d         dk(  sJ |d   d   d   d   d   d   d   dk(  sJ |d   d   d   dk(  sJ |d   d   d   d   d   d   d   dk(  sJ |d   d   d   dk(  sJ y )NTr   %test_get_multiple_predictions_on_taskr   Fr   r   r   r   r   ModelAz'http://test.ml.backend.for.sdk.com:9093r   r   r   r   r   r      r   r   r   r    r!   r"   r   label_BModelBr#   r-   s        r2   r5   r5   2   sp    :
 !!G" v)*G4 ##zz<
 $ H 3&&& ""^GJJ<u#EFHjj))*G w}%&!+++=!!$X.q1':9EaHIUUU=!!$_5AAA=!!$X.q1':9EaHIUUU=!!$_5AAAr3   )	r)   pytestlabel_studio.tests.utilsr   r   mark	django_dbr   r5    r3   r2   <module>r?      sO      < &G &GR (B (Br3   