Gave training size to user

mritunjaysharma394 · Sep 17, 2020 · 205dcb1 · 205dcb1
1 parent 12ecf31
commit 205dcb1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -25,7 +25,7 @@
 
 tags = os.environ["INPUT_TAGS"]
 tagsVar= os.environ["INPUT_TAGSVAR"]
-trainSize = os.environ["INPUT_TRAINSIZE"]
+trainSize = int(os.environ["INPUT_TRAINSIZE"])
 
 tags_str = (str(tags).strip('[]')).split(",")
 tagsVar_str = (str(tagsVar).strip('[]')).split(",")