Style

Signed-off-by: Boris Fomitchev <[email protected]>

Author: borisfom

cuequivariance/cuequivariance/experimental/mace/symmetric_contractions.py

@@ -18,8 +18,8 @@
 import numpy as np
 
 import cuequivariance as cue
-from cuequivariance import descriptors
 import cuequivariance.segmented_tensor_product as stp
+from cuequivariance import descriptors
 from cuequivariance.misc.linalg import round_to_sqrt_rational, triu_array
 
 

cuequivariance/cuequivariance/irreps_array/context_decorator.py

@@ -13,7 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from functools import wraps
-from typing import Optional, Union, Type
+from typing import Optional, Type, Union
 
 import cuequivariance as cue
 import cuequivariance.irreps_array as irreps_array

cuequivariance/cuequivariance/irreps_array/irreps.py

@@ -16,7 +16,7 @@
 
 import dataclasses
 import re
-from typing import NamedTuple, Union, Type, Any, Sequence, Callable, Optional
+from typing import Any, Callable, NamedTuple, Optional, Sequence, Type, Union
 
 import cuequivariance as cue
 

cuequivariance/cuequivariance/irreps_array/misc_ui.py

@@ -15,7 +15,7 @@
 from __future__ import annotations
 
 import warnings
-from typing import Generator, Optional, Union, Any
+from typing import Any, Generator, Optional, Union
 
 import cuequivariance as cue
 

cuequivariance/cuequivariance/irreps_array/reduced_tensor_product.py

@@ -16,7 +16,7 @@
 import itertools
 import logging
 from math import prod
-from typing import FrozenSet, List, Optional, Sequence, Tuple, Iterator, Union
+from typing import FrozenSet, Iterator, List, Optional, Sequence, Tuple, Union
 
 import numpy as np
 

cuequivariance/cuequivariance/misc/sympy_utils.py

@@ -13,8 +13,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from fractions import Fraction
-import numpy as np
 
+import numpy as np
 import sympy
 
 

cuequivariance/cuequivariance/representation/irrep_so3.py

@@ -22,7 +22,7 @@
 import numpy as np
 
 from cuequivariance.misc.linalg import round_to_sqrt_rational
-from cuequivariance.representation import Irrep, SU2
+from cuequivariance.representation import SU2, Irrep
 
 
 # This function is copied from https://github.com/lie-nn/lie-nn/blob/70adebce44e3197ee17f780585c6570d836fc2fe/lie_nn/_src/irreps/so3_real.py

cuequivariance/cuequivariance/representation/irrep_su2.py

@@ -13,6 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from __future__ import annotations
+
 import itertools
 import re
 from dataclasses import dataclass

cuequivariance/cuequivariance/segmented_tensor_product/operand.py

@@ -16,7 +16,7 @@
 
 import dataclasses
 import math
-from typing import Optional, Union, Sequence
+from typing import Optional, Sequence, Union
 
 from cuequivariance import segmented_tensor_product as stp
 

cuequivariance/cuequivariance/segmented_tensor_product/segmented_tensor_product.py

@@ -25,7 +25,7 @@
 import math
 import re
 import zlib
-from typing import Any, Optional, Union, Callable, Sequence
+from typing import Any, Callable, Optional, Sequence, Union
 
 import numpy as np
 import opt_einsum