bugfix

MilaNLProc · Sep 6, 2024 · ab5c009 · ab5c009
1 parent 407aaba
commit ab5c009
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git 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 <[email protected]>"]
 license = "Apache Software License 2.0"

diff --git 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(