code: def test_wrap_override_defaults(self):
      with enable_wrap(
          wrapper_cls=FSDP,
          flatten_parameters=False,
          process_group=self.process_group,
      ):
          layer = wrap(nn.Linear(5, 5), flatten_parameters=True)
      assert isinstance(layer, FSDP)
      assert layer.flatten_parameters
  
imports: import functools
  import os
  import random
  import unittest
  import torch
  import torch.nn
  import torch.nn.functional
  from fairscale.fair_dev.testing.testing import DummyProcessGroup
  from fairscale.nn import FullyShardedDataParallel
  from fairscale.nn import auto_wrap, default_auto_wrap_policy, enable_wrap, wrap
label_desc: Use image node_id 4 for calling the TestAutoWrap obj's underlying member method code with example usage: obj.test_wrap_override_defaults() without return types