refactor so tests/ exist under each sub-package with unit tests

Signed-off-by: Terry Kong <[email protected]>
NVIDIA · Jan 21, 2025 · 36bd74e · 36bd74e
1 parent 01f3478
commit 36bd74e
Show file tree

Hide file tree

Showing 11 changed files with 4 additions and 2 deletions.
diff --git a/nemo_aligner/data/nlp/tests/__init__.py b/nemo_aligner/data/nlp/tests/__init__.py
diff --git a/nemo_aligner/data/nlp/cai_utils_test.py → ..._aligner/data/nlp/tests/cai_utils_test.py b/nemo_aligner/data/nlp/cai_utils_test.py → ..._aligner/data/nlp/tests/cai_utils_test.py
diff --git a/nemo_aligner/data/nlp/datasets_test.py → nemo_aligner/data/nlp/tests/datasets_test.py b/nemo_aligner/data/nlp/datasets_test.py → nemo_aligner/data/nlp/tests/datasets_test.py
diff --git a/nemo_aligner/experimental/README.md b/nemo_aligner/experimental/README.md
@@ -15,13 +15,15 @@ NeMo-Aligner/
 │   ├── algorithms/
 │   ├── data/
 │   │   ├── datasets.py
-│   │   └── datasets_test.py
+│   │   └── tests/
+│   │       └── datasets_test.py
 │   └── experimental/               <----- experimental sub-package
 │       ├── <proj-name>/
 │           ├── dataset.py          <----- experimental dataset
 │           ├── new_algo.py         <----- experimental algo
 │           ├── model.py            <----- experimental model
-│           └── model_test.py       <----- experimental model test
+│           └── tests/
+│               └── model_test.py   <----- experimental model test
 └── tests/
     └── functional/
         └── dpo.sh

diff --git a/nemo_aligner/utils/tests/__init__.py b/nemo_aligner/utils/tests/__init__.py
diff --git a/nemo_aligner/utils/distributed_test.py → nemo_aligner/utils/tests/distributed_test.py b/nemo_aligner/utils/distributed_test.py → nemo_aligner/utils/tests/distributed_test.py
diff --git a/nemo_aligner/utils/ppo_utils_test.py → nemo_aligner/utils/tests/ppo_utils_test.py b/nemo_aligner/utils/ppo_utils_test.py → nemo_aligner/utils/tests/ppo_utils_test.py
diff --git a/...igner/utils/text_generation_utils_test.py → ...utils/tests/text_generation_utils_test.py b/...igner/utils/text_generation_utils_test.py → ...utils/tests/text_generation_utils_test.py
diff --git a/nemo_aligner/utils/trainer_utils_test.py → ...aligner/utils/tests/trainer_utils_test.py b/nemo_aligner/utils/trainer_utils_test.py → ...aligner/utils/tests/trainer_utils_test.py
diff --git a/nemo_aligner/utils/trt_llm_test.py → nemo_aligner/utils/tests/trt_llm_test.py b/nemo_aligner/utils/trt_llm_test.py → nemo_aligner/utils/tests/trt_llm_test.py
diff --git a/nemo_aligner/utils/utils_test.py → nemo_aligner/utils/tests/utils_test.py b/nemo_aligner/utils/utils_test.py → nemo_aligner/utils/tests/utils_test.py