import * as ort from 'onnxruntime-web';
import { bq as PolicyActionChunk } from './types-Bu8Aki_j.js';
import 'react';
import '@react-three/fiber';
import 'three';

/**
 * @license
 * SPDX-License-Identifier: Apache-2.0
 *
 * Optional ONNX Runtime Web helpers for browser policy demos.
 *
 * This entry point is exported as `mujoco-react/onnx` so the main package does
 * not import or bundle `onnxruntime-web`.
 */

type OnnxPolicyDtype = 'float32' | 'float64' | 'int32' | 'int64' | 'bool' | string;
interface OnnxPolicyTensorSpec {
    name: string;
    shape: number[];
    dtype: OnnxPolicyDtype;
}
interface OnnxPolicyImageSpec {
    width: number;
    height: number;
    channels?: number;
    layout?: 'CHW' | 'HWC' | string;
    range?: readonly [number, number];
}
interface OnnxPolicyManifest {
    model: string;
    variants?: Record<string, string>;
    fps?: number;
    joints?: string[];
    cameras?: string[];
    image?: OnnxPolicyImageSpec;
    chunk_size?: number;
    n_action_steps?: number;
    inputs: OnnxPolicyTensorSpec[];
    output: OnnxPolicyTensorSpec & {
        units?: string;
    };
    [key: string]: unknown;
}
interface LoadOnnxPolicyManifestResult<TManifest extends OnnxPolicyManifest = OnnxPolicyManifest> {
    manifest: TManifest;
    manifestUrl: URL;
    modelUrl: URL;
}
interface CreateOnnxPolicySessionOptions<TManifest extends OnnxPolicyManifest = OnnxPolicyManifest> {
    manifestUrl: string | URL;
    variant?: string;
    runtime: typeof ort;
    sessionOptions?: ort.InferenceSession.SessionOptions;
    fetcher?: typeof fetch;
    readManifest?: (response: Response) => Promise<TManifest>;
}
interface OnnxPolicySession<TManifest extends OnnxPolicyManifest = OnnxPolicyManifest> extends LoadOnnxPolicyManifestResult<TManifest> {
    session: ort.InferenceSession;
}
declare function loadOnnxPolicyManifest<TManifest extends OnnxPolicyManifest = OnnxPolicyManifest>(manifestUrlInput: string | URL, options?: Pick<CreateOnnxPolicySessionOptions<TManifest>, 'variant' | 'fetcher' | 'readManifest'>): Promise<LoadOnnxPolicyManifestResult<TManifest>>;
declare function createOnnxPolicySession<TManifest extends OnnxPolicyManifest = OnnxPolicyManifest>(options: CreateOnnxPolicySessionOptions<TManifest>): Promise<OnnxPolicySession<TManifest>>;
declare function onnxTensorToPolicyActionChunk(tensor: ort.Tensor, actionSize?: number, maxActions?: number): PolicyActionChunk;

export { type CreateOnnxPolicySessionOptions, type LoadOnnxPolicyManifestResult, type OnnxPolicyDtype, type OnnxPolicyImageSpec, type OnnxPolicyManifest, type OnnxPolicySession, type OnnxPolicyTensorSpec, createOnnxPolicySession, loadOnnxPolicyManifest, onnxTensorToPolicyActionChunk };