{/* Auto-generated by scripts/mdx_snippets_gen.py. Do not edit manually. */}

export const PySampleDataset = "from tempfile import TemporaryDirectory\n\nimport lancedb\nfrom lancedb.permutation import Permutation, permutation_builder\n\nwith TemporaryDirectory(prefix=\"lancedb-sampling-\") as uri:\n    db = lancedb.connect(uri)\n    table = db.create_table(\"records\", [{\"id\": i} for i in range(10)])\n    splits = (\n        permutation_builder(table).split_random(counts=[3, 7], seed=42).execute()\n    )\n    sample = Permutation.from_tables(table, splits, split=0).with_format(\"arrow\")\n    selected_ids = []\n    for batch in sample.iter(batch_size=3):\n        selected_ids.extend(batch.column(\"id\").to_pylist())\n    print(f\"Selected {len(selected_ids)} rows: {selected_ids}\")\n";