truefoundry 0.5.0rc5__py3-none-any.whl → 0.5.0rc7__py3-none-any.whl

truefoundry/ml/autogen/models/exceptions.py

@@ -0,0 +1,30 @@
+INTERNAL_ERROR = 1
+INVALID_PARAMETER_VALUE = 1000
+
+
+class MlflowException(Exception):
+    """
+    Generic exception thrown to surface failure information about external-facing operations.
+    The error message associated with this exception may be exposed to clients in HTTP responses
+    for debugging purposes. If the error text is sensitive, raise a generic `Exception` object
+    instead.
+    """
+
+    def __init__(self, message, error_code=INTERNAL_ERROR, **kwargs):
+        """
+        :param message: The message describing the error that occured. This will be included in the
+                        exception's serialized JSON representation.
+        :param error_code: An appropriate error code for the error that occured; it will be included
+                           in the exception's serialized JSON representation. This should be one of
+                           the codes listed in the `mlflow.protos.databricks_pb2` proto.
+        :param kwargs: Additional key-value pairs to include in the serialized JSON representation
+                       of the MlflowException.
+        """
+        self.error_code = error_code
+        self.message = str(message)
+        self.json_kwargs = kwargs
+        super().__init__(message)
+
+    @classmethod
+    def invalid_parameter_value(cls, message, **kwargs):
+        return cls(message, error_code=INVALID_PARAMETER_VALUE, **kwargs)