Make exceptions less verbose by default

Author: ricardoV94

pytensor/compile/function/types.py

@@ -950,32 +950,27 @@ def __call__(self, *args, output_subset=None, **kwargs):
 
  except Exception as e:
  i = input_storage.index(arg_container)
- function_name = "pytensor function"
- argument_name = "argument"
- if self.name:
- function_name += ' with name "' + self.name + '"'
- if hasattr(arg, "name") and arg.name:
-  argument_name += ' with name "' + arg.name + '"'
- where = get_variable_trace_string(self.maker.inputs[i].variable)
- if len(e.args) == 1:
- e.args = (
-  "Bad input "
-  + argument_name
-  + " to "
-  + function_name
-  + f" at index {int(i)} (0-based). {where}"
- + e.args[0],
+ function_name = (
+  "pytensor function"
+  if self.name
+ else f"pytensor function '{self.name}'"
+ )
+ argument_name = (
+  f"argument {arg.name}"
+  if hasattr(arg, "name")
+ else "argument"
+ )
+ where = (
+ ""
+ if config.exception_verbosity == "low"
+ else get_variable_trace_string(
+ self.maker.inputs[i].variable
  )
- else:
- e.args = (
- "Bad input "
- + argument_name
- + " to "
- + function_name
- + f" at index {int(i)} (0-based). {where}"
- ) + e.args
- self._restore_defaults()
- raise
+ )
+ e.add_note(
+ f"\nInvalid {argument_name} to {function_name} at index {i}.{where}"
+ )
+ raise e
  arg_container.provided += 1
 
  # Set keyword arguments

pytensor/configdefaults.py

@@ -644,15 +644,8 @@ def add_error_and_warning_configvars():
  # on all important apply nodes.
  config.add(
  "exception_verbosity",
- "If 'low', the text of exceptions will generally refer "
- "to apply nodes with short names such as "
- "Elemwise{add_no_inplace}. If 'high', some exceptions "
- "will also refer to apply nodes with long descriptions "
- """ like:
- A. Elemwise{add_no_inplace}
- B. log_likelihood_v_given_h
- C. log_likelihood_h""",
- EnumStr("low", ["high"]),
+ "Verbosity of exceptions generated by PyTensor functions.",
+ EnumStr("low", ["medium", "high"]),
  in_c_key=False,
  )
 

pytensor/link/utils.py

@@ -313,6 +313,12 @@ def raise_with_op(
  # print a simple traceback from KeyboardInterrupt
  raise exc_value.with_traceback(exc_trace)
 
+ if verbosity == "low":
+ exc_value.add_note(
+ "\nHINT: Set PyTensor `config.exception_verbosity` to `medium` or `high` for more information about the source of the error."
+ )
+ raise exc_value.with_traceback(exc_trace)
+
  trace = getattr(node.outputs[0].tag, "trace", ())
 
  exc_value.__thunk_trace__ = trace