Fix E2E nightly tests: move missing conftest.py (#15501)

pzelasko · web-flow · commit be64db1e1ec4 · 2026-03-16T16:07:44.000-04:00
Signed-off-by: Piotr Żelasko &lt;petezor@gmail.com&gt;
diff --git a/tests/e2e_nightly/conftest.py b/tests/e2e_nightly/conftest.py
@@ -50,14 +50,3 @@ def prepare_for_training_step(model):
     # to log the learning rate. Provide a minimal stand-in if no optimizer is set.
     if getattr(model, '_optimizer', None) is None:
         model._optimizer = torch.optim.SGD([torch.nn.Parameter(torch.zeros(1))], lr=1e-4)
-
-
-def prepare_for_transcribe(model):
-    """Placeholder for any pre-transcribe model preparation.
-
-    Previously disabled CUDA graph decoding, but the root cause
-    (CUDA 12/13 API mismatch in cudaStreamGetCaptureInfo) is now
-    fixed in the NeMo source. Kept as a no-op for compatibility
-    with test files that call it.
-    """
-    pass
diff --git a/tests/e2e_nightly/test_model_support_nvidia__canary_180m_flash.py b/tests/e2e_nightly/test_model_support_nvidia__canary_180m_flash.py
@@ -118,10 +118,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(
diff --git a/tests/e2e_nightly/test_model_support_nvidia__canary_1b.py b/tests/e2e_nightly/test_model_support_nvidia__canary_1b.py
@@ -128,10 +128,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(
diff --git a/tests/e2e_nightly/test_model_support_nvidia__canary_1b_flash.py b/tests/e2e_nightly/test_model_support_nvidia__canary_1b_flash.py
@@ -118,10 +118,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(
diff --git a/tests/e2e_nightly/test_model_support_nvidia__canary_1b_v2.py b/tests/e2e_nightly/test_model_support_nvidia__canary_1b_v2.py
@@ -145,10 +145,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(
diff --git a/tests/e2e_nightly/test_model_support_nvidia__nemotron_speech_streaming_en_0_6b.py b/tests/e2e_nightly/test_model_support_nvidia__nemotron_speech_streaming_en_0_6b.py
@@ -76,10 +76,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__parakeet_ctc_0_6b.py b/tests/e2e_nightly/test_model_support_nvidia__parakeet_ctc_0_6b.py
@@ -74,10 +74,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__parakeet_ctc_0_6b_Vietnamese.py b/tests/e2e_nightly/test_model_support_nvidia__parakeet_ctc_0_6b_Vietnamese.py
@@ -74,10 +74,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__parakeet_ctc_1_1b.py b/tests/e2e_nightly/test_model_support_nvidia__parakeet_ctc_1_1b.py
@@ -74,10 +74,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__parakeet_realtime_eou_120m_v1.py b/tests/e2e_nightly/test_model_support_nvidia__parakeet_realtime_eou_120m_v1.py
@@ -75,10 +75,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__parakeet_rnnt_0_6b.py b/tests/e2e_nightly/test_model_support_nvidia__parakeet_rnnt_0_6b.py
@@ -76,10 +76,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__parakeet_rnnt_110m_da_dk.py b/tests/e2e_nightly/test_model_support_nvidia__parakeet_rnnt_110m_da_dk.py
@@ -76,10 +76,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__parakeet_rnnt_1_1b.py b/tests/e2e_nightly/test_model_support_nvidia__parakeet_rnnt_1_1b.py
@@ -76,10 +76,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__parakeet_tdt_0_6b_v2.py b/tests/e2e_nightly/test_model_support_nvidia__parakeet_tdt_0_6b_v2.py
@@ -76,10 +76,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__parakeet_tdt_0_6b_v3.py b/tests/e2e_nightly/test_model_support_nvidia__parakeet_tdt_0_6b_v3.py
@@ -76,10 +76,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__parakeet_tdt_1_1b.py b/tests/e2e_nightly/test_model_support_nvidia__parakeet_tdt_1_1b.py
@@ -76,10 +76,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__parakeet_tdt_ctc_0_6b_ja.py b/tests/e2e_nightly/test_model_support_nvidia__parakeet_tdt_ctc_0_6b_ja.py
@@ -76,10 +76,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__parakeet_tdt_ctc_110m.py b/tests/e2e_nightly/test_model_support_nvidia__parakeet_tdt_ctc_110m.py
@@ -76,10 +76,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__parakeet_tdt_ctc_1_1b.py b/tests/e2e_nightly/test_model_support_nvidia__parakeet_tdt_ctc_1_1b.py
@@ -76,10 +76,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_ar_fastconformer_hybrid_large_pc_v1_0.py b/tests/e2e_nightly/test_model_support_nvidia__stt_ar_fastconformer_hybrid_large_pc_v1_0.py
@@ -76,10 +76,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_ar_fastconformer_hybrid_large_pcd_v1_0.py b/tests/e2e_nightly/test_model_support_nvidia__stt_ar_fastconformer_hybrid_large_pcd_v1_0.py
@@ -76,10 +76,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_be_fastconformer_hybrid_large_pc.py b/tests/e2e_nightly/test_model_support_nvidia__stt_be_fastconformer_hybrid_large_pc.py
@@ -76,10 +76,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_de_fastconformer_hybrid_large_pc.py b/tests/e2e_nightly/test_model_support_nvidia__stt_de_fastconformer_hybrid_large_pc.py
@@ -76,10 +76,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_en_fastconformer_ctc_large.py b/tests/e2e_nightly/test_model_support_nvidia__stt_en_fastconformer_ctc_large.py
@@ -74,10 +74,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_en_fastconformer_ctc_xlarge.py b/tests/e2e_nightly/test_model_support_nvidia__stt_en_fastconformer_ctc_xlarge.py
@@ -74,10 +74,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_en_fastconformer_ctc_xxlarge.py b/tests/e2e_nightly/test_model_support_nvidia__stt_en_fastconformer_ctc_xxlarge.py
@@ -74,10 +74,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_en_fastconformer_hybrid_large_pc.py b/tests/e2e_nightly/test_model_support_nvidia__stt_en_fastconformer_hybrid_large_pc.py
@@ -76,10 +76,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_en_fastconformer_hybrid_large_streaming_multi.py b/tests/e2e_nightly/test_model_support_nvidia__stt_en_fastconformer_hybrid_large_streaming_multi.py
@@ -76,10 +76,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_en_fastconformer_hybrid_medium_streaming_80ms.py b/tests/e2e_nightly/test_model_support_nvidia__stt_en_fastconformer_hybrid_medium_streaming_80ms.py
@@ -76,10 +76,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_en_fastconformer_hybrid_medium_streaming_80ms_pc.py b/tests/e2e_nightly/test_model_support_nvidia__stt_en_fastconformer_hybrid_medium_streaming_80ms_pc.py
@@ -76,10 +76,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_en_fastconformer_tdt_large.py b/tests/e2e_nightly/test_model_support_nvidia__stt_en_fastconformer_tdt_large.py
@@ -76,10 +76,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_en_fastconformer_transducer_large.py b/tests/e2e_nightly/test_model_support_nvidia__stt_en_fastconformer_transducer_large.py
@@ -76,10 +76,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_en_fastconformer_transducer_xlarge.py b/tests/e2e_nightly/test_model_support_nvidia__stt_en_fastconformer_transducer_xlarge.py
@@ -76,10 +76,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_en_fastconformer_transducer_xxlarge.py b/tests/e2e_nightly/test_model_support_nvidia__stt_en_fastconformer_transducer_xxlarge.py
@@ -76,10 +76,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_es_fastconformer_hybrid_large_pc.py b/tests/e2e_nightly/test_model_support_nvidia__stt_es_fastconformer_hybrid_large_pc.py
@@ -76,10 +76,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_es_fastconformer_hybrid_large_pc_nc.py b/tests/e2e_nightly/test_model_support_nvidia__stt_es_fastconformer_hybrid_large_pc_nc.py
@@ -76,10 +76,6 @@ def test_model_inference():
     model = _load_model()
     model.eval()
 
-    from conftest import prepare_for_transcribe
-
-    prepare_for_transcribe(model)
-
     audio = np.random.randn(16000).astype(np.float32)
 
     result = model.transcribe(audio=[audio], batch_size=1)
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_fa_fastconformer_hybrid_large.py b/tests/e2e_nightly/test_model_support_nvidia__stt_fa_fastconformer_hybrid_large.py
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_fr_fastconformer_hybrid_large_pc.py b/tests/e2e_nightly/test_model_support_nvidia__stt_fr_fastconformer_hybrid_large_pc.py
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_hr_fastconformer_hybrid_large_pc.py b/tests/e2e_nightly/test_model_support_nvidia__stt_hr_fastconformer_hybrid_large_pc.py
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_hy_fastconformer_hybrid_large_pc.py b/tests/e2e_nightly/test_model_support_nvidia__stt_hy_fastconformer_hybrid_large_pc.py
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_it_fastconformer_hybrid_large_pc.py b/tests/e2e_nightly/test_model_support_nvidia__stt_it_fastconformer_hybrid_large_pc.py
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_ka_fastconformer_hybrid_large_pc.py b/tests/e2e_nightly/test_model_support_nvidia__stt_ka_fastconformer_hybrid_large_pc.py
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_ka_fastconformer_hybrid_transducer_ctc_large_streaming_80ms_pc.py b/tests/e2e_nightly/test_model_support_nvidia__stt_ka_fastconformer_hybrid_transducer_ctc_large_streaming_80ms_pc.py
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_kk_ru_fastconformer_hybrid_large.py b/tests/e2e_nightly/test_model_support_nvidia__stt_kk_ru_fastconformer_hybrid_large.py
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_nl_fastconformer_hybrid_large_pc.py b/tests/e2e_nightly/test_model_support_nvidia__stt_nl_fastconformer_hybrid_large_pc.py
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_pl_fastconformer_hybrid_large_pc.py b/tests/e2e_nightly/test_model_support_nvidia__stt_pl_fastconformer_hybrid_large_pc.py
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_pt_fastconformer_hybrid_large_pc.py b/tests/e2e_nightly/test_model_support_nvidia__stt_pt_fastconformer_hybrid_large_pc.py
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_ru_fastconformer_hybrid_large_pc.py b/tests/e2e_nightly/test_model_support_nvidia__stt_ru_fastconformer_hybrid_large_pc.py
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_ua_fastconformer_hybrid_large_pc.py b/tests/e2e_nightly/test_model_support_nvidia__stt_ua_fastconformer_hybrid_large_pc.py
diff --git a/tests/e2e_nightly/test_model_support_nvidia__stt_uz_fastconformer_hybrid_large_pc.py b/tests/e2e_nightly/test_model_support_nvidia__stt_uz_fastconformer_hybrid_large_pc.py
diff --git a/tests/e2e_nightly/test_model_support_stt_en_fastconformer_hybrid_large_streaming_1040ms.py b/tests/e2e_nightly/test_model_support_stt_en_fastconformer_hybrid_large_streaming_1040ms.py
diff --git a/tests/e2e_nightly/test_model_support_stt_multilingual_fastconformer_hybrid_large_pc_blend_eu.py b/tests/e2e_nightly/test_model_support_stt_multilingual_fastconformer_hybrid_large_pc_blend_eu.py
