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 Summary: 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: 63aff6f1849cac30110f9dc7fe40a8f5df8ef2f0
- Loading branch information