summaryrefslogtreecommitdiff
path: root/utils/tuning/libtuning/average.py
blob: e28770d7b865129ddd64a13779cf5090ac952b42 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
# SPDX-License-Identifier: GPL-2.0-or-later
#
# Copyright (C) 2022, Paul Elder <paul.elder@ideasonboard.com>
#
# average.py - Wrapper for numpy averaging functions to enable duck-typing

import numpy as np


# @brief Wrapper for np averaging functions so that they can be duck-typed
class Average(object):
    def __init__(self):
        pass

    def average(self, np_array):
        raise NotImplementedError


class Mean(Average):
    def average(self, np_array):
        return np.mean(np_array)