From c1a3377055754c7682bb6a9473457916ed35d22f Mon Sep 17 00:00:00 2001
From: HessTaha <tahahessane2@gmail.com>
Date: Sun, 31 May 2020 17:14:34 +0200
Subject: [PATCH] add max_length

---
 src/main.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/main.py b/src/main.py
index 3b5ae3d..fd274d9 100644
--- a/src/main.py
+++ b/src/main.py
@@ -56,7 +56,8 @@ def main(path_to_data: str,
     
     config_dict = {
         "vocab_size" : vocab_size,
-        "n_classes"  : n_classes
+        "n_classes"  : n_classes,
+        "max_length" : optimal_length
     }
 
     if n_classes > 2: