fix lint

sail-sg · Jan 12, 2024 · 37be812 · 37be812
1 parent c1cac2c
commit 37be812
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/autofd/__init__.py b/autofd/__init__.py
@@ -16,6 +16,13 @@
 import jax
 import optax
 
+from . import operators  # noqa
+from .general_array import SpecTree  # noqa
+from .general_array import (
+  Arg, Grid, Ret, Spec, dummy_array, dummy_input, dummy_output, function,
+  is_function, num_args, random_input, with_spec, zeros_like
+)
+
 
 def scale_by_learning_rate(
   learning_rate,
@@ -33,13 +40,6 @@ def scale_by_learning_rate(
 
 optax._src.alias._scale_by_learning_rate = scale_by_learning_rate
 
-from . import operators  # noqa
-from .general_array import SpecTree  # noqa
-from .general_array import (
-  Arg, Grid, Ret, Spec, dummy_array, dummy_input, dummy_output, function,
-  is_function, num_args, random_input, with_spec, zeros_like
-)
-
 __all__ = [
   "Spec",
   "SpecTree",