Fix CI lint and precomputed predict path

Author: sergeyf

s2apler/arrow_io.py

@@ -398,7 +398,8 @@ def _prepare_output_dir(output_dir: Path, *, overwrite: bool) -> None:
     existing_bundle_files = [file_name for file_name in bundle_file_names if (output_dir / file_name).exists()]
     if existing_bundle_files and not overwrite:
         raise FileExistsError(
-            f"Arrow bundle files already exist at {output_dir}: {existing_bundle_files}; pass overwrite=True to replace them"
+            "Arrow bundle files already exist at "
+            f"{output_dir}: {existing_bundle_files}; pass overwrite=True to replace them"
         )
     if overwrite:
         for file_name in bundle_file_names:

s2apler/data.py

@@ -827,9 +827,9 @@ def fixed_pairs(
         -------
         train/val/test pairs, where each pair is (paper_id_1, paper_id_2, label)
         """
-        assert self.train_pairs is not None and self.test_pairs is not None, (
-            "You need to pass in train and test pairs to use this function"
-        )
+        assert (
+            self.train_pairs is not None and self.test_pairs is not None
+        ), "You need to pass in train and test pairs to use this function"
         self.train_pairs.loc[:, "label"] = self.train_pairs["label"].map(
             {"NO": 0, "YES": 1, "0": 0, 0: 0, "1": 1, 1: 1}
         )