cleanup

Genentech · Aug 28, 2024 · 31d3d07 · 31d3d07
1 parent 7303f20
commit 31d3d07
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/beignet/_chandrupatla.py b/src/beignet/_chandrupatla.py
@@ -3,14 +3,14 @@
 import torch
 from torch import Tensor
 
-from beignet._root_scalar import RootSolutionInfo
+from ._root_scalar import RootSolutionInfo
 
 
 def chandrupatla(
     func: Callable,
     *args,
-    lower: float,
-    upper: float,
+    lower: float | Tensor,
+    upper: float | Tensor,
     rtol: float | None = None,
     atol: float | None = None,
     maxiter: int = 100,