diff --git a/kg_microbe/transform_utils/traits/traits.py b/kg_microbe/transform_utils/traits/traits.py index 53d2e6ba..0dcadb85 100644 --- a/kg_microbe/transform_utils/traits/traits.py +++ b/kg_microbe/transform_utils/traits/traits.py @@ -154,7 +154,12 @@ def run(self, data_file: Union[Optional[Path], Optional[str]] = None, show_statu if not (self.nlp_output_dir / go_result_fn).is_file(): annotate( - go_nlp_df, GO_PREFIX, exclusion_list, self.nlp_output_dir / go_result_fn, False, None + go_nlp_df, + GO_PREFIX, + exclusion_list, + self.nlp_output_dir / go_result_fn, + False, + None, ) go_result = pd.read_csv( str(self.nlp_output_dir / go_result_fn), sep="\t", low_memory=False