rishil2005/github-issue-triage-env
0
1+
2 dH�ih3 � � � R t ^ RIHt ^ RIt^ RIHt ^ RIt^ RIt ^ RIH4t5 ^ RI H t ^ RI
Ht ! R R]P ] P ]3, 4 tR# ) z=6gym_env.py - Gymnasium-compatible wrapper for issue triage.7)� annotationsN)�Any)�spaces)�IssueEnvironment)�featurize_observationc � a � ] tR t^tRtRR./tRR V 3R llltRRRR/R V 3R8 llltR R ltR
R lt R R lt9Rt V ;t # )�IssueTriageGymEnvz5Gymnasium environment for stable-baselines3 training.� render_modes�humanc �$ � V ^8� d QhRRRRRR/# )� � dataset_path�str�shuffle_each_episode�bool�return�None� )�formats "�-C:\Users\rishi\OneDrive\Desktop\RL\gym_env.py� __annotate__� IssueTriageGymEnv.__annotate__ s# � � 4� 4�S� 4�QU� 4�ae� 4� c � <� \ SV ` 4 \ VR 7 V n W n / V n \ P ! ^4 V n \ P ! \ P ! . RO\ P R7 \ P ! . RO\ P R7 \ P R7 V n
RV n R# ))r
��dtype)�low�highr N)� r r r )� r r r )�super�__init__r �base_envr �10_last_infor �Discrete� action_space�Box�np�array�int32�observation_space� _current_obs)�selfr
r � __class__s &&&�r r! �IssueTriageGymEnv.__init__ s� �� �
����(�l�C��
�$8�!�*,��� #�O�O�A� .��� "(������ �R�X�X�6����,�b�h�h�7��(�(�"11�� � 04��r �seedN�optionsc �$ � V ^8� d QhRRRRRR/# )r r/ z12int | Noner0 zdict[str, Any] | Noner z!tuple[np.ndarray, dict[str, Any]]r )r s "r r r * s$ � � � �!��3H�� *�r c � <� \ SV ` VR 7 Ve6 \ P ! V4 \ P P V4 V P13 P 4 pV P ' dV \ P ! V P14 P 4 ^ V P15 n V P16 P 4 p\ P ! \ V4 \ P R7 pW@n / V n V/ 3# ))r/ r )r �reset�randomr/ r' r" r �shuffle�_issues�_index�_current_observationr( r r) r+ r# )r, r/ r0 �obs�stater- s &$$ �r r3 �IssueTriageGymEnv.reset* s� �� � ��
�4�
� � � � �K�K�� � �I�I�N�N�4� ��m�m�!�!�#�� � $� $� $� �N�N�4�=�=�0�0� 1�#$�D�M�M� ��-�-�4�4�6�C����.�s�3�2�8�8�D�� !�� ����b�y�r c � � V ^8� d QhRRRR/# )r �action�intr z4tuple[np.ndarray, float, bool, bool, dict[str, Any]]r )r s "r r r > s � � F� F�3� F�#W� Fr c � � V P P \ V4 , pV P P V4 w r4rVVf( \ P17 ! ^\ P R7 pM/\ P ! \ V4 \ P R7 pWpn W`n18 \ V4 pRp V\ V4 W�V3# )Nr F)
r" r% r> �stepr' �zerosr) r( r r+ r# r �float)19r, r= �
mapped_action�next_obs�reward�done�info�20next_state�21terminated� truncateds22 && r r@ �IssueTriageGymEnv.step> s� � � �
�
�2�2�3�v�;�?�
�'+�}�}�'9�'9�-�'H�$��$� � ����!�2�8�8�4�J����"7��"A����R�J� &�� ���$�Z�23�� ��5��=�*��E�Er c � � V ^8� d QhRR/# �r r r r )r s "r r r N s � � #� #�� #r c �X � V P ' d \ V P 4 R # R # �N)r# �print�r, s &r �render�IssueTriageGymEnv.renderN s � � �?�?�?� �$�/�/� "� r c � � V ^8� d QhRR/# rM r )r s "r r r R s � � � �t� r c � � R # rO r rQ s &r �close�IssueTriageGymEnv.closeR s � �r )r+ r# r% r" r* r )z dataset.jsonT)
�__name__�24__module__� __qualname__�__firstlineno__�__doc__�metadatar! r3 r@ rR rV �__static_attributes__�
__classcell__)r- s @r r r sK �� �?� �� �*�H�4� 4�&�$(��KO�� �(F� #�� r r )r\ �25__future__r r4 �typingr � gymnasium�gym�numpyr' r � environmentr �rl_agentr �Env�ndarrayr> r r r r �<module>ri sC ��� #�
� � � � � (� *�A�����26�27�C��0� Ar