remove_space.py only handled correctly last class

Added .rstrip("\n") at line reading so that it reads properly all clases, instead of only the last one which does not contain the "\n" character.
jaffe-fly · Jun 11, 2018 · 66fbfe9 · 66fbfe9
1 parent 6f78b0b
commit 66fbfe9
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!')