fixing lint errors (#760)

Summary:
Pull Request resolved: #760

Fixing the lint errors of D74334633

Reviewed By: HuanyuZhang

Differential Revision: D74656618

fbshipit-source-id: 85363cc06a05c3ac96c4afb7f63d99dd347c8a2b

Author: aparna-aketi

opacus/grad_sample/__init__.py

@@ -27,9 +27,9 @@
 from .gsm_exp_weights import GradSampleModuleExpandedWeights
 from .gsm_no_op import GradSampleModuleNoOp
 from .instance_norm import compute_instance_norm_grad_sample  # noqa
-from .rms_norm import compute_rms_norm_grad_sample  # noqa
 from .layer_norm import compute_layer_norm_grad_sample  # noqa
 from .linear import compute_linear_grad_sample  # noqa
+from .rms_norm import compute_rms_norm_grad_sample  # noqa
 from .utils import (
     get_gsm_class,
     register_grad_sampler,