Merge pull request Cartucho#1 from laxos96/laxos96-patch-1

remove_space.py only handled correctly last class
lnt28 · Jun 11, 2018 · 6ef4482 · 6ef4482
2 parents 6f78b0b + 66fbfe9
commit 6ef4482
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/extra/remove_space.py b/extra/remove_space.py
@@ -77,7 +77,7 @@ def rename_class(current_class_name, new_class_name):
 
 with open('class_list.txt') as f:
     for line in f:
-        current_class_name = line
+        current_class_name = line.rstrip("\n")
         new_class_name = line.replace(' ', args.delimiter)
         if line == new_class_name:
             continue
@@ -94,4 +94,4 @@ def rename_class(current_class_name, new_class_name):
           os.chdir("../predicted")
           rename_class(current_class_name, new_class_name)
 
-print('Done!')
+print('Done!')