add file creation

ChEB-AI · Nov 29, 2023 · 17a8760 · 17a8760
1 parent ac4017b
commit 17a8760
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/chebai/preprocessing/reader.py b/chebai/preprocessing/reader.py
@@ -89,7 +89,7 @@ def __init__(self, *args, **kwargs):
         dirname = os.path.dirname(__file__)
         os.makedirs(os.path.join(dirname, "bin", self.name()), exist_ok=True)
         self.tokens_path = os.path.join(dirname, "bin", self.name(), "tokens.txt")
-        with open(self.tokens_path, "r") as pk:
+        with open(self.tokens_path, "r+") as pk:
             self.cache = [x.strip() for x in pk]
 
     def _get_token_index(self, token):