From 063291d816015cbdb04b66a8db6c6020b619252a Mon Sep 17 00:00:00 2001
From: Mohammad Nasirifar <m@fireworks.ai>
Date: Fri, 17 Jan 2025 19:50:38 +0000
Subject: [PATCH] Fix readme to reflect routing through enum

---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index d34b0984d..e3e6dc24d 100644
--- a/README.md
+++ b/README.md
@@ -343,7 +343,7 @@ class Operation(Enum):
     mul = partial(mul)
 
 model = outlines.models.transformers("WizardLM/WizardMath-7B-V1.1")
-generator = outlines.generate.json(model, add)
+generator = outlines.generate.json(model, Operation)
 result = generator("Return json with two float named c and d respectively. c is negative and d greater than 1.0.")
 
 print(result)