Commit fa3ea2a2 authored by Vasyl Bodnaruk's avatar Vasyl Bodnaruk

Fix bug with empty part of tags

parent 7240c7e8
......@@ -8,12 +8,13 @@ News = namedtuple('News', 'id, tags')
class MLDataMaker:
def __init__(self, tags_list):
self.tags = [self.split_tags(i, ' /') for i in tags_list]
self.tags = [self.split_tags(i, '/() ') for i in tags_list]
def split_tags(self, tags, char, replace=' '):
if len(char) > 1:
for i in char:
tags = tags.replace(i, replace)
tags = ' '.join([i for i in tags.split(replace) if len(i) > 0])
return tags.split(replace)
return tags.split(char)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment