pre-commit

Author: willow-ahrens

src/finchlite/compile/stages.py

@@ -1,11 +1,13 @@
 from abc import abstractmethod
+
 from .. import finch_assembly as asm
-from ..symbolic import Stage
 from .. import finch_notation as ntn
+from ..symbolic import Stage
+
 
 class NotationLowerer(Stage):
     @abstractmethod
     def __call__(self, term: ntn.Module) -> asm.Module:
         """
         Compile the given notation term into an assembly term.
-        """
+        """

src/finchlite/finch_notation/__init__.py

@@ -47,8 +47,8 @@
     "Module",
     "NotationExpression",
     "NotationInterpreter",
-    "NotationNode",
     "NotationLoader",
+    "NotationNode",
     "NotationTransform",
     "Read",
     "Repack",

src/finchlite/finch_notation/interpreter.py

@@ -5,6 +5,7 @@
 
 import numpy as np
 
+from .. import finch_assembly as asm
 from ..algebra import (
     Tensor,
     TensorFType,
@@ -14,10 +15,9 @@
     register_property,
     shape_type,
 )
-from .. import finch_assembly as asm
-from .stages import NotationLoader
 from ..symbolic import ScopedDict, fisinstance, ftype
 from . import nodes as ntn
+from .stages import NotationLoader
 
 
 class TensorViewFType(TensorFType):

src/finchlite/finch_notation/stages.py

@@ -1,18 +1,21 @@
 from abc import abstractmethod
+
 from .. import finch_assembly as asm
 from ..symbolic import Stage
 from . import nodes as ntn
 
+
 class NotationLoader(Stage):
     @abstractmethod
     def __call__(self, term: ntn.Module) -> asm.AssemblyLibrary:
         """
         Load the given notation program into a runnable module.
         """
 
+
 class NotationTransform(Stage):
     @abstractmethod
     def __call__(self, term: ntn.Module) -> ntn.Module:
         """
         Transform the given assembly term into another assembly term.
-        """
+        """