Skip to content

Commit

Permalink
exception added
Browse files Browse the repository at this point in the history
shaheennabi committed Dec 2, 2024
1 parent ca14869 commit 1fdbcc9
Showing 1 changed file with 31 additions and 0 deletions.
31 changes: 31 additions & 0 deletions src/finetuning/exception/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,31 @@
import sys


def error_message_detail(error, error_detail: sys):
_, _, exc_tb = error_detail.exc_info()

file_name = exc_tb.tb_frame.f_code.co_filename

error_message = "Error occurred python script name [{0}] line number [{1}] error message [{2}]".format(
file_name, exc_tb.tb_lineno, str(error)
)

return error_message


class CustomException(Exception):
def __init__(self, error_message, error_detail):
"""
:param error_message: error message in string format
"""
super().__init__(error_message)

self.error_message = error_message_detail(
error_message, error_detail=error_detail
)

def __str__(self):
return self.error_message



0 comments on commit 1fdbcc9

Please sign in to comment.