add badges to README

ggordonhall · Dec 8, 2018 · b503a02 · b503a02
1 parent 5cbc86e
commit b503a02
Showing 4 changed files with 7 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -1,4 +1,7 @@
 # Measurement Tagger
+[![Build Status](https://travis-ci.org/ggordonhall/measurement_tagger.png)](https://travis-ci.org/ggordonhall/measurement_tagger)
+[![codecov](https://codecov.io/gh/ggordonhall/measurement_tagger/branch/master/graph/badge.svg)](https://codecov.io/gh/ggordonhall/measurement_tagger)
+
 ## A dependency parse based measurement tagger.
 
 Text to be tagged should be stored in the `text/` directory. The file to tag is specific with `-t`.

diff --git a/modules/extractor.py b/modules/extractor.py
@@ -47,5 +47,4 @@ def extract(self):
         for sent in self._sentences:
             measures = self._tagger.tag(sent)
             if measures:
-                for measure in self._converter.convert(measures):
-                    yield measure
+                yield from self._converter.convert(measures)
diff --git a/modules/loader.py b/modules/loader.py
@@ -30,15 +30,13 @@ def _line_iter(self):
         """
 
         with open(self._filepath, 'r') as f:
-            for line in f:
-                yield self._formatter.format(line)
+            yield from (self._formatter.format(line) for line in f)
 
     def _tokenise(self):
         raise NotImplementedError
 
     def __iter__(self):
-        for sentence in self._tokenise():
-            yield sentence
+        yield from self._tokenise()
 
 
 class SentenceLoader(AbstractLoader):

diff --git a/modules/tagger.py b/modules/tagger.py
@@ -77,6 +77,7 @@ def _measurements(self, token, unit):
                 r_measure = Measurement(
                     right_mod, self._right_mod_tokens[token.lemma_])
                 modifiers.append(r_measure)
+
         return modifiers
 
     def _find_mod(self, direction, token):