This repository has been archived by the owner on Nov 22, 2022. It is now read-only.
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Replace unit-tests on LMModel and FLLanguageModelingTask by LiteLMMod…
…el and FLLiteLMTask (#1296) Summary: Pull Request resolved: #1296 Previously, fl_language_model_test.py covers unit-tests on LMModel by comparing LMTask and FLLMTask. We will ship LiteLMModel instead of LMModel, hence the unit-tests should be using LiteLMModel, LiteLMTask, and FLLiteLMTask instead. One caveat: we need to import LiteLMTask from fb/tasks.py, however, the import section in fb/tasks.py misses tons of dependencies (I guess no unit-tests was ran on tasks.py?), hence I have to replicate LiteLMTask. Reviewed By: kmalik22 Differential Revision: D20651264 fbshipit-source-id: d4843efdea6e4dbc5ffd97963a275ac87337580b
- Loading branch information