Skip to content

#316 improve task type#369

Merged
MarleneKress79789 merged 25 commits into
mainfrom
refactoring/#316-improve-task-type
Jun 23, 2026
Merged

#316 improve task type#369
MarleneKress79789 merged 25 commits into
mainfrom
refactoring/#316-improve-task-type

Conversation

@MarleneKress79789

@MarleneKress79789 MarleneKress79789 commented Feb 17, 2026

Copy link
Copy Markdown
Collaborator

All Submissions:

  • Is the title of the Pull Request correct?
  • Is the title of the corresponding issue correct?
  • Have you updated the changelog?
  • Have you updated the User/Developer Guide?
  • Have you checked to ensure there aren't other open Pull Requests for the same update/change?
  • Are you mentioning the issue which this PullRequest fixes ("Fixes...")

fixes #316

Comment thread doc/user_guide/manage_models.md Outdated
Comment on lines +85 to +87
* `fill-mask`
* `question-answering`
* `text-classification`

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

So sequence classification and filling mask change. That is actually less bad than I assumed.
Filling mask is likely not used to often. Sequence classification is a bit annoying, because it many valid applications.

@@ -82,13 +82,15 @@ Specific parameters
Some models can be used for multiple types of tasks, but Hugging Face Transformers stores different metadata depending on the task of the model, which affects how the model is loaded later. Setting an incorrect task type, or leaving the task type empty may affect the models performance severely.

Available task types are:

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

We actually need feature-extraction as an allowed task type, because this is used in text-ai and our GPU example in ai-lab. Later, we want to have it also for a ComputeEmbedding UDF in this extension.

Comment thread doc/changes/unreleased.md Outdated
Comment thread doc/changes/unreleased.md Outdated
Comment thread doc/user_guide/manage_models.md Outdated
Comment thread doc/user_guide/manage_models.md Outdated
Comment thread doc/user_guide/manage_models.md Outdated
Comment thread doc/user_guide/manage_models.md Outdated
Comment thread exasol_transformers_extension/utils/model_specification.py Outdated
Comment thread exasol_transformers_extension/utils/model_specification.py Outdated
Comment thread exasol_transformers_extension/utils/model_specification.py Outdated
Comment thread test/fixtures/model_fixture.py Outdated
# Conflicts:
#	doc/changes/unreleased.md
#	doc/user_guide/manage_models.md
#	exasol_transformers_extension/deployment/default_udf_parameters.py
#	exasol_transformers_extension/udfs/callers/ai_classify_extended_udf.py
#	exasol_transformers_extension/udfs/callers/with_spans/span_ai_classify_extended_udf_call.py
#	exasol_transformers_extension/udfs/models/ai_classify_extended_udf.py
#	exasol_transformers_extension/udfs/models/prediction_tasks/zero_shot.py
#	exasol_transformers_extension/udfs/models/zero_ai_classify_extended_udf.py
#	exasol_transformers_extension/utils/model_specification.py
#	poetry.lock
#	test/fixtures/model_fixture.py
#	test/integration_tests/with_db/test_upload_model.py
#	test/integration_tests/without_db/udfs/test_ai_classify_extended_udf.py
#	test/unit/udfs/base_model_dummy_implementation.py
#	test/unit/udfs/test_ai_classify_extended.py
#	test/unit/udfs/test_delete_model_udf.py
#	test/unit/utils/test_load_local_model.py
Co-authored-by: Ariel Schulz <43442541+ArBridgeman@users.noreply.github.com>
@MarleneKress79789 MarleneKress79789 deployed to manual-approval June 22, 2026 12:33 — with GitHub Actions Active
@ArBridgeman ArBridgeman self-requested a review June 23, 2026 06:54
@MarleneKress79789 MarleneKress79789 enabled auto-merge (squash) June 23, 2026 08:22
@MarleneKress79789 MarleneKress79789 merged commit 51c0337 into main Jun 23, 2026
34 checks passed
@MarleneKress79789 MarleneKress79789 deleted the refactoring/#316-improve-task-type branch June 23, 2026 09:27
@sonarqubecloud

Copy link
Copy Markdown

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Improve task_type handling

3 participants