chore: style

setup.py

@@ -1,20 +1,20 @@
-from setuptools import setup, find_packages
+from setuptools import find_packages, setup
 
-with open('README.md') as f:
+with open("README.md") as f:
     readme = f.read()
 
-with open('requirements.txt') as f:
-    requirements = f.read().split('\n')
+with open("requirements.txt") as f:
+    requirements = f.read().split("\n")
 
 setup(
-    name='stripedhyena',
-    version='0.2.1',
-    description='Model and inference code for beyond Transformer architectures',
+    name="stripedhyena",
+    version="0.2.1",
+    description="Model and inference code for beyond Transformer architectures",
     long_description=readme,
-    long_description_content_type='text/markdown',
-    author='Michael Poli',
-    url='http://github.com/togethercomputer/stripedhyena',
-    license='Apache-2.0',
+    long_description_content_type="text/markdown",
+    author="Michael Poli",
+    url="http://github.com/togethercomputer/stripedhyena",
+    license="Apache-2.0",
     packages=find_packages(where="stripedhyena"),
     install_requires=requirements,
-)
+)

test/test_backward.py

@@ -4,7 +4,6 @@
 import torch
 import torch.nn as nn
 import yaml
-
 from src.layers import RMSNorm
 from src.model import StripedHyena
 from src.utils import dotdict

test/test_forward.py

@@ -4,11 +4,10 @@
 import torch
 import torch.nn as nn
 import yaml
-from torch.autograd import grad
-
 from src.layers import RMSNorm
 from src.model import StripedHyena
 from src.utils import dotdict
+from torch.autograd import grad
 
 try:
     from flashfftconv import FlashFFTConv

test/test_numerics.py

@@ -4,7 +4,6 @@
 import torch
 import torch.nn as nn
 import yaml
-
 from src.layers import RMSNorm
 from src.utils import dotdict
 

test/test_prefill.py

@@ -4,7 +4,6 @@
 import torch
 import torch.nn as nn
 import yaml
-
 from src.layers import RMSNorm
 from src.model import StripedHyena
 from src.utils import dotdict