Merge pull request #31 from uncountableinc/catherine/ceiling-handle-t…

…ensor-input fix non tensor input to ceiling function
uncountableinc · Oct 25, 2023 · e0c43f5 · e0c43f5
2 parents 4828d52 + 5da5c3a
commit e0c43f5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/hotxlfp/formulas/mathtrig.py b/hotxlfp/formulas/mathtrig.py
@@ -245,14 +245,16 @@ def SUMIF(args, criteria):
 
 @dispatcher.register_for("CEILING", "CEILING.MATH", "CEILING.PRECISE")
 def CEILING(number, significance=1):
-    number = torch.tensor(utils.parse_number(number))
+    number = utils.parse_number(number)
     significance = utils.parse_number(significance)
-    if not isinstance(significance, torch.Tensor) or significance.size(dim=0) == 1:
+    if isinstance(number, torch.Tensor) and (not isinstance(significance, torch.Tensor) or significance.dim() == 0 or significance.size(dim=0) == 1):
         significance = torch.broadcast_to(torch.tensor(significance), number.size())
 
+    number = torch.tensor(number)
+    significance = torch.tensor(significance)
     if utils.any_is_error((number, significance)):
         return error.VALUE
-    if number.size(dim=0) != significance.size(dim=0):
+    if number.size() != significance.size():
         return error.VALUE
 
     positive_number = torch.where(

diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setup(
     name="hotxlfp",
-    version="0.0.11+unc.27",
+    version="0.0.11+unc.28",
     packages=[
         "hotxlfp",
         "hotxlfp._compat",