Merge pull request #7 from explosion/rename-update

update names and update version

.github/workflows/tests.yml

@@ -44,6 +44,14 @@ jobs:
           python -m prodigy db-in textcat-binary tests/datasets/textcat-binary.jsonl
           python -m prodigy db-in textcat-multi tests/datasets/textcat-multi.jsonl
       
+      - name: Confirm recipe names
+        run: | 
+          python -m prodigy hf.ner.correct --help
+          python -m prodigy hf.textcat.correct --help
+          python -m prodigy hf.train.ner --help
+          python -m prodigy hf.train.textcat --help
+          python -m prodigy hf.upload --help
+      
       - name: Run ruff
         if: always()
         shell: bash

prodigy_hf/ner.py

@@ -276,7 +276,7 @@ def get_hf_config_labels(hf_mod: TokenClassificationPipeline):
 
 
 @recipe(
-    "hf.correct.ner",
+    "hf.ner.correct",
     # fmt: off
     dataset=Arg(help="Dataset to write annotations into"),
     model=Arg(help="Path to transformer model. Can also point to model on hub."),

prodigy_hf/textcat.py

@@ -207,7 +207,7 @@ def add_model_predictions(stream, hf_pipeline, model_labels):
 
 
 @recipe(
-    "hf.correct.textcat",
+    "hf.textcat.correct",
     # fmt: off
     dataset=Arg(help="Dataset to write annotations into"),
     model=Arg(help="Path to transformer model. Can also point to model on hub."),

setup.cfg

@@ -19,8 +19,8 @@ prodigy_recipes =
     hf.upload = prodigy_hf:hf_upload
     hf.train.ner = prodigy_hf:hf_train_ner
     hf.train.textcat = prodigy_hf:hf_train_textcat
-    hf.correct.ner = prodigy_hf:hf_ner_correct
-    hf.correct.textcat = prodigy_hf:hf_textcat_correct
+    hf.ner.correct = prodigy_hf:hf_ner_correct
+    hf.textcat.correct = prodigy_hf:hf_textcat_correct
 
 [bdist_wheel]
 universal = true