cange default outputs to 1

treebeardtech · Oct 25, 2023 · 412b0ab · 412b0ab
1 parent ff1ce56
commit 412b0ab
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 9 deletions.
diff --git a/README.md b/README.md
@@ -35,13 +35,7 @@ export OPENAI_API_KEY='sk-xxxx'
 e.g. nbwrite/example1.yaml:
 ```yaml
 task: |
-  Create a hello world notebook 'x.ipynb', use nbmake's NotebookRun class to test it from a Python application
-steps:
-  - Create a hello world notebook using nbformat
-  - Use nbmake's NotebookRun class to execute it from a Python application
-  - Check the output notebook printed what we were expecting
-packages:
-  - nbmake
+  Plot the iris dataset using pandas
 generation:
   count: 2
 ```

diff --git a/src/nbwrite/config.py b/src/nbwrite/config.py
@@ -11,7 +11,7 @@
 
 
 class GenerationConfig(BaseModel):
-    count: int = 2
+    count: int = 1
     system_prompt: str = DEFAULT_SYSTEM_PROMPT
     llm_kwargs: Dict[str, Any] = DEFAULT_LLM_KWARGS
     retriever_kwargs: Dict[str, Any] = DEFAULT_RETRIEVER_KWARGS

diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -85,4 +85,4 @@ def test_cli_minimal(tmpdir: local):
 
         logger.warn(f"Checking outputs in {outdir}")
         outputs = list(Path(outdir).glob("*.ipynb"))
-        assert len(outputs) == 2
+        assert len(outputs) == 1