diff --git a/fslite/fs/methods.py b/fslite/fs/methods.py index adf67fe..133b528 100644 --- a/fslite/fs/methods.py +++ b/fslite/fs/methods.py @@ -184,10 +184,10 @@ def set_params(self, **kwargs): # return self.__str__() -class InvalidMethodError(Exception): +class InvalidMethodError(ValueError): """ Error raised when an invalid feature selection method is used. """ def __init__(self, message): - super().__init__(message) + super().__init__(f"Invalid feature selection method: {message}")