From be10f9a1a468aaa53f519f558c8c2fe9298c9af2 Mon Sep 17 00:00:00 2001
From: Gabefire <33893811+Gabefire@users.noreply.github.com>
Date: Wed, 20 Dec 2023 17:54:23 -0600
Subject: [PATCH] SN-94 added update_data_row_labeling_priority to notebook

---
 .../queue_management.ipynb                    | 73 ++++++++++++-------
 1 file changed, 45 insertions(+), 28 deletions(-)

diff --git a/examples/project_configuration/queue_management.ipynb b/examples/project_configuration/queue_management.ipynb
index 6de3a7c41..0919625d6 100644
--- a/examples/project_configuration/queue_management.ipynb
+++ b/examples/project_configuration/queue_management.ipynb
@@ -73,14 +73,14 @@
       "metadata": {},
       "source": [
         "# API Key and Client\n",
-        "Provide a valid api key below in order to properly connect to the Labelbox Client."
+        "See the developer guide for [creating an API key](https://docs.labelbox.com/reference/create-api-key)."
       ],
       "cell_type": "markdown"
     },
     {
       "metadata": {},
       "source": [
-        "# Add your api key\n",
+        "# Add your API key\n",
         "API_KEY = \"\"\n",
         "client = lb.Client(api_key=API_KEY)"
       ],
@@ -98,9 +98,7 @@
     {
       "metadata": {},
       "source": [
-        "## Create two Labelbox  projects\n",
-        "# Project defaults to batch mode with benchmark quality settings if queue mode argument is not provided\n",
-        "# Note that queue mode will be deprecated once dataset mode is deprecated \n",
+        "# Create Labelbox  project\n",
         "\n",
         "project = client.create_project(name=\"batch-test-project\",\n",
         "                                description=\"a description\",\n",
@@ -139,7 +137,7 @@
         "data_rows = dataset.create_data_rows(uploads)\n",
         "data_rows.wait_till_done()\n",
         "print(\"Errors\" , data_rows.errors)\n",
-        "print(\"Dataset status: \", data_rows.status)"
+        "print(\"Dataset status: \", data_rows.status)\n"
       ],
       "cell_type": "code",
       "outputs": [],
@@ -196,33 +194,48 @@
       "source": [
         "## Queue Order\n",
         "- Add priority for each data row\n",
-        "- Remove all the batch priority in your project"
+        "- Update priority for each data row"
       ],
       "cell_type": "markdown"
     },
     {
       "metadata": {},
       "source": [
-        "## See current LPOs\n",
-        "print(project.labeling_parameter_overrides().get_many(4))\n",
+        "# Select data rows from batches \n",
+        "data_rows = []\n",
+        "for batch in list(project.batches()):\n",
+        "  for data_row in batch.export_data_rows(): \n",
+        "    data_rows.append(data_row)\n",
         "\n",
+        "# Get label parameter overrides (LPOs)\n",
+        "project_lpos = list(project.labeling_parameter_overrides())\n",
         "\n",
-        "## Select data rows from batches \n",
-        "data_rows = []\n",
-        "for b in list(project.batches()):\n",
-        "  for dr in b.export_data_rows(): \n",
-        "    data_rows.append(dr)\n",
-        "  \n",
-        "## Add LPOs\n",
-        "lpos1 = []\n",
-        "p=1;\n",
-        "for dr in data_rows: \n",
-        "  lpos1.append((dr, p, 1))\n",
-        "  p+=1\n",
+        "for lpo in project_lpos:\n",
+        "  print(lpo)\n",
+        "  print(\"Data row:\", lpo.data_row().uid)"
+      ],
+      "cell_type": "code",
+      "outputs": [],
+      "execution_count": null
+    },
+    {
+      "metadata": {},
+      "source": [
+        "# Add LPOs\n",
+        "lpos = []\n",
+        "priority=1\n",
+        "for data_row in data_rows: \n",
+        "  lpos.append((data_row, priority, 1))\n",
+        "  priority+=1\n",
         "\n",
         "\n",
-        "project.set_labeling_parameter_overrides(lpos1)\n",
-        "# Get the project's LPOs"
+        "project.set_labeling_parameter_overrides(lpos)\n",
+        "\n",
+        "# Check results\n",
+        "project_lpos = list(project.labeling_parameter_overrides())\n",
+        "\n",
+        "for lpo in project_lpos:\n",
+        "  print(lpo)"
       ],
       "cell_type": "code",
       "outputs": [],
@@ -231,14 +244,18 @@
     {
       "metadata": {},
       "source": [
-        "## Verify LPOs\n",
+        "# Update LPOs\n",
+        "global_keys = []\n",
+        "for data_row in data_rows:\n",
+        "    global_keys.append(data_row.global_key)\n",
+        "\n",
+        "project.update_data_row_labeling_priority(data_rows=lb.GlobalKeys(global_keys), priority=1)\n",
+        "\n",
+        "# Check results\n",
         "project_lpos = list(project.labeling_parameter_overrides())\n",
-        "## Remove LPOs\n",
-        "# project.unset_labeling_parameter_overrides(dataset.export_data_rows())\n",
         "\n",
         "for lpo in project_lpos:\n",
-        "  print(lpo)\n",
-        "  print(\"Data row:\", lpo.data_row().uid)"
+        "  print(lpo)"
       ],
       "cell_type": "code",
       "outputs": [],