Revise vowpal_wabbit notebook

9 months ago · a3a2ce623e
parent 8fafa1af91
commit a3a2ce623e
3 changed files with 841 additions and 834 deletions
--- a/docs/extras/modules/chains/how_to/learned_prompt_optimization.ipynb
+++ b/docs/extras/modules/chains/how_to/learned_prompt_optimization.ipynb
--- a/docs/extras/use_cases/more/learned_prompt_optimization.ipynb
+++ b/docs/extras/use_cases/more/learned_prompt_optimization.ipynb
--- a/libs/experimental/langchain_experimental/rl_chain/model_repository.py
+++ b/libs/experimental/langchain_experimental/rl_chain/model_repository.py
@ -45,7 +45,13 @@ class ModelRepository:
            shutil.copyfile(self.model_path, self.folder / f"model-{self.get_tag()}.vw")

    def load(self, commandline: List[str]) -> "vw.Workspace":
-        import vowpal_wabbit_next as vw
+        try:
+            import vowpal_wabbit_next as vw
+        except ImportError as e:
+            raise ImportError(
+                "Unable to import vowpal_wabbit_next, please install with "
+                "`pip install vowpal_wabbit_next`."
+            ) from e

        model_data = None
        if self.model_path.exists():