-
Notifications
You must be signed in to change notification settings - Fork 2
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Added task_type parameter for correct model loading #245
Added task_type parameter for correct model loading #245
Conversation
exasol_transformers_extension/utils/current_model_specification.py
Outdated
Show resolved
Hide resolved
exasol_transformers_extension/utils/current_model_specification.py
Outdated
Show resolved
Hide resolved
tests/fixtures/model_fixture.py
Outdated
|
||
@pytest.fixture(scope="session") | ||
def prepare_base_model_for_local_bucketfs(tmpdir_factory) -> PurePosixPath: | ||
def prepare_question_answering_model_for_local_bucketfs(tmpdir_factory) -> PurePosixPath: | ||
model_specification = model_params.base_model_specs |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I think, you should select one model per task for testing, because the base_model "bert_uncased" can't be used for all tasks. I think, it can be used for mask filling, sequence classification, zero shot classification and maybe text generation. I think, it can't used for translation, question answering.
tests/integration_tests/with_db/udfs/test_sequence_classification_text_pair_script.py
Outdated
Show resolved
Hide resolved
tests/integration_tests/with_db/udfs/test_token_classification_script.py
Outdated
Show resolved
Hide resolved
tests/integration_tests/with_db/udfs/test_zero_shot_text_classification_script.py
Outdated
Show resolved
Hide resolved
tests/integration_tests/without_db/udfs/test_sequence_classification_single_text_udf.py
Outdated
Show resolved
Hide resolved
bucketfs_connections = [Connection(address=f"file:///test{i}") for i in range(count)] | ||
bfs_conn_name = [f"bfs_conn_name_{i}" for i in bucketfs_connections] | ||
|
||
mock_cmss = [create_autospec(CurrentModelSpecification, | ||
mock_cmss = [create_autospec(BucketFSModelSpecification, | ||
model_name=base_model_names[i], |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
wow, cool, that I didn't knew
…aiting for the SLC build while the SaaS DB is already running [CodeBuild]
…e without need and make db_conn a session fixture [CodeBuild]
…ach test. It seems to be a bug that a session scope fixture is called for every test. This might happen because backend is parameterized. [CodeBuild]
…CodeBuild [CodeBuild]
… them [CodeBuild]
|
||
_BACKEND_OPTION = '--backend' | ||
|
||
def pytest_addoption(parser): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
we should rename this on to pytest_add_option at some point, adoption is something else.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This function name is defined by pytest and can't be changed
All Submissions:
[CodeBuild]
to the commit messagethis pr is a mismash of tickets to get to a working release as fast as possible. it aims to fix the main problems presented in #213. in order to archive this it partly does tasks in these tickets:
#221
#220
#219
and fixes #218