RobustArchitectureSearch/process.py at master · shashankkotyan/RobustArchitectureSearch · GitHub

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
28
29
30
31
32
33
34
35
36
37
38
39
40
#!/usr/bin/env python

"""
Author: Shashank Kotyan
Email:  shashankkotyan@gmail.com
"""

import traceback, multiprocessing as mp


class Process(mp.Process):


    def __init__(self, *args, **kwargs):

        mp.Process.__init__(self, *args, **kwargs)

        self._pconn, self._cconn = mp.Pipe()
        self._exception = None


    def run(self):

        try:

            mp.Process.run(self)
            self._cconn.send(None)

        except Exception as e:

            tb = traceback.format_exc()
            self._cconn.send((e, tb))


    @property
    def exception(self):

        if self._pconn.poll(): self._exception = self._pconn.recv()

        return self._exception