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') ...@@ -8,12 +8,13 @@ News = namedtuple('News', 'id, tags')
class MLDataMaker: class MLDataMaker:
def __init__(self, tags_list): 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=' '): def split_tags(self, tags, char, replace=' '):
if len(char) > 1: if len(char) > 1:
for i in char: for i in char:
tags = tags.replace(i, replace) 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(replace)
return tags.split(char) 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