[Build] Fix notebooks (#1467)

Need an extra package for notebook testing.

Author: riedgar-ms

pyproject.toml

@@ -50,6 +50,7 @@ test = [
     "diskcache",
     "azure-identity",
     "bitsandbytes",
+    "datasets",
     "jupyter",
     "papermill",
     "pillow",