Skip to main content

RuntimeDriver

Struct RuntimeDriver 

Source
pub struct RuntimeDriver<E, M> { /* private fields */ }
Expand description

Drives one ready model/env session through its reset -> predict -> step loop. Inert until a run* method is awaited.

Implementations§

Source§

impl<E, M> RuntimeDriver<E, M>
where E: RuntimeEnv + Clone + 'static, M: RuntimeModel,

Source

pub fn new( spec: RuntimeSessionSpec, env: E, model: M, hooks: Arc<dyn RuntimeHooks>, ) -> Self

Source

pub fn with_prefetch(self, lead: u32) -> Self

Enable async inference: a group asks for its next action chunk while lead (or fewer) replay frames of the current one remain, so the forward overlaps the env steps. The chunk sees an observation up to lead steps stale, so results are NOT comparable to the synchronous loop — callers must label runs accordingly. lead == 0 leaves it off.

Source

pub fn with_scheduler(self, scheduler: Box<dyn PredictScheduler>) -> Self

Replace the predict scheduling policy (default: EagerScheduler).

Source

pub fn with_relay_policy(self, relay_policy: Arc<dyn RelayPolicy>) -> Self

Replace the relay policy (default: RefusingRelayPolicy).

Source

pub async fn run(self) -> Result<RuntimeReport, RuntimeError>

Source

pub async fn run_with_cancellation( self, cancellation: CancellationToken, ) -> Result<RuntimeReport, RuntimeError>

Source

pub async fn run_with_cancellation_reason( self, cancellation: CancellationToken, reason: impl Into<String>, ) -> Result<RuntimeReport, RuntimeError>

Runs the session, attributing any cancellation of cancellation to reason.

The reason is carried into RuntimeError::RouteCancelled, the session_failed hook event, and the ReleaseAdapter reason, so callers (e.g. an owner that cancels for Ctrl+C, a deadline, or a sibling-route failure) can supply an accurate cause instead of a hardcoded one.

Auto Trait Implementations§

§

impl<E, M> !Freeze for RuntimeDriver<E, M>

§

impl<E, M> !RefUnwindSafe for RuntimeDriver<E, M>

§

impl<E, M> !Sync for RuntimeDriver<E, M>

§

impl<E, M> !UnwindSafe for RuntimeDriver<E, M>

§

impl<E, M> Send for RuntimeDriver<E, M>
where E: Send, Option<Arc<M>>: Send,

§

impl<E, M> Unpin for RuntimeDriver<E, M>
where E: Unpin, Option<Arc<M>>: Unpin,

§

impl<E, M> UnsafeUnpin for RuntimeDriver<E, M>

Blanket Implementations§

Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
where ST: ?Sized, DT: ?Sized,

Source§

impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
where ST: ?Sized, DT: ?Sized,

Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

impl<T> Instrument for T

Source§

fn instrument(self, span: Span) -> Instrumented<Self> ⓘ

Instruments this type with the provided Span, returning an Instrumented wrapper. Read more
Source§

fn in_current_span(self) -> Instrumented<Self> ⓘ

Instruments this type with the current Span, returning an Instrumented wrapper. Read more
Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T> IntoRequest<T> for T

Source§

fn into_request(self) -> Request<T>

Wrap the input message T in a tonic::Request
Source§

impl<L> LayerExt<L> for L

Source§

fn named_layer<S>(&self, service: S) -> Layered<<L as Layer<S>>::Service, S>
where L: Layer<S>,

Applies the layer to a service and wraps it in Layered.
Source§

impl<T> Read<Exclusive, BecauseExclusive> for T
where T: ?Sized,

Source§

impl<T> Same for T

Source§

type Output = T

Should always be Self
Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = !

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, !>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.
Source§

impl<T> WithSubscriber for T

Source§

fn with_subscriber<S>(self, subscriber: S) -> WithDispatch<Self> ⓘ
where S: Into<Dispatch>,

Attaches the provided Subscriber to this type, returning a WithDispatch wrapper. Read more
Source§

fn with_current_subscriber(self) -> WithDispatch<Self> ⓘ

Attaches the current default Subscriber to this type, returning a WithDispatch wrapper. Read more