Skip to content

brahmap.utilities.modify_numpy_context

Bases: object

Source code in brahmap/utilities/tools.py 
43
44
45
46
47
48
49
50
51
52
class modify_numpy_context(object):
    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