o
    HEDi                     @   sv   d dl Z d dlZd dlmZ d dlZd dlmZmZ d dlm	Z	m
Z
 ejje	 e
dejje edd	ddZdS )
    N)NamedTemporaryFile)	HAS_NUMPYNUMPY_INSTRUCTIONS)
HAS_PANDASPANDAS_INSTRUCTIONS)reasonreturnc               
   C   s   d} | d }d}|d }||d| |d||dg}t ddd3}t|j |D ]}|t|d  |  q(tjd	|j gtj	d
dtj	dd
d}W d   n1 sVw   Y  |j
dksbJ d|jv siJ |jS )z
    Ensures that long_examples_validator() handles previously applied recommendations,
    namely dropped duplicates, without resulting in a KeyError.
    z	a prompt i  za completion )prompt
completionz.jsonlw)suffixmode
z(openai tools fine_tunes.prepare_data -f Tz	y
y
y
y
yzutf-8)stdouttextinputstderrencodingshellN z(indices of the long examples has changed)r   printnamewritejsondumpsflush
subprocessrunPIPEr   r   )short_promptlong_promptshort_completionlong_completionunprepared_training_datatraining_dataprompt_completion_rowprepared_data_cmd_output r'   x/var/www/www-root/data/www/ovozai.pdev.uz/venv/lib/python3.10/site-packages/openai/tests/test_long_examples_validator.pytest_long_examples_validator   s4   	

r)   )r   N)r   r   tempfiler   pytestopenai.datalib.numpy_helperr   r   openai.datalib.pandas_helperr   r   markskipifr)   r'   r'   r'   r(   <module>   s    