BUG: make a failing Monte Carlo worker say so - #1182
Conversation
fa81b7f to
65a2da2
Compare
Codecov Report❌ Patch coverage is
Additional details and impacted files@@ Coverage Diff @@
## develop #1182 +/- ##
===========================================
+ Coverage 91.24% 91.50% +0.26%
===========================================
Files 131 131
Lines 17619 17723 +104
===========================================
+ Hits 16076 16217 +141
+ Misses 1543 1506 -37 ☔ View full report in Codecov by Harness. 🚀 New features to boost your workflow:
|
6b260bb to
2adc9d4
Compare
There was a problem hiding this comment.
Pull request overview
Fixes parallel Monte Carlo seed handling and ensures failed workers cannot silently hang or produce incomplete successful runs.
Changes:
- Normalizes
SeedSequencevalues without collapsing worker streams. - Adds worker failure reporting, exit-code validation, and bounded shutdown polling.
- Adds regression coverage for parallel execution and worker failure modes.
Reviewed changes
Copilot reviewed 8 out of 8 changed files in this pull request and generated 2 comments.
Show a summary per file
| File | Description |
|---|---|
rocketpy/tools.py |
Adds deterministic SeedSequence conversion. |
rocketpy/stochastic/stochastic_model.py |
Accepts worker seed sequences. |
rocketpy/simulation/monte_carlo.py |
Improves worker reporting and lifecycle handling. |
tests/unit/stochastic/test_seed_types.py |
Tests seed compatibility and stream independence. |
tests/unit/simulation/test_monte_carlo_parallel_runs.py |
Exercises real serial and parallel runs. |
tests/unit/simulation/test_monte_carlo_worker_reporting.py |
Tests worker failure diagnostics. |
tests/unit/simulation/test_monte_carlo_worker_join.py |
Tests polling and shutdown behavior. |
tests/unit/simulation/test_monte_carlo_worker_exit.py |
Tests abnormal worker exit detection. |
💡 Add a code-review agent skill or configure MCP servers for context-aware, tailored reviews. Learn more in the docs.
|
Thanks for the approve, @Gui-FernandesBR. Flagging that the branch has moved a fair way since you looked at it, so you can decide whether you want another pass. TL;DR: four commits landed after your approve ( What landed, and why:
The part worth your eye is the reversal. I had made a reported failure end the parent's wait, which reads sensibly and turns out to be wrong: a worker that reports has left nothing behind, and its siblings stop on their own once they finish the simulation in hand. With the event treated as a reason to stop, a sibling that needed forty polls was terminated after three. It was not stuck, it was mid-simulation, and it would have exited cleanly with its rows intact. Only a worker that died can hold the shared lock for good, so only that ends the wait now. There is a test pinning it, and reverting the line turns that test red. One thing I have not covered and would rather say than imply: the monotonic clock is a judgement call, not a tested one. Pinning it would mean moving the system clock, which I do not think belongs in CI. Nothing here needs action from you if you are happy with it as is. If you would rather the extra work went somewhere separate, I am glad to move those four out into their own PR and put this back to what you approved. Apologies for the moving target: I kept finding things while reviewing my own answer to the last round, and pushed each as I went rather than waiting. It is settled now. |
bb743e4 to
629cd87
Compare
|
Heads up that I pushed 75b902e after this was approved, so the approval no longer covers the head. Three things changed, all in the failure path. The reporting lock. Index types. Paired logs. The two were validated independently. A record goes into both under one lock and the inputs write rolls back if the outputs write fails, so their index sequences have to match. They are compared now. Full |
|
Another push after the approval, e2e0b64, so the approval no longer covers the head. It answers a liveness gap that I had described wrongly above, which is worth being explicit about. The description was wrong. It said a reported failure ends the wait. The code said the opposite, in the join docstring, and the code is what ran: The event ends the wait now, on a longer grace than the exit-code path, since a sibling that only read the event is working rather than blocked on a lock nobody owns. Slowness alone still ends nothing: with no failure reported a healthy worker is given as long as it needs, which the existing control test pins. This does change a decision the branch had made deliberately, and the test that recorded it. That test said a sibling mid-simulation should be left alone. I think the trade goes the other way: once a failure is reported the run is short a simulation and the completeness check refuses it, so that sibling's row belongs to a run that has already failed, while the hang costs the caller any error at all. The sixty seconds is a judgement call with nothing principled behind it, and I would happily take a different number. Three smaller things in the same commit.
Full |
|
Conflicts are gone. The branch is squashed to one commit and rebased onto current On the stacking: this one sat on #1181, which has landed, so the diff here is its own work again. 317 lines of I ran the suite against a |
Six ways a worker that failed got away without saying so. It died inside its own handler on an unbound name, so the manager lock it was holding was never given back and the run waited for good. A worker that was killed ran no handler, set no event, and left only an exit code nobody read, so simulate() returned normally with rows missing. Reporting a failure could itself block on a lock a dead sibling still held. Both names are bound before the try now, the handler reports through a bounded lock and releases it from a finally, and the parent reads exit codes and the failure event instead of joining unbounded. A run that is only slow is still never cut short: how a worker ended decides that, not how long it took. An exit code cannot show a worker that left between claiming an index and recording it, so a run is checked against its own logs at the end. Both must hold exactly the simulations asked for, none twice, none unreadable. Scope is the parallel producer. __run_in_serial has the same unbound name and belongs to RocketPy-Team#1177. Signed-off-by: thc1006 <84045975+thc1006@users.noreply.github.com>
5450887 to
81711b8
Compare
A Monte Carlo run cannot be seeded on develop: simulate() takes no random_seed. Seeding it per simulation index is not enough on its own, because an append then derives a fresh root and writes it into the same file, so a study resumed after a restart holds two lineages with nothing afterwards to say which simulation came from which. Both halves are here. A simulation takes its seed from its own index, so a serial run and a run split over workers draw the same inputs for the same index. Every input row records the root that drew it, and an append reads it back rather than needing to be given it again. A seed that disagrees with the rows is refused, as is a log whose rows disagree with each other, and one whose rows carry no root at all, which is how a log written before this looks. Output rows carry a digest of that root, so a log belonging to another study is refused even when its indices line up with this one's. RocketPy-Team#1182's worker tests drive the producer with a stand-in monitor, so they move to the claim along with it. A reseed failure now names the index it was seeding for rather than worker startup, because the seeding happens after the claim rather than once above the loop. The seeding half was RocketPy-Team#1054, closed in favour of this. Addresses RocketPy-Team#1053 and RocketPy-Team#1075. Signed-off-by: thc1006 <84045975+thc1006@users.noreply.github.com>
Six ways a failing Monte Carlo worker got away without saying so. Three of them came out of review of the first three, which is why this is longer than it started.
Was stacked on #1181, which is the parallel run being unable to start at all. That has landed and this branch now sits on top of it, so it no longer shows in the diff here. What is left is this pull request's own work: 317 lines of
monte_carlo.pyand five test files.Pull request type
Checklist
ruff check/ruff format --check,pylint) has passed locallydevelop, listed below.CHANGELOG.md: one line for this one. The workflow that writes them has not run since CI: run the changelog job for pull requests from forks #1112, which is BUG: the changelog workflow stopped running, and CHANGELOG.md is 37 merged pull requests behind #1173.Current behavior
A worker that fails early dies inside its own handler.
__sim_producerbindssim_idxandinputs_jsoninside the simulation loop:and its
exceptblock reads both. A worker that fails in the seeding above the loop, or in the claim that opens it, arrives there with neither name assigned:That happens after
mutex.acquire()and beforemutex.release(), so a manager lock is left held by a process that no longer exists. The next worker blocks onacquire()and the run waits forever.error_event.set()is the handler's last line, so the parent is never told either.A worker that is killed is not noticed at all. The parent joins each worker and reads the event. A signal runs no handler, so the event stays clear and the join returns because the process is gone. With a
SIGKILLin the second of six simulations across two workers:I would rather have the hang than that one. A hang is at least visible.
New behavior
Both names are bound before the
try, and the message saysworker startupwhen no index was claimed instead of naming one that does not exist. The handler now finishes, releases the mutex and sets the event, so a worker failure surfaces as aRuntimeErrornaming the error file rather than as a wait.After the join,
_refuse_a_worker_that_did_not_finishreads the exit codes. Anything other than zero is refused. It sits before the event check because the two are disjoint: a worker that reports through its handler exits cleanly, and one that was killed only leaves its exit code behind.The handler holds the shared mutex while it reports, so a failure in the reporting used to leave the lock held by a process that had already gone. The event is set first and from outside the lock, the lock is released from a
finally, and each reporting step is separate so an unwritable log cannot replace the failure being reported. A startup failure writes a row of its own now, since the caller is told to read that file and a traceback a worker printed is not there once its output is redirected.The lock belongs to the manager and is not released when its holder is killed, so a sibling can block on a lock nobody owns while an unbounded join waits with it. That is the case the exit-code check exists for and the one it could not see.
_join_the_workerspolls instead, and acts only once a worker has actually ended badly. A run that is merely slow is never bounded: an exit code, not a duration, decides. This is deliberately not a general worker-lifecycle change; it is the smallest thing that makes the check above reachable.Scope is the parallel producer.
__run_in_serialhas the same unboundinputs_jsonand belongs to #1177, whose tests cover it already; I have deliberately not touched it.Breaking change
A run that used to hang now raises, and a run that used to return with missing rows now raises. Both were already failures.
Additional information
Verification, on a clean tree:
The ten are the same ten on both, which is why I ran
developat the commit this branch sits on. Each of them is an optional dependency this machine does not have:statsmodelsandprettytablefor the sensitivity tests,imageiofor the ellipses one, andtimezonefinder,windrose,ipywidgetsandjsonpicklefor the environment analysis ones.Mutations, each leaving a control standing:
tryinputs_jsonis bound, which is what the traceback points attest_a_worker_that_leaves_early_does_not_pass_as_a_finished_runif process.exitcodein place of!= 0Nonecasetry/finallytest_a_reported_failure_ends_the_waittest_no_failure_path_waits_on_a_worker_without_a_boundtest_a_worker_that_leaves_cleanly_without_recording_is_not_a_successCounts are from this branch's own test files, with a green control before each. The last two rows include
test_monte_carlo_parallel_runs.pyand the rest do not, so those two are over five files and the others over four. They were measured before the branch was squashed and rebased ontobc3fe735, which touches nothing this pull request changes. The later runs take their baseline fromgit show HEAD:rather than from a copy, after one run left a file mutated and the next took that as its baseline; the file was checked clean againstHEADafter every run.Two rows there are worth pointing at. Fixing only the name the traceback names looks complete and leaves the message raising on the other one, and the same 14 tests catch either version. And ending the wait as soon as any worker has finished, rather than on how one finished, takes down the test that says a slow run must be left alone. That is the failure mode that would hurt real users most, since it stops a run that was only taking its time.
Three more, each of which let a broken run still look finished. They were raised in review of the commits above and all three reproduced.
The join waited only on exit status. A worker that fails the ordinary way is caught by the producer, reports through the event and returns, so it exits cleanly. With a sibling stuck, the parent saw one clean exit and one live process and never stopped. A reported failure ends the wait now too. A manager that cannot be asked is not taken as evidence either way, which has its own control test.
The bounded shutdown was undone one level up. After the exit-code check raised, the outer handler joined every process again with no timeout, so the stubborn worker the bound exists for was waited on anyway. That handler uses the same teardown now, and a test walks the parallel path's syntax to keep an unbounded join out of it.
An exit code says how a process ended, never whether the index it claimed reached the logs. The monitor counts claims, not rows, so it cannot stand in either. Measured, six simulations across two workers leaving through
os._exit(0):The run is checked against its own logs at the end: both must hold exactly the simulations asked for, none twice, none unreadable, none numbered past the run.
Still not covered, and I would rather say so than imply otherwise: a worker killed between
mutex.acquire()and the write it is guarding leaves a torn record behind. The run is stopped and reported now, but the rows it was midway through are not repaired._refuse_a_worker_that_did_not_finishis a module-level function rather than a method because the run paths are driven by stub objects in the tests, whereself.__helper()would not resolve.The stand-in worker in the join tests gives up after two hundred polls. Reproducing a real hang there would take a CI job down with it rather than report, and
requirements-tests.txthas nopytest-timeout; I found that out by hanging my own mutation run for the full fifteen minutes.The Monte Carlo objects in these tests are built on
tmp_pathrather than retargeted after construction.filenameis a plain attribute and the three log paths are settled in__init__, so assigning it leaves a test writing the fixture's own files into the working directory, which is what these were doing.#1169 and #1170 have landed since I first ran that, so it is not a throwaway tree any more. This branch is rebased onto current
developand the suite ran on the result. The fixed-seed baseline came out the same:stochastic_calistounder seed 42 readsmass=14.906007947 radius=0.063501935.Every row above was measured again on the current tree, since a mutation table is a claim about the tests and the tests had moved. Two of the counts had been written as five and as eight and are fourteen and twelve now, because the branch gained tests after the table was first written. Two anchors were wrong on the first attempt and gave nothing:
bool(exitcode)is the same function asnot in (None, 0)for an int orNone, and shortening the join poll does nothing to a stand-in worker. Both were redone against the real shape.