Fall back to old typing and ignore star imports for asdfghjkl_src

laplace/curvature/asdfghjkl_src/__init__.py

@@ -1,10 +1,10 @@
 from .core import extend
-from .fisher import *
-from .gradient import *
-from .hessian import *
-from .kernel import *
-from .matrices import *
-from .mvp import *
-from .operations import *
-from .symmatrix import *
-from .utils import *
+from .fisher import *  # noqa: F403
+from .gradient import *  # noqa: F403
+from .hessian import *  # noqa: F403
+from .kernel import *  # noqa: F403
+from .matrices import *  # noqa: F403
+from .mvp import *  # noqa: F403
+from .operations import *  # noqa: F403
+from .symmatrix import *  # noqa: F403
+from .utils import *  # noqa: F403

laplace/curvature/asdfghjkl_src/mvp.py

@@ -155,7 +155,7 @@ def _group_product(xs, ys):
     return sum([torch.sum(x * y) for (x, y) in zip(xs, ys)])
 
 
-def _group_add(xs, ys, alpha: float | torch.Tensor=1.):
+def _group_add(xs, ys, alpha=1.):
     return [x.add(y.mul(alpha)) for x, y in zip(xs, ys)]