PR feedback

Author: Val Brodsky

libs/labelbox/src/labelbox/exceptions.py

@@ -152,7 +152,7 @@ class ProcessingWaitTimeout(Exception):
     """Raised when waiting for the data rows to be processed takes longer than allowed"""
 
 
-def _error_message_for_unparsed_graphql_error(error_string: str) -> str:
+def error_message_for_unparsed_graphql_error(error_string: str) -> str:
     """
     Since our client only parses certain graphql errors, this function is used to
     extract the error message from the error string when the error is not

libs/labelbox/src/labelbox/schema/project.py

@@ -13,7 +13,7 @@
 
 from labelbox import parser
 from labelbox import utils
-from labelbox.exceptions import _error_message_for_unparsed_graphql_error
+from labelbox.exceptions import error_message_for_unparsed_graphql_error
 from labelbox.exceptions import (InvalidQueryError, LabelboxError,
                                  ProcessingWaitTimeout, ResourceConflict,
                                  ResourceNotFoundError)
@@ -1275,7 +1275,7 @@ def add_model_config(self, model_config_id: str) -> str:
             if e.message.startswith(
                     "Unknown error: "
             ):  # unfortunate hack to handle unparsed graphql errors
-                error_content = _error_message_for_unparsed_graphql_error(
+                error_content = error_message_for_unparsed_graphql_error(
                     e.message)
             else:
                 error_content = e.message

libs/labelbox/src/labelbox/schema/project_model_config.py

@@ -1,6 +1,6 @@
 from labelbox.orm.db_object import DbObject
 from labelbox.orm.model import Field, Relationship
-from labelbox.exceptions import LabelboxError, _error_message_for_unparsed_graphql_error
+from labelbox.exceptions import LabelboxError, error_message_for_unparsed_graphql_error
 
 
 class ProjectModelConfig(DbObject):
@@ -38,7 +38,7 @@ def delete(self) -> bool:
             if e.message.startswith(
                     "Unknown error: "
             ):  # unfortunate hack to handle unparsed graphql errors
-                error_content = _error_message_for_unparsed_graphql_error(
+                error_content = error_message_for_unparsed_graphql_error(
                     e.message)
             else:
                 error_content = e.message

libs/labelbox/tests/unit/test_exceptions.py

@@ -1,13 +1,13 @@
 import pytest
 
-from labelbox.exceptions import _error_message_for_unparsed_graphql_error
+from labelbox.exceptions import error_message_for_unparsed_graphql_error
 
 
 @pytest.mark.parametrize('exception_message, expected_result', [
-    ("Unparsed errors on query execution: [{'message': 'Cannot create model config for project because model setup is complete'",
+    ("Unparsed errors on query execution: [{'message': 'Cannot create model config for project because model setup is complete'}]",
      "Cannot create model config for project because model setup is complete"),
     ("blah blah blah", "Unknown error"),
 ])
 def test_client_unparsed_exception_messages(exception_message, expected_result):
-    assert _error_message_for_unparsed_graphql_error(
+    assert error_message_for_unparsed_graphql_error(
         exception_message) == expected_result