rishil2005/github-issue-triage-env
0
1+
2 c�i� � �b � ^ RI Ht ^ RIt^ RIHt ^ RIHt ^ RIHtH t R R lt3 ! R R 4 t R# )4� )� annotationsN)�Path)�Any)� AgentAction�IssueObservationc �( � V ^8� d QhRRRRRRRR/# ) � �actionr �expected_action�str�expected_labelz5str | None�return�int� )�formats "�1C:\Users\rishi\OneDrive\Desktop\RL\environment.py� __annotate__r 6 s) � � � �K� �#� �z� �^a� � c � � VR 8X d V P R 8X d ^# R# V P R8X d V P V8X d ^# R# )�RequestMoreInfo�AddLabel�����)� action_type�label�r7 r r
s &&&r �_compute_rewardr 8 sE � ��+�+��&�&�*;�;�q�C��C�
���Z�'�F�L�L�N�,J��
�Ir c �V � ] tR t^tR
R R lltR R ltR R ltR R ltR R9 ltR t R # )�IssueEnvironmentc � � V ^8� d QhRRRR/# )r � dataset_pathz10str | Pathr �Noner )r s "r r � IssueEnvironment.__annotate__ s � � � �Z� �T� r c �B � \ V4 V n . V n ^ V n R# )r N)r r �_issues�_index)�selfr s &&r �__init__�IssueEnvironment.__init__ s � � � � .���-/�� ��� r c � � V ^8� d QhRR/# �r r r r )r s "r r r" s � � +� +�'� +r c � � \ V P R R7 ;_uu_ 4 p\ P ! V4 V n RRR4 V P ' g \ R4 h^ V n V P 4 # + ' g i LD; i)zutf-8)�encodingNzdataset.json is empty.)�openr �json�loadr$ �11ValueErrorr% �_current_observation)r& �fhs & r �reset�IssueEnvironment.reset s_ � �
�$�#�#�g�
6�
6�"� �9�9�R�=�D�L� 7��|�|�|� � 5�6� 6��� ��(�(�*�*� 7�
6�s � A8�8 B c � � V ^8� d QhRRRR/# )r r12 r r z9tuple[IssueObservation | None, int, bool, dict[str, Any]]r )r s "r r r" s � � ,� ,�;� ,�+d� ,r c � � V P V P , p\ VVR , VP R4 R7 pRVR, R VR , RVP R4 RVP RVP13 RV/pV ;P ^,
un V P \
V P 4 8� pV' d RMV P 4 pWcWT3# )r r
r �issue_id� agent_action� agent_label�rewardN)r$ r% r �getr r �lenr1 )r& r14 �currentr: �info�done�next_obss && r �step�IssueEnvironment.step s� � ��,�,�t�{�{�+�� �� #�$5� 6�"�;�;�'7�8�15��
��16�+� �w�'8� 9� �g�k�k�*:� ;� �F� .� .� �6�<�<� �f�
17��
� � �q�� ��{�{�c�$�,�,� /�/�� �4�T�%>�%>�%@����+�+r c � � V ^8� d QhRR/# )r r r! r )r s "r r r" 6 s � � � �t� r c � � R # )Nr )r& s &r �close�IssueEnvironment.close6 s � �r c � � V ^8� d QhRR/# r* r )r s "r r r" 9 s � � 18� 19�&6� 20r c � � V P V P , p\ VR , VR, VR, R7 # )r7 �title�body)r7 rI rJ )r$ r% r )r& �issues & r r1 �%IssueEnvironment._current_observation9 s7 � �� � �T�[�[�)�� ��:�&���.��v� �21� 22r )r% r$ r N)z dataset.json)23�__name__�24__module__� __qualname__�__firstlineno__r' r3 rA rE r1 �__static_attributes__r r r r r s � ��25+�,�,�26� 27r r ) �28__future__r r. �pathlibr �typingr �modelsr r r r r r r �<module>rV s# �� "� � � � 0��-29� -30r