fix: cache key as string

dottxt-ai · Sep 4, 2024 · 8ae4086 · 8ae4086
1 parent 72377db
commit 8ae4086
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/outlines/caching.py b/outlines/caching.py
@@ -126,7 +126,8 @@ def wrapper(*args, **kwargs):
 
         def __cache_key__(*args, **kwargs):
             """Make key for cache given function arguments."""
-            return args_to_key(base, args, kwargs, typed, ignore)
+            # return args_to_key(base, args, kwargs, typed, ignore)
+            return str(args_to_key(base, args, kwargs, typed, ignore))
 
         wrapper.__cache_key__ = __cache_key__  # type: ignore
         wrapper.__memory__ = memory  # type: ignore