put 'how to disable-asserts' tip in doc instead of tests docstring

Signed-off-by: Sophie du Couédic <[email protected]>

Author: sducouedic

docs/contributing/README.md

@@ -55,6 +55,9 @@ For additional features and advanced configurations, refer to the official [MkDo
 
 ## Testing
 
+!!! tip
+    When running tests, if errors occur, these can be analyzed/debugged by setting `DISABLE_ASSERTS = True` in spyre_util.py and by rerunning the test using `pytest --capture=no tests/spyre/test_spyre_basic.py`. After debugging, `DISABLE_ASSERTS` should be reset to `False`.
+
 ### Testing Locally on CPU (No Spyre card)
 
 Optionally, download the `ibm-ai-platform/micro-g3.3-8b-instruct-1b` model:

tests/e2e/test_spyre_basic.py

@@ -1,10 +1,5 @@
 """Verification of vLLM output by comparing with HF
 
-If errors occur, these can be analyzed/debugged by setting
-`DISABLE_ASSERTS = True` in spyre_util.py and by rerunning the
-test using `pytest --capture=no tests/spyre/test_spyre_basic.py`
-After debugging, `DISABLE_ASSERTS` should be reset to `False`.
-
 Run `python -m pytest tests/e2e/test_spyre_basic.py`.
 """
 

tests/e2e/test_spyre_max_new_tokens.py

@@ -31,11 +31,6 @@ def test_output(
     The same prompts are also input to HF. The generated output
     including text, token ids, and logprobs, is verified to be
     identical for vLLM and HF.
-
-    If errors occur, these can be analyzed/debugged by setting
-    'DISABLE_ASSERTS = True' in spyre_util.py and by rerunning the
-    test using 'pytest --capture=no tests/spyre/test_spyre_max_new_tokens.py'
-    After debugging, DISABLE_ASSERTS should be reset to 'False'.
     '''
 
     prompts = get_chicken_soup_prompts(4)

tests/e2e/test_spyre_warmup_shapes.py

@@ -34,11 +34,6 @@ def test_output(
     The same prompts are also input to HF. The generated
     output including text, token ids, and logprobs, is
     verified to be identical for vLLM and HF.
-
-    If errors occur, these can be analyzed/debugged by setting
-    'DISABLE_ASSERTS = True' in spyre_util.py and by rerunning the
-    test using 'pytest --capture=no tests/spyre/test_spyre_warmup_shapes.py'
-    After debugging, DISABLE_ASSERTS should be reset to 'False'.
     '''
 
     prompts = get_chicken_soup_prompts(4)