From 3756cc068d2e08e573114f69980ae47e9085b92d Mon Sep 17 00:00:00 2001
From: Ryan <18477649+Ryul0rd@users.noreply.github.com>
Date: Thu, 18 May 2023 02:38:02 -0700
Subject: [PATCH] Remove leftover garbage from testing

---
 jsonformer/main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/jsonformer/main.py b/jsonformer/main.py
index f9e0280..4bb2d5d 100644
--- a/jsonformer/main.py
+++ b/jsonformer/main.py
@@ -181,7 +181,7 @@ class Jsonformer:
         # These are necessary because we don't know if we're at the end or middle of an object/array
         terminal_tokens = torch.concat([
             self.tokenizer.encode(s, add_special_tokens=False, return_tensors="pt")[:, 0]
-            for s in ('", "', '"}', '"]dsdsf')
+            for s in ('", "', '"}', '"]')
         ])
 
         highest_probability = 0.0