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