alphagov / govuk-content-metadata

GovNER: an encoder-based language model (RoBERTa) fine-tuned to perform Named Entity Recognition (NER) on GOV.UK content
MIT License
4 stars 1 forks source link

Refactor tests for bulk inference pipeline (following recent changes) #65

Closed exfalsoquodlibet closed 1 year ago

exfalsoquodlibet commented 1 year ago

Following the moving of the bulk inference pipeline to GCP and a VM with GPU (see PR https://github.com/alphagov/govuk-content-metadata/pull/56), tests for the inference functions had to be rewritten. However, there are some issues when these are run locally since the M1 machine does not have CUDA gpu available.

At the moment, we have deleted the tests (tests/test_bulk_inference_pipeline/test_bulk_src.py) but this is not a solution!

rory-hurley-gds commented 1 year ago

Done in pr #85