Adding training dependencies to notebooks, closes #669

examples/16_Train_a_text_labeler.ipynb

@@ -52,7 +52,7 @@
       },
       "source": [
         "%%capture\n",
-        "!pip install git+https://github.com/neuml/txtai datasets pandas"
+        "!pip install git+https://github.com/neuml/txtai#egg=txtai[pipeline-train] datasets pandas"
       ],
       "execution_count": null,
       "outputs": []

examples/17_Train_without_labels.ipynb

@@ -45,7 +45,7 @@
       },
       "source": [
         "%%capture\n",
-        "!pip install git+https://github.com/neuml/txtai datasets pandas"
+        "!pip install git+https://github.com/neuml/txtai#egg=txtai[pipeline-train] datasets pandas"
       ],
       "execution_count": null,
       "outputs": []

examples/19_Train_a_QA_model.ipynb

@@ -47,7 +47,7 @@
       },
       "source": [
         "%%capture\n",
-        "!pip install git+https://github.com/neuml/txtai datasets pandas"
+        "!pip install git+https://github.com/neuml/txtai#egg=txtai[pipeline-train] datasets pandas"
       ],
       "execution_count": null,
       "outputs": []
@@ -292,4 +292,4 @@
       ]
     }
   ]
-}
+}

examples/20_Extractive_QA_to_build_structured_data.ipynb

@@ -42,7 +42,7 @@
       },
       "source": [
         "%%capture\n",
-        "!pip install git+https://github.com/neuml/txtai"
+        "!pip install git+https://github.com/neuml/txtai#egg=txtai[pipeline-train]"
       ],
       "execution_count": null,
       "outputs": []
@@ -526,4 +526,4 @@
       ]
     }
   ]
-}
+}

examples/41_Train_a_language_model_from_scratch.ipynb

@@ -57,7 +57,7 @@
       "outputs": [],
       "source": [
         "%%capture\n",
-        "!pip install git+https://github.com/neuml/txtai datasets sentence-transformers onnxruntime onnx"
+        "!pip install git+https://github.com/neuml/txtai#egg=txtai[pipeline-train] datasets sentence-transformers onnxruntime onnx"
       ]
     },
     {
@@ -495,4 +495,4 @@
       }
     }
   ]
-}
+}