Merge pull request #309 from HazyResearch/bugfix/flash-fill-max-token

Bugfix/flash fill max token
HazyResearch · Mar 18, 2023 · a0564c6 · a0564c6
2 parents bfd44e0 + bb6de15
commit a0564c6
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/meerkat/interactive/app/src/lib/component/contrib/flash_fill/__init__.py b/meerkat/interactive/app/src/lib/component/contrib/flash_fill/__init__.py
@@ -25,8 +25,6 @@ def __init__(
         manifest_cache_dir: str = "~/.cache/manifest",
         max_tokens: int = 1,
     ):
-        self.max_tokens = max_tokens
-
         df = df.view()
         if target_column not in df.columns:
             df[target_column] = ""
@@ -41,6 +39,7 @@ def __init__(
             os.path.expanduser(manifest_cache_dir)
         )
         os.makedirs(self.manifest_cache_dir, exist_ok=True)
+        self.max_tokens = max_tokens
 
     @property
     def prompt(self):