Move rlhf tests out of modules (#1592)

pytorch · Sep 17, 2024 · 60a7e3d · 60a7e3d
1 parent df7cb46
commit 60a7e3d
Show file tree

Hide file tree

Showing 6 changed files with 0 additions and 0 deletions.
diff --git a/tests/torchtune/modules/rlhf/__init__.py → tests/torchtune/rlhf/__init__.py b/tests/torchtune/modules/rlhf/__init__.py → tests/torchtune/rlhf/__init__.py
diff --git a/...s/torchtune/modules/rlhf/loss/__init__.py → tests/torchtune/rlhf/loss/__init__.py b/...s/torchtune/modules/rlhf/loss/__init__.py → tests/torchtune/rlhf/loss/__init__.py
diff --git a/...chtune/modules/rlhf/loss/test_dpo_loss.py → tests/torchtune/rlhf/loss/test_dpo_loss.py b/...chtune/modules/rlhf/loss/test_dpo_loss.py → tests/torchtune/rlhf/loss/test_dpo_loss.py
diff --git a/...chtune/modules/rlhf/loss/test_ppo_loss.py → tests/torchtune/rlhf/loss/test_ppo_loss.py b/...chtune/modules/rlhf/loss/test_ppo_loss.py → tests/torchtune/rlhf/loss/test_ppo_loss.py
diff --git a/tests/torchtune/modules/rlhf/test_rewards.py → tests/torchtune/rlhf/test_rewards.py b/tests/torchtune/modules/rlhf/test_rewards.py → tests/torchtune/rlhf/test_rewards.py
diff --git a/.../modules/rlhf/test_sequence_processing.py → ...orchtune/rlhf/test_sequence_processing.py b/.../modules/rlhf/test_sequence_processing.py → ...orchtune/rlhf/test_sequence_processing.py