Merge pull request #24 from Arize-ai/juliagomes/update-repo-in-notebook

Juliagomes/fix notebook
Arize-ai · Jul 2, 2024 · a748440 · a748440
2 parents 1eaab90 + 589bd34
commit a748440
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/validator/__init__.py b/validator/__init__.py
@@ -1,3 +1,3 @@
-from .main import DatasetEmbeddings
+from .main import ArizeDatasetEmbeddings
 
-__all__ = ["DatasetEmbeddings"]
+__all__ = ["ArizeDatasetEmbeddings"]
diff --git a/validator/arize_demo_dataset_embeddings_guard.ipynb b/validator/arize_demo_dataset_embeddings_guard.ipynb
@@ -182,7 +182,7 @@
         "\n",
         "\n",
         "# can uncomment this once its public\n",
-        "!pip install  --no-cache-dir  -qq git+https://github.com/Arize-ai/jailbreak-embeddings\n",
+        "!pip install  --no-cache-dir  -qq git+https://github.com/Arize-ai/dataset-embeddings-guardrails\n",
         "\n",
         "from openinference.instrumentation.guardrails import GuardrailsInstrumentor\n",
         "from guardrails import Guard\n",