Skip to content

Commit 4f57ec2

Browse files
committed
Remove conftest diffs
Signed-off-by: John St. John <jstjohn@nvidia.com>
1 parent 5736e04 commit 4f57ec2

1 file changed

Lines changed: 1 addition & 28 deletions

File tree

  • bionemo-recipes/recipes/evo2_megatron/tests/bionemo/evo2

bionemo-recipes/recipes/evo2_megatron/tests/bionemo/evo2/conftest.py

Lines changed: 1 addition & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -14,42 +14,15 @@
1414
# limitations under the License.
1515

1616

17-
# conftest.py
18-
# ruff: noqa: E402, I001
1917
import copy
2018
import gc
21-
import importlib
2219
import os
2320
import shlex
2421
import subprocess
25-
import sys
2622
from pathlib import Path
27-
from types import ModuleType
2823

2924
import pytest
30-
31-
# Load Transformer Engine's core library before torch; this avoids a CUDA library
32-
# symbol-resolution failure seen when Megatron imports torch before TE is loaded.
33-
importlib.import_module("transformer_engine.pytorch")
34-
torch = importlib.import_module("torch")
35-
36-
# Megatron Bridge imports Transformers conversion modules during collection. In
37-
# this NGC image, torchvision is present but its custom ops are unavailable, so
38-
# make Transformers skip optional vision imports for these non-vision tests.
39-
transformers_import_utils = importlib.import_module("transformers.utils.import_utils")
40-
transformers_import_utils._torchvision_available = False
41-
42-
43-
# Megatron Bridge eagerly imports VLM recipe modules through its recipes package.
44-
# Those modules import qwen_vl_utils, which imports torchvision. Provide the
45-
# non-vision tests with a stub so Evo2 collection does not depend on VLM extras.
46-
def _unavailable_process_vision_info(*args, **kwargs):
47-
raise RuntimeError("qwen_vl_utils is unavailable in Evo2 tests")
48-
49-
50-
qwen_vl_utils = ModuleType("qwen_vl_utils")
51-
qwen_vl_utils.process_vision_info = _unavailable_process_vision_info
52-
sys.modules["qwen_vl_utils"] = qwen_vl_utils
25+
import torch
5326

5427
from bionemo.core.data.load import load as bionemo_load
5528
from bionemo.evo2.data.dataset_tokenizer import DEFAULT_HF_TOKENIZER_MODEL_PATH_512

0 commit comments

Comments
 (0)