RelCAT: flake8 fixes.

Author: vladd-bit

medcat/utils/relation_extraction/ml_utils.py

@@ -23,7 +23,7 @@ def split_list_train_test_by_class(data: List, sample_limit: int = -1, test_size
     Args:
         data (List): "output_relations": relation_instances, <-- see create_base_relations_from_doc/csv
                     for data columns
-        sample_limit (int, optional): limit the number of samples per class, useful for dataset balancing . Defaults to -1.
+        sample_limit (int): limit the number of samples per class, useful for dataset balancing . Defaults to -1.
         test_size (float): Defaults to 0.2.
         shuffle (bool): shuffle data randomly. Defaults to True.
 

medcat/utils/relation_extraction/rel_dataset.py

@@ -194,10 +194,10 @@ def _create_relation_validation(self,
             ent2_end_char_pos (int): ent2 end char pos
             ent1_token_start_pos (int): ent1_token_start_pos. Defaults to -1.
             ent2_token_start_pos (int): ent2_token_start_pos. Defaults to -1.
-            ent1_token_end_pos (int, optional): ent1_token_end_pos. Defaults to -1.
-            ent2_token_end_pos (int, optional): ent2_token_end_pos. Defaults to -1.
-            is_spacy_doc (bool, optional): checks if doc is spacy docs. Defaults to False.
-            is_mct_export (bool, optional): chekcs if doc is a mct export. Defaults to False.
+            ent1_token_end_pos (int): ent1_token_end_pos. Defaults to -1.
+            ent2_token_end_pos (int): ent2_token_end_pos. Defaults to -1.
+            is_spacy_doc (bool): checks if doc is spacy docs. Defaults to False.
+            is_mct_export (bool): chekcs if doc is a mct export. Defaults to False.
 
         Returns:
             List: row containing rel data ["relation_token_span_ids", "ent1_ent2_start", "ent1", "ent2", "label",