grammarly / gector

Official implementation of the papers "GECToR – Grammatical Error Correction: Tag, Not Rewrite" (BEA-20) and "Text Simplification by Tagging" (BEA-21)
Apache License 2.0
891 stars 216 forks source link

Upstream/2 write a predictor class to mimic the gec model #172

Open damien2012eng opened 1 year ago

damien2012eng commented 1 year ago

As discussed before, our team has been working on updating GECToR dependencies to the latest versions (Inference only). This is the second PR, one more PR is coming soon. Link to the successful run: here We also checked the impact for the training process. No breaks were identified.

First PR: Add unit tests and regression test. Additionally, bump Python version. Second PR: Write a predictor class to mimic the GEC_model. Third PR: Unpin all the dependency requirements

How to review:

Unit test: pytest -v tests Regression test: python regression_tests/test_gector_roberta.py python regression_tests/test_regression_data_predictor.py Our plan is to submit one more PR. Suggestions and comments are welcome!