metaopt · XuehaiPan · Feb 2, 2023 · Dec 8, 2022 · Dec 8, 2022 · Dec 26, 2022
diff --git a/torchopt/diff/implicit/nn/module.py b/torchopt/diff/implicit/nn/module.py
@@ -110,7 +110,7 @@ def enable_implicit_gradients(
         raise TypeError('Implicit gradients are already enabled for the `solve` method.')
 
     if cls.linear_solve is not None:
-        solve_kwargs = dict(solve=cls.linear_solve)
+        solve_kwargs = {'solve': cls.linear_solve}
     else:
         solve_kwargs = {}
 

diff --git a/torchopt/distributed/api.py b/torchopt/distributed/api.py
@@ -1,4 +1,4 @@
-# Copyright 2022 MetaOPT Team. All Rights Reserved.
+# Copyright 2022-2023 MetaOPT Team. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -202,7 +202,7 @@ def __reduce__(
         return (
             TensorDimensionPartitioner,
             (self.dim,),
-            dict(exclusive=self.exclusive, keepdim=self.keepdim, workers=self.workers),
+            {'exclusive': self.exclusive, 'keepdim': self.keepdim, 'workers': self.workers},
         )
 
 

diff --git a/torchopt/visual.py b/torchopt/visual.py
@@ -1,4 +1,4 @@
-# Copyright 2022 MetaOPT Team. All Rights Reserved.
+# Copyright 2022-2023 MetaOPT Team. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -138,16 +138,16 @@ def make_dot(
                 else:
                     param_map.update({v: k for k, v in cast(Mapping, param).items()})
 
-    node_attr = dict(
-        style='filled',
-        shape='box',
-        align='left',
-        fontsize='10',
-        ranksep='0.1',
-        height='0.2',
-        fontname='monospace',
-    )
-    dot = Digraph(node_attr=node_attr, graph_attr=dict(size='12,12'))
+    node_attr = {
+        'style': 'filled',
+        'shape': 'box',
+        'align': 'left',
+        'fontsize': '10',
+        'ranksep': '0.1',
+        'height': '0.2',
+        'fontname': 'monospace',
+    }
+    dot = Digraph(node_attr=node_attr, graph_attr={'size': '12,12'})
     seen = set()
 
     def size_to_str(size):