addressed linting errors with precommit

Author: Advaith Anand

src/kirin/dialects/math/__init__.py

@@ -1,120 +1,160 @@
-"math dialect, modeling functions in python's `math` stdlib"# This file is generated by gen.py
+"math dialect, modeling functions in python's `math` stdlib"  # This file is generated by gen.py
+
+import math as pymath
+
+from kirin import lowering
 from kirin.dialects.math.dialect import dialect as dialect
+
 from . import stmts as stmts, interp as interp
-import math as pymath
+
 pi = pymath.pi
 e = pymath.e
 tau = pymath.tau
-from kirin import lowering
+
 
 @lowering.wraps(stmts.acos)
 def acos(x: float) -> float: ...
 
+
 @lowering.wraps(stmts.asin)
 def asin(x: float) -> float: ...
 
+
 @lowering.wraps(stmts.asinh)
 def asinh(x: float) -> float: ...
 
+
 @lowering.wraps(stmts.atan)
 def atan(x: float) -> float: ...
 
+
 @lowering.wraps(stmts.atan2)
 def atan2(y: float, x: float) -> float: ...
 
+
 @lowering.wraps(stmts.atanh)
 def atanh(x: float) -> float: ...
 
+
 @lowering.wraps(stmts.ceil)
 def ceil(x: float) -> int: ...
 
+
 @lowering.wraps(stmts.copysign)
 def copysign(x: float, y: float) -> float: ...
 
+
 @lowering.wraps(stmts.cos)
 def cos(x: float) -> float: ...
 
+
 @lowering.wraps(stmts.cosh)
 def cosh(x: float) -> float: ...
 
+
 @lowering.wraps(stmts.degrees)
 def degrees(x: float) -> float: ...
 
+
 @lowering.wraps(stmts.erf)
 def erf(x: float) -> float: ...
 
+
 @lowering.wraps(stmts.erfc)
 def erfc(x: float) -> float: ...
 
+
 @lowering.wraps(stmts.exp)
 def exp(x: float) -> float: ...
 
+
 @lowering.wraps(stmts.expm1)
 def expm1(x: float) -> float: ...
 
+
 @lowering.wraps(stmts.fabs)
 def fabs(x: float) -> float: ...
 
+
 @lowering.wraps(stmts.floor)
 def floor(x: float) -> int: ...
 
+
 @lowering.wraps(stmts.fmod)
 def fmod(x: float, y: float) -> float: ...
 
+
 @lowering.wraps(stmts.gamma)
 def gamma(x: float) -> float: ...
 
+
 @lowering.wraps(stmts.isfinite)
 def isfinite(x: float) -> bool: ...
 
+
 @lowering.wraps(stmts.isinf)
 def isinf(x: float) -> bool: ...
 
+
 @lowering.wraps(stmts.isnan)
 def isnan(x: float) -> bool: ...
 
+
 @lowering.wraps(stmts.lgamma)
 def lgamma(x: float) -> float: ...
 
+
 @lowering.wraps(stmts.log)
 def log(x: float, base: float) -> float: ...
 
+
 @lowering.wraps(stmts.log10)
 def log10(x: float) -> float: ...
 
+
 @lowering.wraps(stmts.log1p)
 def log1p(x: float) -> float: ...
 
+
 @lowering.wraps(stmts.log2)
 def log2(x: float) -> float: ...
 
+
 @lowering.wraps(stmts.pow)
 def pow(x: float, y: float) -> float: ...
 
+
 @lowering.wraps(stmts.radians)
 def radians(x: float) -> float: ...
 
+
 @lowering.wraps(stmts.remainder)
 def remainder(x: float, y: float) -> float: ...
 
+
 @lowering.wraps(stmts.sin)
 def sin(x: float) -> float: ...
 
+
 @lowering.wraps(stmts.sinh)
 def sinh(x: float) -> float: ...
 
+
 @lowering.wraps(stmts.sqrt)
 def sqrt(x: float) -> float: ...
 
+
 @lowering.wraps(stmts.tan)
 def tan(x: float) -> float: ...
 
