make chained test actually pass

.github/workflows/run-tests.yaml

@@ -76,12 +76,12 @@ jobs:
           sleep 30  # wait for server to download layers
           
           # test individual blocks
-          export PYTHONPATH=. REF_NAME=bloom-testing/test-bloomd-350m
-          BLOCK_UID=bloom-testing/test-bloomd-350m.0 REF_INDEX=0 pytest tests/test_block_exact_match.py
-          BLOCK_UID=bloom-testing/test-bloomd-350m.19 REF_INDEX=19 pytest tests/test_block_exact_match.py
+          export PYTHONPATH=. MODEL_NAME=bloom-testing/test-bloomd-350m
+          BLOCK_UID=bloom-testing/test-bloomd-350m.0 REF_NAME=$MODEL_NAME REF_INDEX=0 pytest tests/test_block_exact_match.py
+          BLOCK_UID=bloom-testing/test-bloomd-350m.19 REF_NAME=$MODEL_NAME REF_INDEX=19 pytest tests/test_block_exact_match.py
+
+          REF_NAME=$MODEL_NAME pytest tests/test_chained_calls.py
           
-          BLOCK_UID=bloom-testing/test-bloomd-350m.3 REF_INDEX=3 pytest tests/test_chained_forward_backward.py
-          unset REF_NAME
           MODEL_NAME=bloom-testing/test-bloomd-350m REF_NAME=bigscience/bloom-350m pytest tests/test_full_model.py
           
           kill -s SIGINT $SERVER1_PID $SERVER2_PID