improve error handling and fix gr.Warning typo

text_classification_ui_helpers.py

@@ -58,8 +58,8 @@ def check_dataset(dataset_id):
         configs = datasets.get_dataset_config_names(dataset_id, trust_remote_code=True)
         if len(configs) == 0:
             return (
-                gr.update(),
-                gr.update(),
+                gr.update(visible=False),
+                gr.update(visible=False),
                 ""
             )
         splits = datasets.get_dataset_split_names(
@@ -72,11 +72,13 @@ def check_dataset(dataset_id):
         )
     except Exception as e:
         logger.warn(f"Check your dataset {dataset_id}: {e}")
+        if "doesn't exist" in str(e):
+            gr.Warning(get_dataset_fetch_error_raw(e))
         if "forbidden" in str(e).lower(): # GSK-2770
-            gr.warning(get_dataset_fetch_error_raw(e))
+            gr.Warning(get_dataset_fetch_error_raw(e))
         return (
-            gr.update(),
-            gr.update(),
+            gr.update(visible=False),
+            gr.update(visible=False),
             ""
         )
 

wordings.py

@@ -67,7 +67,7 @@ HF_TOKEN_INVALID_STYLED= """
                 """
 
 def get_dataset_fetch_error_raw(error):
-    return f"""Sorry you cannot use this dataset because {error} Contact HF team to support this dataset."""
+    return f"""Sorry you cannot use this dataset because {error}. Contact HF team to support this dataset."""
 
 def get_styled_input(input):
     return f"""<h3 style="text-align: center;color: #4ca154; background-color: #e2fbe8; border-radius: 8px; padding: 10px; ">