trustformers-debug 0.1.1

Advanced debugging tools for TrustformeRS models
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

//! # AggregationFunction - Trait Implementations
//!
//! This module contains trait implementations for `AggregationFunction`.
//!
//! ## Implemented Traits
//!
//! - `Hash`
//! - `Eq`
//! - `PartialEq`
//!
//! 🤖 Generated with [SplitRS](https://github.com/cool-japan/splitrs)

use super::types::*;
use std::mem;

impl std::hash::Hash for AggregationFunction {
    fn hash<H: std::hash::Hasher>(&self, state: &mut H) {
        mem::discriminant(self).hash(state);
        match self {
            AggregationFunction::Percentile(p) => {
                p.to_bits().hash(state);
            },
            AggregationFunction::Custom(s) => s.hash(state),
            _ => {},
        }
    }
}

impl Eq for AggregationFunction {}

impl PartialEq for AggregationFunction {
    fn eq(&self, other: &Self) -> bool {
        match (self, other) {
            (AggregationFunction::Percentile(a), AggregationFunction::Percentile(b)) => {
                a.to_bits() == b.to_bits()
            },
            (AggregationFunction::Custom(a), AggregationFunction::Custom(b)) => a == b,
            _ => mem::discriminant(self) == mem::discriminant(other),
        }
    }
}