def test_custom_mixed_precision():
    """Test to ensure that passing a custom mixed precision config works."""
    config = MixedPrecision()
    strategy = FSDPStrategy(mixed_precision=config)
    assert strategy.mixed_precision_config == config
Test to ensure that passing a custom mixed precision config works.
tests/tests_pytorch/strategies/test_fsdp.py