chg: Linting with black

Author: ivyleavedtoadflax

deep_reference_parser/__init__.py

@@ -19,12 +19,16 @@
 
 from .common import download_model_artefact
 from .deep_reference_parser import DeepReferenceParser
-from .io import (load_tsv, read_jsonl, read_pickle, write_jsonl, write_pickle,
-                 write_to_csv, write_tsv)
+from .io import (
+    load_tsv,
+    read_jsonl,
+    read_pickle,
+    write_jsonl,
+    write_pickle,
+    write_to_csv,
+    write_tsv,
+)
 from .logger import logger
 from .model_utils import get_config
 from .reference_utils import break_into_chunks
 from .tokens_to_references import tokens_to_references
-
-
-

deep_reference_parser/prodigy/labels_to_prodigy.py

@@ -1,4 +1,3 @@
-
 def labels_to_prodigy(tokens, labels):
     """
     Converts a list of tokens and labels like those used by Rodrigues et al,

deep_reference_parser/prodigy/misc.py

@@ -1,11 +1,13 @@
 import spacy
 
+
 def _join_prodigy_tokens(text):
     """Return all prodigy tokens in a single string
     """
 
     return "\n".join([str(i) for i in text])
 
+
 def prodigy_to_conll(docs):
     """
     Expect list of jsons loaded from a jsonl

deep_reference_parser/prodigy/prodigy_to_tsv.py

@@ -20,7 +20,7 @@
 
 msg = Printer()
 
-ROWS_TO_PRINT=15
+ROWS_TO_PRINT = 15
 
 
 class TokenLabelPairs:

deep_reference_parser/reference_utils.py

@@ -6,6 +6,7 @@
 
 from .logger import logger
 
+
 def yield_token_label_pairs(tokens, labels):
     """
     Convert matching lists of tokens and labels to tuples of (token, label) but
@@ -24,6 +25,7 @@ def yield_token_label_pairs(tokens, labels):
         else:
             yield (None, None)
 
+
 def break_into_chunks(doc, max_words=250):
     """
     Breaks a list into lists of lists of length max_words

tests/prodigy/test_misc.py

@@ -1,5 +1,6 @@
 from deep_reference_parser.prodigy import prodigy_to_conll
 
+
 def test_prodigy_to_conll():
 
     before = [

tests/test_io.py

@@ -5,7 +5,13 @@
 
 import pytest
 
-from deep_reference_parser.io.io import read_jsonl, write_jsonl, load_tsv, write_tsv, _split_list_by_linebreaks
+from deep_reference_parser.io.io import (
+    read_jsonl,
+    write_jsonl,
+    load_tsv,
+    write_tsv,
+    _split_list_by_linebreaks,
+)
 from deep_reference_parser.reference_utils import yield_token_label_pairs
 
 from .common import TEST_JSONL, TEST_TSV_TRAIN, TEST_TSV_PREDICT, TEST_LOAD_TSV
@@ -41,6 +47,7 @@ def test_write_tsv(tmpdir):
 
     assert expected == actual
 
+
 def test_load_tsv_train():
     """
     Text of TEST_TSV_TRAIN:
@@ -129,6 +136,7 @@ def test_load_tsv_predict():
 
     assert actual == expected
 
+
 def test_load_tsv_train_multiple_labels():
     """
     Text of TEST_TSV_TRAIN:
@@ -219,6 +227,7 @@ def test_yield_toke_label_pairs():
 
     assert expected == actual
 
+
 def test_read_jsonl():
 
     expected = [
@@ -280,20 +289,23 @@ def test_write_jsonl(tmpdir):
 
     assert expected == actual
 
+
 def test_split_list_by_linebreaks():
 
     lst = ["a", "b", "c", None, "d"]
     expected = [["a", "b", "c"], ["d"]]
 
     actual = _split_list_by_linebreaks(lst)
 
+
 def test_list_by_linebreaks_ending_in_None():
 
     lst = ["a", "b", "c", float("nan"), "d", None]
     expected = [["a", "b", "c"], ["d"]]
 
     actual = _split_list_by_linebreaks(lst)
 
+
 def test_list_by_linebreaks_starting_in_None():
 
     lst = [None, "a", "b", "c", None, "d"]

tests/test_reference_utils.py

@@ -3,9 +3,8 @@
 
 import pytest
 
-from deep_reference_parser.reference_utils import (
-    break_into_chunks
-)
+from deep_reference_parser.reference_utils import break_into_chunks
+
 
 def test_break_into_chunks():
 
@@ -15,4 +14,3 @@ def test_break_into_chunks():
     actual = break_into_chunks(before, max_words=2)
 
     assert expected == actual
-