+
 @lowering.wraps(stmts.tanh)
 def tanh(x: float) -> float: ...
 
+
 @lowering.wraps(stmts.trunc)
 def trunc(x: float) -> int: ...
 
+
 @lowering.wraps(stmts.ulp)
 def ulp(x: float) -> float: ...
-

src/kirin/dialects/math/interp.py

@@ -1,8 +1,9 @@
 # This file is generated by gen.py
 import math
-from kirin.dialects.math.dialect import dialect
+
+from kirin.interp import Frame, MethodTable, impl
 from kirin.dialects.math import stmts
-from kirin.interp import MethodTable, Frame, impl
+from kirin.dialects.math.dialect import dialect
 
 
 @dialect.register
@@ -13,217 +14,181 @@ def acos(self, interp, frame: Frame, stmt: stmts.acos):
         values = frame.get_values(stmt.args)
         return (math.acos(values[0]),)
 
-
     @impl(stmts.asin)
     def asin(self, interp, frame: Frame, stmt: stmts.asin):
         values = frame.get_values(stmt.args)
         return (math.asin(values[0]),)
 
-
     @impl(stmts.asinh)
     def asinh(self, interp, frame: Frame, stmt: stmts.asinh):
         values = frame.get_values(stmt.args)
         return (math.asinh(values[0]),)
 
-
     @impl(stmts.atan)
     def atan(self, interp, frame: Frame, stmt: stmts.atan):
         values = frame.get_values(stmt.args)
         return (math.atan(values[0]),)
 
-
     @impl(stmts.atan2)
     def atan2(self, interp, frame: Frame, stmt: stmts.atan2):
         values = frame.get_values(stmt.args)
         return (math.atan2(values[0], values[1]),)
 
-
     @impl(stmts.atanh)
     def atanh(self, interp, frame: Frame, stmt: stmts.atanh):
         values = frame.get_values(stmt.args)
         return (math.atanh(values[0]),)
 
-
     @impl(stmts.ceil)
     def ceil(self, interp, frame: Frame, stmt: stmts.ceil):
         values = frame.get_values(stmt.args)
         return (math.ceil(values[0]),)
 
-
     @impl(stmts.copysign)
     def copysign(self, interp, frame: Frame, stmt: stmts.copysign):
         values = frame.get_values(stmt.args)
         return (math.copysign(values[0], values[1]),)
 
-
     @impl(stmts.cos)
     def cos(self, interp, frame: Frame, stmt: stmts.cos):
         values = frame.get_values(stmt.args)
         return (math.cos(values[0]),)
 
-
     @impl(stmts.cosh)
     def cosh(self, interp, frame: Frame, stmt: stmts.cosh):
         values = frame.get_values(stmt.args)
         return (math.cosh(values[0]),)
 
-
     @impl(stmts.degrees)
     def degrees(self, interp, frame: Frame, stmt: stmts.degrees):
         values = frame.get_values(stmt.args)
         return (math.degrees(values[0]),)
 
-
     @impl(stmts.erf)
     def erf(self, interp, frame: Frame, stmt: stmts.erf):
         values = frame.get_values(stmt.args)
         return (math.erf(values[0]),)
 
-
     @impl(stmts.erfc)
     def erfc(self, interp, frame: Frame, stmt: stmts.erfc):
         values = frame.get_values(stmt.args)
         return (math.erfc(values[0]),)
 
-
     @impl(stmts.exp)
     def exp(self, interp, frame: Frame, stmt: stmts.exp):
         values = frame.get_values(stmt.args)
         return (math.exp(values[0]),)
 
-
     @impl(stmts.expm1)
     def expm1(self, interp, frame: Frame, stmt: stmts.expm1):
         values = frame.get_values(stmt.args)
         return (math.expm1(values[0]),)
 
-
     @impl(stmts.fabs)
     def fabs(self, interp, frame: Frame, stmt: stmts.fabs):
         values = frame.get_values(stmt.args)
         return (math.fabs(values[0]),)
 
-
     @impl(stmts.floor)
     def floor(self, interp, frame: Frame, stmt: stmts.floor):
         values = frame.get_values(stmt.args)
         return (math.floor(values[0]),)
 
