fixed unit test import

sugunav14 · sugunav14 · commit da137ab5427f · 2025-10-14T17:12:09.000Z
Signed-off-by: Suguna Velury &lt;178320438+sugunav14@users.noreply.github.com&gt;
diff --git a/tests/gpu/torch/export/test_fsdp2_export.py b/tests/gpu/torch/export/test_fsdp2_export.py
@@ -27,8 +27,7 @@
     _export_quantized_weight,
     requantize_resmooth_fused_llm_layers,
 )
-from modelopt.torch.quantization.qtensor.base_qtensor import fsdp2_aware_weight_update
-from modelopt.torch.quantization.utils import patch_fsdp_mp_dtypes
+from modelopt.torch.quantization.utils import fsdp2_aware_weight_update, patch_fsdp_mp_dtypes
 
 orig_init_mp_dtypes = patch_fsdp_mp_dtypes()
 

Original file line number	Diff line number	Diff line change
`@@ -27,8 +27,7 @@`
`27`	`27`	`_export_quantized_weight,`
`28`	`28`	`requantize_resmooth_fused_llm_layers,`
`29`	`29`	`)`
`30`		`-from modelopt.torch.quantization.qtensor.base_qtensor import fsdp2_aware_weight_update`
`31`		`-from modelopt.torch.quantization.utils import patch_fsdp_mp_dtypes`
	`30`	`+from modelopt.torch.quantization.utils import fsdp2_aware_weight_update, patch_fsdp_mp_dtypes`
`32`	`31`
`33`	`32`	`orig_init_mp_dtypes = patch_fsdp_mp_dtypes()`
`34`	`33`