Style: PEP8 them all

tests/test_cky.py

@@ -1,10 +1,10 @@
 import torch
 from hypothesis import given, strategies as st
 from torch_struct import TreeCRF
-from torchnyan import assert_close, assert_grad_close, BATCH_SIZE, device, sizes, TINY_TOKEN_SIZE
+from torchnyan import BATCH_SIZE, TINY_TOKEN_SIZE, assert_close, assert_grad_close, device, sizes
 from torchrua import C
 
-from torchlatent.cky import cky_partitions, cky_scores, CkyDecoder
+from torchlatent.cky import CkyDecoder, cky_partitions, cky_scores
 from torchlatent.semiring import Log
 
 

tests/test_crf.py

@@ -1,10 +1,10 @@
 import torch
 from hypothesis import given, strategies as st
 from torchcrf import CRF
-from torchnyan import assert_close, assert_grad_close, assert_sequence_close, BATCH_SIZE, device, sizes, TOKEN_SIZE
+from torchnyan import BATCH_SIZE, TOKEN_SIZE, assert_close, assert_grad_close, assert_sequence_close, device, sizes
 from torchrua import C, D, P
 
-from torchlatent.crf import crf_partitions, crf_scores, CrfDecoder
+from torchlatent.crf import CrfDecoder, crf_partitions, crf_scores
 from torchlatent.semiring import Log
 
 

tests/test_functional.py

@@ -1,7 +1,7 @@
 import torch
 from hypothesis import given, strategies as st
 from torchnyan.assertion import assert_close, assert_grad_close
-from torchnyan.strategy import device, sizes, TINY_BATCH_SIZE, TINY_TOKEN_SIZE
+from torchnyan.strategy import TINY_BATCH_SIZE, TINY_TOKEN_SIZE, device, sizes
 
 from torchlatent.functional import logaddexp, logsumexp
 

torchlatent/abc.py

@@ -3,7 +3,7 @@
 
 import torch
 import torch.autograd
-from torch import nn, Tensor
+from torch import Tensor, nn
 from torch.distributions.utils import lazy_property
 from torchrua import C, D, P
 

torchlatent/crf.py

@@ -1,7 +1,7 @@
 from typing import Tuple, Type, Union
 
 import torch
-from torch import nn, Tensor
+from torch import Tensor, nn
 from torch.distributions.utils import lazy_property
 from torch.nn import init
 from torchrua import C, D, P