@@ -36,64 +36,11 @@ jobs:
3636 - name : Install the project
3737 run : uv sync --locked --all-extras --dev
3838
39- - name : Run Fireworks Tracing Integration Tests
39+ - name : Run Fireworks Tracing Tests
4040 env :
4141 FIREWORKS_API_KEY : ${{ secrets.FIREWORKS_API_KEY }}
4242 PYTHONWARNINGS : " ignore::DeprecationWarning,ignore::RuntimeWarning"
4343 run : |
44- # Run Fireworks tracing integration tests
45- uv run pytest tests/logging/test_fireworks_tracing_integration.py -v --tb=short
46-
47- # Run RemoteRolloutProcessor with Fireworks tracing (if exists)
48- if [ -f "tests/remote_server/test_remote_fireworks.py" ]; then
49- echo "Running RemoteRolloutProcessor Fireworks integration test..."
50- # Note: This requires manual server startup, so may need to be run separately
51- # uv run pytest tests/remote_server/test_remote_fireworks.py -v --tb=short
52- fi
53-
54- - name : Run Status Propagation Tests
55- env :
56- FIREWORKS_API_KEY : ${{ secrets.FIREWORKS_API_KEY }}
57- PYTHONWARNINGS : " ignore::DeprecationWarning,ignore::RuntimeWarning"
58- run : |
59- # Run any status propagation tests that use Fireworks tracing
60- if [ -f "tests/remote_server/test_remote_fireworks_propagate_status.py" ]; then
61- echo "Running status propagation tests with Fireworks tracing..."
62- uv run pytest tests/remote_server/test_remote_fireworks_propagate_status.py -v --tb=short || true
63- fi
64-
65- fireworks-tracing-smoke-test :
66- name : Fireworks Tracing Smoke Test
67- runs-on : ubuntu-latest
68- needs : fireworks-tracing-tests
69-
70- steps :
71- - name : Checkout repository
72- uses : actions/checkout@v4
73-
74- - name : Set up Python 3.10
75- uses : actions/setup-python@v5
76- with :
77- python-version : " 3.10"
78-
79- - name : Install uv
80- uses : astral-sh/setup-uv@v6
81- with :
82- enable-cache : true
83-
84- - name : Install the project
85- run : uv sync --locked --all-extras --dev
86-
87- - name : Smoke Test - Basic Log Sending
88- env :
89- FIREWORKS_API_KEY : ${{ secrets.FIREWORKS_API_KEY }}
90- run : |
91- # Run just the basic log sending test as a smoke test
92- uv run pytest tests/logging/test_fireworks_tracing_integration.py::test_fireworks_tracing_handler_sends_logs -v
93-
94- - name : Smoke Test - Status Logging
95- env :
96- FIREWORKS_API_KEY : ${{ secrets.FIREWORKS_API_KEY }}
97- run : |
98- # Run just the status logging test as a smoke test
99- uv run pytest tests/logging/test_fireworks_tracing_integration.py::test_fireworks_tracing_handler_logs_status_info -v || true
44+ # Run RemoteRolloutProcessor Propagate Status Smoke Test (now uses Fireworks tracing)
45+ uv run pytest tests/remote_server/test_remote_fireworks_propagate_status.py::test_remote_rollout_and_fetch_fireworks_propagate_status \
46+ -v --tb=short
0 commit comments