allow .whl files for spacy nlp pipeline contributions

This commit is contained in:
Patrick Jentsch 2024-02-14 14:24:21 +01:00
parent 05ab204e5a
commit cf8c164d60
2 changed files with 3 additions and 3 deletions

View File

@ -16,8 +16,8 @@ class CreateSpaCyNLPPipelineModelForm(ContributionBaseForm):
) )
def validate_spacy_model_file(self, field): def validate_spacy_model_file(self, field):
if not field.data.filename.lower().endswith('.tar.gz'): if not field.data.filename.lower().endswith(('.tar.gz', ('.whl'))):
raise ValidationError('.tar.gz files only!') raise ValidationError('.tar.gz or .whl files only!')
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
if 'prefix' not in kwargs: if 'prefix' not in kwargs:

View File

@ -42,7 +42,7 @@
{{ form.hidden_tag() }} {{ form.hidden_tag() }}
<div class="row"> <div class="row">
<div class="col s12 l5"> <div class="col s12 l5">
{{ wtf.render_field(form.spacy_model_file, accept='.tar.gz', placeholder='Choose a .tar.gz file') }} {{ wtf.render_field(form.spacy_model_file, accept='.tar.gz,.whl', placeholder='Choose a .tar.gz or .whl file') }}
</div> </div>
<div class="col s12 l7"> <div class="col s12 l7">
{{ wtf.render_field(form.title, material_icon='title') }} {{ wtf.render_field(form.title, material_icon='title') }}