Hide traceback for epyccel and lambdify errors (pyccel#1869)

Add a `try/except` around internal Pyccel calls to catch all
`PyccelError`s. This allows the shortest possible traceback to be shown
to the user when the error is recognised, handled and reported by
Pyccel. Fixes pyccel#1868.

Author: EmilyBourne

CHANGELOG.md

@@ -22,6 +22,7 @@ All notable changes to this project will be documented in this file.
 -   #1844 : Add line numbers and code to errors from built-in function calls.
 -   #1867 : Add a `use_out` parameter to `pyccel.lambdify` to avoid unnecessary memory allocation.
 -   #1867 : Auto-generate a docstring for functions generated via calls to `pyccel.lambdify`.
+-   #1868 : Hide traceback for `epyccel` and `lambdify` errors.
 -   \[INTERNALS\] Added `container_rank` property to `ast.datatypes.PyccelType` objects.
 -   \[DEVELOPER\] Added an improved traceback to the developer-mode errors for errors in function calls.
 

pyccel/commands/epyccel.py

@@ -18,7 +18,7 @@
 
 from pyccel.utilities.strings  import random_string
 from pyccel.codegen.pipeline   import execute_pyccel
-from pyccel.errors.errors      import ErrorsMode
+from pyccel.errors.errors      import ErrorsMode, PyccelError
 
 __all__ = ['get_source_function', 'epyccel_seq', 'epyccel']
 
@@ -359,6 +359,8 @@ def epyccel( python_function_or_module, **kwargs ):
                 fun_name = python_function_or_module.__name__ if fun else None
                 success  = True
             # error handling carried out after broadcast to prevent deadlocks
+            except PyccelError as e:
+                raise type(e)(str(e)) from None
             except BaseException as e: # pylint: disable=broad-except
                 exc_info = e
                 success  = False
@@ -396,7 +398,10 @@ def epyccel( python_function_or_module, **kwargs ):
 
     # Serial version
     else:
-        mod, fun = epyccel_seq( python_function_or_module, **kwargs )
+        try:
+            mod, fun = epyccel_seq( python_function_or_module, **kwargs )
+        except PyccelError as e:
+            raise type(e)(str(e)) from None
 
     # Return Fortran function (if any), otherwise module
     return fun or mod

pyccel/commands/lambdify.py

@@ -5,8 +5,9 @@
 import sympy as sp
 from packaging import version
 
-from pyccel.commands.epyccel import epyccel
+from pyccel.commands.epyccel  import epyccel
 from pyccel.utilities.strings import random_string
+from pyccel.errors.errors     import PyccelError
 
 if version.parse(sp.__version__) >= version.parse('1.8'):
     from sympy.printing.numpy import NumPyPrinter
@@ -117,6 +118,10 @@ def lambdify(expr : sp.Expr, args : 'dict[sp.Symbol, str]', *, result_type : str
     docstring += '\n    """'
 
     func = '\n'.join((numpy_import, decorators, signature, docstring, code))
-    package = epyccel(func, **kwargs)
+    try:
+        package = epyccel(func, **kwargs)
+    except PyccelError as e:
+        raise type(e)(str(e)) from None
+
     return getattr(package, func_name)