From 1b0bd34b05bcb4f25c12203e215e94b5fba91c95 Mon Sep 17 00:00:00 2001 From: Gagan Kaushik Date: Thu, 19 Dec 2024 21:04:10 +0000 Subject: [PATCH] formatting --- .../tests/bionemo/geneformer/test_model.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/sub-packages/bionemo-geneformer/tests/bionemo/geneformer/test_model.py b/sub-packages/bionemo-geneformer/tests/bionemo/geneformer/test_model.py index 505a12bd51..d679eeb023 100644 --- a/sub-packages/bionemo-geneformer/tests/bionemo/geneformer/test_model.py +++ b/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 ):