[Andrew Polyakov] python-executor imports fix

Author: AndrewPolyakov1

utbot-python-executor/src/main/python/utbot_executor/utbot_executor/deep_serialization/json_converter.py

@@ -14,7 +14,11 @@
     MemoryDump, IteratorMemoryObject, NdarrayMemoryObject,
 )
 from utbot_executor.deep_serialization.utils import PythonId, TypeInfo
-import numpy as np
+try:
+    import numpy as np
+except ImportError:
+    import sys
+    print("numpy is not installed", file=sys.stderr)
 
 
 class MemoryObjectEncoder(json.JSONEncoder):

utbot-python-executor/src/main/python/utbot_executor/utbot_executor/deep_serialization/memory_objects.py

@@ -8,7 +8,11 @@
 import typing
 from itertools import zip_longest
 from typing import Any, Callable, Dict, List, Optional, Set, Type, Iterable
-import numpy as np
+try:
+    import numpy as np
+except ImportError:
+    import sys
+    print("numpy is not installed", file=sys.stderr)
 
 from utbot_executor.deep_serialization.config import PICKLE_PROTO
 from utbot_executor.deep_serialization.iterator_wrapper import IteratorWrapper

utbot-python-executor/src/main/python/utbot_executor/utbot_executor/executor.py

@@ -87,8 +87,9 @@ def add_imports(imports: Iterable[str]):
                         globals()[submodule_name] = importlib.import_module(
                             submodule_name
                         )
-                    except ModuleNotFoundError:
+                    except ModuleNotFoundError as e:
                         logging.warning("Import submodule %s failed", submodule_name)
+                        raise e
                 logging.debug("Submodule #%d: OK", i)
 
     def run_function(self, request: ExecutionRequest) -> ExecutionResponse:
@@ -115,6 +116,9 @@ def run_reduce_function(self, request: ExecutionRequest) -> ExecutionResponse:
             self.add_imports(request.imports)
             loader.add_syspaths(request.syspaths)
             loader.add_imports(request.imports)
+        except ModuleNotFoundError as _:
+            logging.debug("Error \n%s", traceback.format_exc())
+            return ExecutionFailResponse("fail", traceback.format_exc())
         except Exception as _:
             logging.debug("Error \n%s", traceback.format_exc())
             return ExecutionFailResponse("fail", traceback.format_exc())

utbot-python-executor/src/main/python/utbot_executor/utbot_executor/memory_compressor.py

@@ -3,8 +3,11 @@
 from utbot_executor.deep_serialization.memory_objects import MemoryDump
 from utbot_executor.deep_serialization.utils import PythonId
 
-import numpy as np
-
+try:
+    import numpy as np
+except ImportError:
+    import sys
+    print("numpy is not installed", file=sys.stderr)
 
 def compress_memory(
         ids: typing.List[PythonId],