Return the ideal batch size for this function
(&self)
| 98 | |
| 99 | /// Return the ideal batch size for this function |
| 100 | pub fn ideal_batch_size(&self) -> Result<Option<usize>> { |
| 101 | if let Some(expr) = self.func.downcast_ref::<ScalarFunctionExpr>() |
| 102 | && let Some(udf) = expr.fun().inner().downcast_ref::<AsyncScalarUDF>() |
| 103 | { |
| 104 | return Ok(udf.ideal_batch_size()); |
| 105 | } |
| 106 | not_impl_err!("Can't get ideal_batch_size from {:?}", self.func) |
| 107 | } |
| 108 | |
| 109 | /// This (async) function is called for each record batch to evaluate the LLM expressions |
| 110 | /// |
no test coverage detected