Fix style.

Author: tompollard

wfdb/io/_signal.py

@@ -1712,8 +1712,9 @@ def _rd_dat_file(
     if wfdb_archive is not None:
         # If the exact file name isn't found, look for any .dat file
         if not wfdb_archive.exists(file_name):
-            dat_files = [f for f in wfdb_archive.zipfile.namelist()
-                        if f.endswith('.dat')]
+            dat_files = [
+                f for f in wfdb_archive.zipfile.namelist() if f.endswith(".dat")
+            ]
             if not dat_files:
                 raise FileNotFoundError(
                     f"No dat file found in archive for {file_name}"

wfdb/io/archive.py

@@ -32,7 +32,7 @@ def __init__(self, record_name, mode="r"):
         """
         self.record_name = record_name
         # Only append .wfdb if it's not already there
-        if record_name.endswith('.wfdb'):
+        if record_name.endswith(".wfdb"):
             self.archive_path = record_name
         else:
             self.archive_path = f"{record_name}.wfdb"
@@ -45,24 +45,19 @@ def __init__(self, record_name, mode="r"):
                     f"Archive not found: {self.archive_path}"
                 )
             if not zipfile.is_zipfile(self.archive_path):
-                raise ValueError(
-                    f"Invalid WFDB archive: {self.archive_path}"
-                )
-            self.zipfile = zipfile.ZipFile(
-                self.archive_path, mode="r"
-            )
+                raise ValueError(f"Invalid WFDB archive: {self.archive_path}")
+            self.zipfile = zipfile.ZipFile(self.archive_path, mode="r")
 
         elif mode == "w":
             # Create archive file if needed
             if not os.path.exists(self.archive_path):
                 # Create the directory if it doesn't exist
-                os.makedirs(os.path.dirname(os.path.abspath(self.archive_path)), exist_ok=True)
-                WFDBArchive.make_archive_file(
-                    [], self.archive_path
+                os.makedirs(
+                    os.path.dirname(os.path.abspath(self.archive_path)),
+                    exist_ok=True,
                 )
-            self.zipfile = zipfile.ZipFile(
-                self.archive_path, mode="a"
-            )
+                WFDBArchive.make_archive_file([], self.archive_path)
+            self.zipfile = zipfile.ZipFile(self.archive_path, mode="a")
 
     def __enter__(self):
         return self

wfdb/io/record.py

@@ -2054,9 +2054,13 @@ def rdrecord(
         wfdb_archive = get_archive(record_base)
 
         # Find any .hea file in the archive
-        hea_files = [f for f in wfdb_archive.zipfile.namelist() if f.endswith('.hea')]
+        hea_files = [
+            f for f in wfdb_archive.zipfile.namelist() if f.endswith(".hea")
+        ]
         if not hea_files:
-            raise FileNotFoundError(f"No header file found in archive {record_name}")
+            raise FileNotFoundError(
+                f"No header file found in archive {record_name}"
+            )
         hea_file = hea_files[0]  # Use the first header file found
 
         import tempfile