Method map

src/python/py_kernel.rs:111–137

(
        &self,
        spec: &PyKernelSpec,
        kwargs: Option<&Bound<'_, pyo3::types::PyDict>>,
    )

    /// Map a kernel spec over named arguments, returning a `MapSpec` for `rt.go()`.
    #[pyo3(signature = (spec, **kwargs))]
    fn map(
        &self,
        spec: &PyKernelSpec,
        kwargs: Option<&Bound<'_, pyo3::types::PyDict>>,
    ) -> PyResult<PyMapSpec> {
        let mut arg_map: HashMap<String, ArgValue> = HashMap::new();
        if let Some(kw) = kwargs {
            for (key, val) in kw {
                let name: String = key.extract()?;
                if let Ok(buf) = val.extract::<PyBuffer>() {
                    arg_map.insert(name, ArgValue::Buffer(buf.inner));
                } else if let Ok(v) = val.extract::<i64>() {
                    arg_map.insert(name, ArgValue::Scalar(v as f64));
                } else if let Ok(v) = val.extract::<f64>() {
                    arg_map.insert(name, ArgValue::Scalar(v));
                } else {
                    return Err(PyErr::new::<pyo3::exceptions::PyTypeError, _>(format!(
                        "argument '{name}' must be Buffer or numeric"
                    )));
                }
            }
        }
        Ok(PyMapSpec {
            spec: spec.clone(),
            args: arg_map,
        })
    }

    // --- Math functions (unary: inherit temp_args) ---