@@ -206,12 +206,6 @@ impl EnumBindGenerator {
206206 ) ;
207207 write_str ! ( self , " }" ) ;
208208 }
209-
210- fn generate_enum_hash_method ( & mut self ) {
211- write_str ! ( self , " pub fn __hash__(&self) -> u64 {" ) ;
212- write_str ! ( self , " crate::hash_u8(*self as u8)" ) ;
213- write_str ! ( self , " }" ) ;
214- }
215209}
216210
217211impl Generator for EnumBindGenerator {
@@ -247,8 +241,11 @@ impl Generator for EnumBindGenerator {
247241
248242 fn generate_definition ( & mut self ) {
249243 write_str ! ( self , "#[allow(non_camel_case_types)]" ) ;
250- write_str ! ( self , "#[pyclass(module = \" rlbot_flatbuffers\" , frozen)]" ) ;
251- write_str ! ( self , "#[derive(Debug, Default, Clone, Copy)]" ) ;
244+ write_str ! (
245+ self ,
246+ "#[pyclass(module = \" rlbot_flatbuffers\" , frozen, hash, eq, eq_int)]"
247+ ) ;
248+ write_str ! ( self , "#[derive(Debug, Default, Clone, Copy, PartialEq, Eq, Hash)]" ) ;
252249 write_fmt ! ( self , "pub enum {} {{" , self . struct_name) ;
253250 write_str ! ( self , " #[default]" ) ;
254251
@@ -329,7 +326,6 @@ impl Generator for EnumBindGenerator {
329326 self . generate_repr_method ( ) ;
330327 write_str ! ( self , "" ) ;
331328
332- self . generate_enum_hash_method ( ) ;
333329 write_str ! ( self , "}" ) ;
334330 write_str ! ( self , "" ) ;
335331 }
0 commit comments