Skip to content

Commit

Permalink
#252 Fixed language_container_factory
Browse files Browse the repository at this point in the history
  • Loading branch information
ahsimb committed Sep 24, 2024
1 parent 9a87a1a commit 09726f4
Showing 1 changed file with 1 addition and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,6 @@


CONTAINER_NAME = "exasol_transformers_extension_container"
LANGUAGE_ALIAS = "PYTHON3_TE"


def add_pytorch_to_requirements(container_builder: LanguageContainerBuilder) -> None:
Expand All @@ -21,7 +20,7 @@ def add_pytorch_to_requirements(container_builder: LanguageContainerBuilder) ->

@contextmanager
def language_container_factory():
with LanguageContainerBuilder(CONTAINER_NAME, LANGUAGE_ALIAS) as container_builder:
with LanguageContainerBuilder(CONTAINER_NAME) as container_builder:
add_pytorch_to_requirements(container_builder)
project_directory = find_path_backwards("pyproject.toml", __file__).parent
container_builder.prepare_flavor(project_directory, requirement_filter=exclude_cuda)
Expand Down

0 comments on commit 09726f4

Please sign in to comment.