using prodigy.serve for textcat.batch-train

,
7

hello Ines

Any news on this error ?

I use this function:

import logging
from pathlib import Path

import prodigy
from prodigy.recipes.ner import batch_train

@prodigy.recipe('sentinel.ner.batch.train',
                dataset=prodigy.recipe_args['dataset'],
                spacy_model=prodigy.recipe_args['spacy_model'],
                label=prodigy.recipe_args['label']
                )
def batch_train_custom(dataset, spacy_model, label):
    print("Train {} to {}".format(dataset, label))
    output_path = Path('models/'+dataset)

    return batch_train(dataset=dataset,
                       input_model=spacy_model,
                       output_model=output_path,
                       label=label,
                       factor=1,
                       dropout=0.2,
                       n_iter=10,
                       batch_size=-1,
                       beam_width=16,
                       eval_id=None,
                       eval_split=0.2,
                       unsegmented=False,
                       no_missing=False,
                       silent=False)

It works and the model is created. But I have this error:

Model: /tmp/models/temporary_model
Training data: /tmp/models/temporary_model/training.jsonl
Evaluation data: /tmp/models/temporary_model/evaluation.jsonl
Traceback (most recent call last):
  File "/Users/iero/miniconda/lib/python3.6/runpy.py", line 193, in _run_module_as_main
    "__main__", mod_spec)
  File "/Users/iero/miniconda/lib/python3.6/runpy.py", line 85, in _run_code
    exec(code, run_globals)
  File "/Users/iero/code/sentinel/sorcerer.py", line 300, in <module>
    prodigy.serve('sentinel.ner.batch.train', training_name, nlp_models[option.lang], training_labels)
  File "/Users/iero/miniconda/lib/python3.6/site-packages/prodigy/__init__.py", line 23, in serve
    controller.config.update(config)
AttributeError: 'dict' object has no attribute 'config'

Any idea ?