Commit cb7fe593 authored by Joel Collins's avatar Joel Collins
Browse files

Added debug timer

parent aa9f6214
Loading
Loading
Loading
Loading
+12 −0
Original line number Diff line number Diff line
@@ -5,10 +5,22 @@ import base64
from uuid import UUID
import numpy as np
import logging
import time
from collections import abc
from functools import reduce
from contextlib import contextmanager

class Timer(object):
    def __init__(self, name):
        self.name = name
        self.start = None
        self.end = None
    def __enter__(self):
        self.start = time.time()
    def __exit__(self, type, value, traceback):
        self.end = time.time()
        logging.debug(f"{self.name} time: {self.end - self.start}")


def deserialise_array_b64(b64_string, dtype, shape):
    flat_arr = np.fromstring(base64.b64decode(b64_string), dtype)