350f9e107f
Since #4335 was merged, I've noticed that the definition of ServerRunner in the tests is the same as in the test for OpenAI API. I have moved the class to the test utilities to avoid code duplication. (Although it only has been repeated twice so far, I will add another similar test suite in #4200 which would duplicate the code a third time) Also, I have moved the test utilities file (test_utils.py) to under the test directory (tests/utils.py), since none of its code is actually used in the main package. Note that I have added __init__.py to each test subpackage and updated the ray.init() call in the test utilities file in order to relative import tests/utils.py. |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| allclose_default.py | ||
| conftest.py | ||
| test_activation.py | ||
| test_attention.py | ||
| test_cache.py | ||
| test_layernorm.py | ||
| test_moe.py | ||
| test_pos_encoding.py | ||
| test_prefix_prefill.py | ||
| test_rand.py | ||
| test_sampler.py | ||