abd-clam 0.11.0

Clustered Learning of Approximate Manifolds
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

import functools
import time
import typing


class TimeIt:
    """A class to provide a decorator for timing the execution of a function."""

    def __init__(
        self,
        logger,
        template: str = "completed {:s} in {:.3f} seconds",
    ) -> None:
        self.template: str = template
        self.logger = logger

    def __call__(self, function: typing.Callable):
        @functools.wraps(function)
        def wrapper(*args, **kwargs):
            start = time.perf_counter()
            result = function(*args, **kwargs)
            end = time.perf_counter()

            self.logger.info(self.template.format(function.__name__, end - start))
            return result

        return wrapper