def METHOD_NAME(model_cls, block_count):
    model = model_cls()
    traced = symbolic_trace(model)
    assert isinstance(traced, GraphModule)
    inp = torch.randn(1, 3, 224, 224)
    traced_st_depth_no_drop, residual_count = apply_stochastic_residual(traced, 0.0)
    out_traced = traced(inp)
    out_traced_st_depth_no_drop = traced_st_depth_no_drop(inp)
    assert torch.allclose(out_traced,
                          out_traced_st_depth_no_drop), 'mismatch in outputs with 0 drop rate for stochastic modules'
    assert residual_count == block_count