Skip to content

brahmap.utilities.modify_numpy_context

Bases: object

A context manager that replaces np.linalg.norm with parallel_norm

Source code in brahmap/utilities/tools.py 
42
43
44
45
46
47
48
49
50
51
52
53
class modify_numpy_context(object):
    """A context manager that replaces `np.linalg.norm` with `parallel_norm`"""

    def __init__(self):
        self.parallel_norm = parallel_norm
        self.original_norm = np.linalg.norm

    def __enter__(self):
        np.linalg.norm = self.parallel_norm

    def __exit__(self, exc_type, exc_val, exc_tb):
        np.linalg.norm = self.original_norm