style: format code with yapf

metaopt · XuehaiPan · Jul 10, 2022 · Jul 7, 2022 · Jul 7, 2022 · Jul 7, 2022
commit c5d6b802ce16425436741ba5c2d1a44c5e651826
diff --git a/examples/MAML-RL/helpers/__init__.py b/examples/MAML-RL/helpers/__init__.py
@@ -18,6 +18,7 @@
 
 from gym.envs.registration import register
 
+
 register(
     'TabularMDP-v0',
     entry_point='helpers.Tabular_mdp:TabularMDPEnv',

diff --git a/examples/MAML-RL/run_MAML.py b/examples/MAML-RL/run_MAML.py
@@ -25,6 +25,7 @@
 
 from .helpers.policy import CategoricalMLPPolicy
 
+
 TASK_NUM = 40
 TRAJ_NUM = 20
 TRAJ_LEN = 10

diff --git a/examples/few-shot/maml-omniglot.py b/examples/few-shot/maml-omniglot.py
@@ -55,6 +55,7 @@
 
 from .support.omniglot_loaders import OmniglotNShot
 
+
 mpl.use('Agg')
 plt.style.use('bmh')
 

diff --git a/setup.cfg b/setup.cfg
@@ -2,9 +2,10 @@
 based_on_style = yapf
 indent_width = 4
 continuation_indent_width = 4
+column_limit = 100
 spaces_before_comment = 2
 dedent_closing_brackets = true
-column_limit = 100
+blank_lines_between_top_level_imports_and_variables = 2
 
 [flake8]
 exclude =
@@ -16,9 +17,10 @@ convention = google
 
 [isort]
 profile = black
-multi_line_output = 3
 indent = 4
 line_length = 100
+lines_after_imports = 2
+multi_line_output = 3
 
 [mypy]
 allow_redefinition = True

diff --git a/torchopt/__init__.py b/torchopt/__init__.py
@@ -22,6 +22,7 @@
 from torchopt._src.utils import extract_state_dict, recover_state_dict, stop_gradient
 from torchopt.version import __version__
 
+
 __all__ = [
     "accelerated_op_available",
     "clip",

diff --git a/torchopt/_src/base.py b/torchopt/_src/base.py
@@ -36,6 +36,7 @@
 
 from torchopt._src import typing
 
+
 OptState = typing.TensorTree  # States are arbitrary nests of `torch.Tensor`.
 # Parameters are arbitrary nests of `torch.Tensor`.
 Params = typing.TensorTree

diff --git a/torchopt/_src/clip.py b/torchopt/_src/clip.py
@@ -22,6 +22,7 @@
 
 from torchopt._src import base
 
+
 ClipState = base.EmptyState
 
 

diff --git a/torchopt/_src/transform.py b/torchopt/_src/transform.py
@@ -38,6 +38,7 @@
 from torchopt._src import base
 from torchopt._src.typing import ScalarOrSchedule, Schedule
 
+
 ScaleState = base.EmptyState
 
 

diff --git a/torchopt/_src/typing.py b/torchopt/_src/typing.py
@@ -17,6 +17,7 @@
 
 from torch import Tensor
 
+
 Scalar = Union[float, int]
 Numeric = Union[Tensor, Scalar]
 

diff --git a/torchopt/_src/visual.py b/torchopt/_src/visual.py
@@ -24,6 +24,7 @@
 from graphviz import Digraph
 from pkg_resources import parse_version
 
+
 Node = namedtuple('Node', ('name', 'inputs', 'attr', 'op'))
 
 # Saved attrs for grad_fn (incl. saved variables) begin with `._saved_*`
Original file line number	Diff line number	Diff line change
Expand Up		@@ -55,6 +55,7 @@

		from .support.omniglot_loaders import OmniglotNShot


		mpl.use('Agg')
		plt.style.use('bmh')

Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -22,6 +22,7 @@

		from torchopt._src import base


		ClipState = base.EmptyState


Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -38,6 +38,7 @@
		from torchopt._src import base
		from torchopt._src.typing import ScalarOrSchedule, Schedule


		ScaleState = base.EmptyState


Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -17,6 +17,7 @@

		from torch import Tensor


		Scalar = Union[float, int]
		Numeric = Union[Tensor, Scalar]

Expand Down