formatting

sub-packages/bionemo-geneformer/tests/bionemo/geneformer/test_model.py

@@ -14,12 +14,12 @@
 # limitations under the License.
 
 import math
+import re
 import tarfile
 from copy import deepcopy
 from pathlib import Path
 from typing import List, Tuple
 from unittest import mock
-import re
 
 import pytest
 import torch
@@ -261,7 +261,9 @@ def __getitem__(self, idx):
         return {"text": self.input_ids[idx], "attention_mask": self.mask[idx]}
 
 
-@pytest.mark.xfail(re.search(r"h[1-9]00", torch.cuda.get_device_name().lower()) is not None, reason="Known issue on H100 GPUs")
+@pytest.mark.xfail(
+    re.search(r"h[1-9]00", torch.cuda.get_device_name().lower()) is not None, reason="Known issue on H100 GPUs"
+)
 def test_geneformer_nemo1_v_nemo2_inference_golden_values(
     geneformer_config: GeneformerConfig, cells: List[List[str]], seed: int = 42
 ):