Include stack trace when sampling errors are surfaced

[amontanez24]

Problem Description

As a developer, I'd like to be able to see the stack trace of an error so I can figure out what went wrong and where to look.

We have a function that catches and surfaces errors with a better message during sampling.

SDV/sdv/single_table/utils.py

Lines 81 to 108 in 0d203be

	def handle_sampling_error(output_file_path, sampling_error):
	"""Handle sampling errors by printing a user-legible error and then raising.
	Args:
	output_file_path (str):
	The output file path.
	sampling_error:
	The error to raise.
	Side Effects:
	The error will be raised.
	"""
	if 'Unable to sample any rows for the given conditions' in str(sampling_error):
	raise sampling_error
	error_msg = None
	if output_file_path is not None:
	error_msg = f'Error: Sampling terminated. Partial results are stored in {output_file_path}.'
	else:
	error_msg = (
	'Error: Sampling terminated. No results were saved due to unspecified '
	'"output_file_path".'
	)
	if error_msg:
	raise type(sampling_error)(error_msg + '\n' + str(sampling_error))
	raise sampling_error

The problem is that when we raise the new error, we're only including the original error message but not the stack trace. This makes it almost impossible to debug. Below is an example output

Expected behavior

• When an error raises from the handle_sampling_error function, it should show the stack trace.

Additional context

• To do this, we can change this line to use exception chaining

SDV/sdv/single_table/utils.py

Lines 105 to 106 in 0d203be

	if error_msg:
	raise type(sampling_error)(error_msg + '\n' + str(sampling_error))

The line above should be something like

raise type(sampling_error)(error_msg) from sampling_error