Analyse trl (#1998)

## Changes whitelist trl ### Linked issues Progresses #1901 ### Functionality None ### Tests - [x] manually tested Co-authored-by: Eric Vergnaud <[email protected]>
databrickslabs · Jul 2, 2024 · 2f04dc5 · 2f04dc5
1 parent 2f3f8d0
commit 2f04dc5
Showing 1 changed file with 60 additions and 0 deletions.
diff --git a/src/databricks/labs/ucx/source_code/known.json b/src/databricks/labs/ucx/source_code/known.json
@@ -19538,6 +19538,66 @@
   "transformers": {
     "transformers": []
   },
+  "trl": {
+    "tests.slow": [],
+    "tests.slow.test_dpo_slow": [],
+    "tests.slow.test_sft_slow": [],
+    "tests.slow.testing_constants": [],
+    "trl": [],
+    "trl.commands": [],
+    "trl.commands.cli": [],
+    "trl.commands.cli_utils": [],
+    "trl.commands.scripts.bco": [],
+    "trl.commands.scripts.chat": [],
+    "trl.commands.scripts.cpo": [],
+    "trl.commands.scripts.ddpo": [],
+    "trl.commands.scripts.dpo": [],
+    "trl.commands.scripts.kto": [],
+    "trl.commands.scripts.orpo": [],
+    "trl.commands.scripts.ppo": [],
+    "trl.commands.scripts.ppo_multi_adapter": [],
+    "trl.commands.scripts.reward_modeling": [],
+    "trl.commands.scripts.sft": [],
+    "trl.commands.scripts.vsft_llava": [],
+    "trl.core": [],
+    "trl.environment": [],
+    "trl.environment.base_environment": [],
+    "trl.extras": [],
+    "trl.extras.best_of_n_sampler": [],
+    "trl.extras.dataset_formatting": [],
+    "trl.import_utils": [],
+    "trl.models": [],
+    "trl.models.modeling_base": [],
+    "trl.models.modeling_sd_base": [],
+    "trl.models.modeling_value_head": [],
+    "trl.models.sd_utils": [],
+    "trl.models.utils": [],
+    "trl.trainer": [],
+    "trl.trainer.base": [],
+    "trl.trainer.cpo_config": [],
+    "trl.trainer.cpo_trainer": [],
+    "trl.trainer.ddpo_config": [],
+    "trl.trainer.ddpo_trainer": [],
+    "trl.trainer.dpo_config": [],
+    "trl.trainer.dpo_trainer": [],
+    "trl.trainer.iterative_sft_trainer": [],
+    "trl.trainer.kto_config": [],
+    "trl.trainer.kto_trainer": [],
+    "trl.trainer.model_config": [],
+    "trl.trainer.orpo_config": [],
+    "trl.trainer.orpo_trainer": [],
+    "trl.trainer.ppo_config": [],
+    "trl.trainer.ppo_trainer": [],
+    "trl.trainer.ppov2_config": [],
+    "trl.trainer.ppov2_trainer": [],
+    "trl.trainer.reward_config": [],
+    "trl.trainer.reward_trainer": [],
+    "trl.trainer.rloo_config": [],
+    "trl.trainer.rloo_trainer": [],
+    "trl.trainer.sft_config": [],
+    "trl.trainer.sft_trainer": [],
+    "trl.trainer.utils": []
+  },
   "types-pyyaml": {},
   "types-requests": {},
   "typing_extensions": {