pan class='kw'>sample_ids_dtype</span> <span class='kw'>=</span> <span class='kw'>NULL</span>
)</pre>

    <h2 class="hasAnchor" id="arguments"><a class="anchor" href="#arguments"></a>Arguments</h2>
    <table class="ref-arguments">
    <colgroup><col class="name" /><col class="desc" /></colgroup>
    <tr>
      <th>initialize_fn</th>
      <td><p>callable that returns (finished, next_inputs) for the first iteration.</p></td>
    </tr>
    <tr>
      <th>sample_fn</th>
      <td><p>callable that takes (time