format, docu update, see if underscores work

Author: MarleneKress79789

doc/user_guide/manage_models.md

@@ -89,16 +89,24 @@ may affect the model's performance severely.
 
 Available task types are:
 
-| task_type                    | UDFs using this task_type    |
-|------------------------------|-------------------------------|
-| `fill-mask`                  | AI_FILL_MASK_EXTENDED         |
-| `question-answering`         | AI_ANSWER_EXTENDED            |
-| `text-classification`        | AI_CUSTOM_CLASSIFY_EXTENDED   |
-| `text-classification`        | AI_ENTAILMENT_EXTENDED        |
-| `text-generation`            | AI_COMPLETE_EXTENDED          |
-| `token-classification`       | AI_TRANSLATE_EXTENDED         |
-| `translation`                | AI_EXTRACT_EXTENDED           |
-| `zero-shot-classification`   | AI_CLASSIFY_EXTENDED          |
+| task_type                     | UDFs using this task_type   |
+|-------------------------------|-----------------------------|
+| `fill-mask`                   | AI_FILL_MASK_EXTENDED       |
+| `question-answering`          | AI_ANSWER_EXTENDED          |
+| `text-classification`         | AI_CUSTOM_CLASSIFY_EXTENDED |
+| `text-classification`         | AI_ENTAILMENT_EXTENDED      |
+| `text-classification`         | AI_SENTIMENT                |
+| `text-classification`         | AI_ENTAILMENT_EXTENDED      |
+| `text-generation`             | AI_COMPLETE_EXTENDED        |
+| `token-classification`        | AI_EXTRACT_EXTENDED         |
+| `token-classification`        | AI_EXTRACT_ENTITIES         |
+| `translation`                 | AI_TRANSLATE_EXTENDED       |
+| `zero-shot-classification`    | AI_CLASSIFY_EXTENDED        |
+| `zero-shot-classification`    | AI_CLASSIFY                 |
+| `document-question-answering` | NONE                        |
+| `mask-generation`             | NONE                        |
+| `table-question-answering`    | NONE                        |
+| `feature-extraction`          | NONE                        |
 
 
 Note that you may use underscores (`_`) instead of dashes (`-`).

exasol_transformers_extension/deployment/default_udf_parameters.py

@@ -39,5 +39,5 @@
         task_type="token-classification",
         bucketfs_conn_name=DEFAULT_BUCKETFS_CONN_NAME,
         sub_dir=Path(DEFAULT_SUBDIR),
-    )
+    ),
 }

test/fixtures/model_fixture.py

@@ -6,8 +6,8 @@
     PurePosixPath,
 )
 from test.fixtures.model_fixture_utils import (
-    prepare_default_model_for_local_bucketfs,
     download_model_to_standard_local_save_path,
+    prepare_default_model_for_local_bucketfs,
     prepare_model_for_local_bucketfs,
     upload_model,
     upload_model_to_bucketfs,
@@ -19,13 +19,12 @@
 import exasol.bucketfs as bfs
 import pytest
 
-from exasol_transformers_extension.utils.bucketfs_model_specification import (
-    BucketFSModelSpecification,
-)
-
 from exasol_transformers_extension.deployment.default_udf_parameters import (
     DEFAULT_MODEL_SPECS,
 )
+from exasol_transformers_extension.utils.bucketfs_model_specification import (
+    BucketFSModelSpecification,
+)
 
 
 @pytest.fixture(scope="session")

test/integration_tests/with_db/test_upload_model.py

@@ -57,7 +57,7 @@ def run_model_upload_test(
             get_cli_arg(StdParams.path_in_bucket, str(bucketfs_location)),
             get_cli_arg(MODEL_NAME_ARG, model_name),
             get_cli_arg(SUBDIR_ARG, sub_dir),
-            get_cli_arg(TASK_TYPE_ARG, "fill-mask"),#todo  "ai_fill_mask_extended"?
+            get_cli_arg(TASK_TYPE_ARG, "fill-mask"),
         ]
     )
 

test/utils/parameters.py

@@ -46,7 +46,7 @@ def create_illegal_tiny_model_specs():
 
 model_params = ModelParams(
     fill_model_specs=ModelSpecification(
-        "bert-base-uncased", "fill-mask"
+        "bert-base-uncased", "fill_mask"
     ),  # todo if this fixed the test, something goes wrong with the underscores
     seq2seq_model_specs=ModelSpecification("t5-small", "translation"),
     q_a_model_specs=ModelSpecification(