From ab5c009c7a1df7216046a457455f05e737f1965a Mon Sep 17 00:00:00 2001
From: Giuseppe Attanasio <giuseppeattanasio6@gmail.com>
Date: Fri, 6 Sep 2024 18:11:10 +0000
Subject: [PATCH] bugfix

---
 pyproject.toml                         | 2 +-
 simple_generation/simple_generation.py | 4 ----
 2 files changed, 1 insertion(+), 5 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index ed4d05b..ed4a90f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "simple-generation"
-version = "0.6.2"
+version = "0.6.3"
 description = "A python package to run inference with HuggingFace checkpoints wrapping many convenient features."
 authors = ["Giuseppe Attanasio <giuseppeattanasio6@gmail.com>"]
 license = "Apache Software License 2.0"
diff --git a/simple_generation/simple_generation.py b/simple_generation/simple_generation.py
index b050579..9984429 100644
--- a/simple_generation/simple_generation.py
+++ b/simple_generation/simple_generation.py
@@ -344,10 +344,6 @@ def __call__(
             )
             text_inputs = [text for _, text in idx2text_mapping]
 
-        import pdb
-
-        pdb.set_trace()
-
         # Processing the input text
         dataset = Dataset.from_dict({"text": text_inputs})
         dataset = dataset.map(