fix metadata removal for ExecutionCountRestartHelper (#2705)
This commit is contained in:
parent
b3d3c38b29
commit
7623aa46f9
@ -699,12 +699,12 @@ impl ExecutionCountRestartHelper {
|
||||
}
|
||||
|
||||
/// Clear progress for the stage this wrapper wraps.
|
||||
pub fn clear_progress<S>(&mut self, state: &mut S) -> Result<(), Error>
|
||||
pub fn clear_progress<S>(&mut self, state: &mut S, name: &str) -> Result<(), Error>
|
||||
where
|
||||
S: HasMetadata,
|
||||
S: HasNamedMetadata,
|
||||
{
|
||||
self.started_at_execs = None;
|
||||
let _metadata = state.remove_metadata::<ExecutionCountRestartHelperMetadata>();
|
||||
let _metadata = state.remove_named_metadata::<ExecutionCountRestartHelperMetadata>(name);
|
||||
debug_assert!(_metadata.is_some(), "Called clear_progress, but should_restart was not called before (or did mutational stages get nested?)");
|
||||
Ok(())
|
||||
}
|
||||
|
@ -260,7 +260,7 @@ where
|
||||
}
|
||||
|
||||
fn clear_progress(&mut self, state: &mut Self::State) -> Result<(), Error> {
|
||||
self.restart_helper.clear_progress(state)
|
||||
self.restart_helper.clear_progress(state, &self.name)
|
||||
}
|
||||
|
||||
fn perform(
|
||||
|
@ -296,7 +296,7 @@ where
|
||||
}
|
||||
|
||||
fn clear_progress(&mut self, state: &mut Self::State) -> Result<(), Error> {
|
||||
self.restart_helper.clear_progress(state)
|
||||
self.restart_helper.clear_progress(state, &self.name)
|
||||
}
|
||||
}
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user