-
     @impl(stmts.fmod)
     def fmod(self, interp, frame: Frame, stmt: stmts.fmod):
         values = frame.get_values(stmt.args)
         return (math.fmod(values[0], values[1]),)
 
-
     @impl(stmts.gamma)
     def gamma(self, interp, frame: Frame, stmt: stmts.gamma):
         values = frame.get_values(stmt.args)
         return (math.gamma(values[0]),)
 
-
     @impl(stmts.isfinite)
     def isfinite(self, interp, frame: Frame, stmt: stmts.isfinite):
         values = frame.get_values(stmt.args)
         return (math.isfinite(values[0]),)
 
-
     @impl(stmts.isinf)
     def isinf(self, interp, frame: Frame, stmt: stmts.isinf):
         values = frame.get_values(stmt.args)
         return (math.isinf(values[0]),)
 
-
     @impl(stmts.isnan)
     def isnan(self, interp, frame: Frame, stmt: stmts.isnan):
         values = frame.get_values(stmt.args)
         return (math.isnan(values[0]),)
 
-
     @impl(stmts.lgamma)
     def lgamma(self, interp, frame: Frame, stmt: stmts.lgamma):
         values = frame.get_values(stmt.args)
         return (math.lgamma(values[0]),)
 
-
     @impl(stmts.log)
     def log(self, interp, frame: Frame, stmt: stmts.log):
         values = frame.get_values(stmt.args)
         return (math.log(values[0], values[1]),)
 
-
     @impl(stmts.log10)
     def log10(self, interp, frame: Frame, stmt: stmts.log10):
         values = frame.get_values(stmt.args)
         return (math.log10(values[0]),)
 
-
     @impl(stmts.log1p)
     def log1p(self, interp, frame: Frame, stmt: stmts.log1p):
         values = frame.get_values(stmt.args)
         return (math.log1p(values[0]),)
 
-
     @impl(stmts.log2)
     def log2(self, interp, frame: Frame, stmt: stmts.log2):
         values = frame.get_values(stmt.args)
         return (math.log2(values[0]),)
 
-
     @impl(stmts.pow)
     def pow(self, interp, frame: Frame, stmt: stmts.pow):
         values = frame.get_values(stmt.args)
         return (math.pow(values[0], values[1]),)
 
-
     @impl(stmts.radians)
     def radians(self, interp, frame: Frame, stmt: stmts.radians):
         values = frame.get_values(stmt.args)
         return (math.radians(values[0]),)
 
-
     @impl(stmts.remainder)
     def remainder(self, interp, frame: Frame, stmt: stmts.remainder):
         values = frame.get_values(stmt.args)
         return (math.remainder(values[0], values[1]),)
 
-
     @impl(stmts.sin)
     def sin(self, interp, frame: Frame, stmt: stmts.sin):
         values = frame.get_values(stmt.args)
         return (math.sin(values[0]),)
 
-
     @impl(stmts.sinh)
     def sinh(self, interp, frame: Frame, stmt: stmts.sinh):
         values = frame.get_values(stmt.args)
         return (math.sinh(values[0]),)
 
-
     @impl(stmts.sqrt)
     def sqrt(self, interp, frame: Frame, stmt: stmts.sqrt):
         values = frame.get_values(stmt.args)
         return (math.sqrt(values[0]),)
 
-
     @impl(stmts.tan)
     def tan(self, interp, frame: Frame, stmt: stmts.tan):
         values = frame.get_values(stmt.args)
         return (math.tan(values[0]),)
 
-
     @impl(stmts.tanh)
     def tanh(self, interp, frame: Frame, stmt: stmts.tanh):
         values = frame.get_values(stmt.args)
         return (math.tanh(values[0]),)
 
-
     @impl(stmts.trunc)
     def trunc(self, interp, frame: Frame, stmt: stmts.trunc):
         values = frame.get_values(stmt.args)
         return (math.trunc(values[0]),)
 
-
     @impl(stmts.ulp)
     def ulp(self, interp, frame: Frame, stmt: stmts.ulp):
         values = frame.get_values(stmt.args)