Minor adjustments to the example notebooks and tests (#864)

* Minor adjustments to the example notebooks and tests

Since some of the upstream code re: shapes and sending requests changed in conjunction with the recent Dataloaders and Models changes, there are some small tweaks required to get the example tests passing again.

* Uncomment the `shutil.rmtree` clean-up

examples/getting-started-movielens/04-Triton-Inference-with-TF.ipynb

@@ -78,7 +78,10 @@
     "df_lib = get_lib()\n",
     "\n",
     "import tritonclient.grpc as grpcclient\n",
-    "import nvtabular.inference.triton as nvt_triton"
+    "import merlin.systems.triton as merlin_triton\n",
+    "\n",
+    "import merlin.dtypes as md\n",
+    "from merlin.schema import ColumnSchema, Schema"
    ]
   },
   {
@@ -343,7 +346,12 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "inputs = nvt_triton.convert_df_to_triton_input([\"userId\", \"movieId\"], batch, grpcclient.InferInput)\n",
+    "request_schema = Schema([\n",
+    "    ColumnSchema(\"userId\", dtype=md.int64),\n",
+    "    ColumnSchema(\"movieId\", dtype=md.int64),\n",
+    "])\n",
+    "\n",
+    "inputs = merlin_triton.convert_df_to_triton_input(request_schema, batch, grpcclient.InferInput)\n",
     "\n",
     "outputs = [\n",
     "    grpcclient.InferRequestedOutput(col)\n",

tests/unit/examples/test_building_deploying_multi_stage_RecSys.py

@@ -74,10 +74,11 @@ def test_func():
             response = run_ensemble_on_tritonserver(
                 "/tmp/examples/poc_ensemble", ensemble.graph.input_schema, batch, outputs,  "executor_model"
             )
-            response = [x.tolist()[0] for x in response["ordered_ids"]]
+            ordered_ids = [x.tolist() for x in response["ordered_ids"]]
             shutil.rmtree("/tmp/examples/", ignore_errors=True)
             """
         )
         tb2.execute_cell(NUM_OF_CELLS - 2)
-        response = tb2.ref("response")
-        assert len(response) == top_k
+        ordered_ids = tb2.ref("ordered_ids")
+
+        assert len(ordered_ids[0]) == top_k