chg: Update __init__.py files and imports

Author: ivyleavedtoadflax

deep_reference_parser/__init__.py

@@ -2,9 +2,9 @@
 # distracting on the command line. These lines here (while undesireable)
 # reduce the level of verbosity.
 
+import os
 import sys
 import warnings
-import os
 
 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"
 
@@ -19,20 +19,12 @@
 
 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 .logger import logger
 from .model_utils import get_config
-from .reference_utils import (
-    break_into_chunks,
-    labels_to_prodigy,
-    load_tsv,
-    prodigy_to_conll,
-    prodigy_to_lists,
-    read_jsonl,
-    read_pickle,
-    write_json,
-    write_jsonl,
-    write_pickle,
-    write_to_csv,
-    write_txt,
-)
+from .reference_utils import break_into_chunks
 from .tokens_to_references import tokens_to_references
+
+
+

deep_reference_parser/deep_reference_parser.py

@@ -47,7 +47,7 @@
     save_confusion_matrix,
     word2vec_embeddings,
 )
-from .reference_utils import load_tsv, read_pickle, write_pickle, write_to_csv
+from .io import load_tsv, read_pickle, write_pickle, write_to_csv
 
 
 class DeepReferenceParser:

deep_reference_parser/prodigy/__init__.py

@@ -6,3 +6,5 @@
 from .reach_to_prodigy import ReachToProdigy, reach_to_prodigy
 from .reference_to_token_annotations import TokenTagger, reference_to_token_annotations
 from .spacy_doc_to_prodigy import SpacyDocToProdigy
+from .misc import prodigy_to_conll
+from .labels_to_prodigy import labels_to_prodigy