remove duplicate import in test/pytorch/selective_layernorm_mlp/test_recipe.py

jaimec00 · jaimec00 · commit 4aadd049e4a0 · 2025-10-28T03:33:40.000Z
Signed-off-by: Jaime Cardenas &lt;jaime@evolutionaryscale.ai&gt;
diff --git a/tests/pytorch/selective_layernorm_mlp/test_recipe.py b/tests/pytorch/selective_layernorm_mlp/test_recipe.py
@@ -26,7 +26,6 @@
 )
 import transformer_engine.pytorch.ops as te_ops
 from transformer_engine.common.recipe import DelayedScaling, Float8BlockScaling, MXFP8BlockScaling
-import transformer_engine_torch as tex
 
 # Check if FP8 is supported
 fp8_available, reason_for_no_fp8 = te.is_fp8_available(return_reason=True)

Original file line number	Diff line number	Diff line change
`@@ -26,7 +26,6 @@`
`26`	`26`	`)`
`27`	`27`	`import transformer_engine.pytorch.ops as te_ops`
`28`	`28`	`from transformer_engine.common.recipe import DelayedScaling, Float8BlockScaling, MXFP8BlockScaling`
`29`		`-import transformer_engine_torch as tex`
`30`	`29`
`31`	`30`	`# Check if FP8 is supported`
`32`	`31`	`fp8_available, reason_for_no_fp8 = te.is_fp8_available(return_reason=True)`