custom intent split name (#238)

Author: Samoed

autointent/_dataset/_dataset.py

@@ -94,18 +94,19 @@ def from_json(cls, filepath: str | Path) -> "Dataset":
         return JsonReader().read(filepath)
 
     @classmethod
-    def from_hub(cls, repo_name: str) -> "Dataset":
+    def from_hub(cls, repo_name: str, intent_subset_name: str = Split.INTENTS) -> "Dataset":
         """Loads a dataset from the Hugging Face Hub.
 
         Args:
             repo_name: The name of the Hugging Face repository, like `DeepPavlov/clinc150`.
+            intent_subset_name: The name of the intent subset to load, defaults to `intents`.
         """
         from ._reader import DictReader
 
         splits = load_dataset(repo_name, "default")
         mapping = dict(**splits)
-        if Split.INTENTS in get_dataset_config_names(repo_name):
-            mapping["intents"] = load_dataset(repo_name, Split.INTENTS)[Split.INTENTS].to_list()
+        if intent_subset_name in get_dataset_config_names(repo_name):
+            mapping[Split.INTENTS] = load_dataset(repo_name, intent_subset_name, split=Split.INTENTS).to_list()
 
         return DictReader().read(mapping)