Fix

OpShin · Dec 29, 2024 · 33e4256 · 33e4256
1 parent 2968732
commit 33e4256
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/uplc/cost_model.py b/uplc/cost_model.py
@@ -380,7 +380,7 @@ class LiteralInYOrLinearInZ(CostingFun):
     slope: int = 0
 
     def cost(self, *memories: int, values=[]) -> int:
-        y = values[1]
+        y = values[1].value
         if y == 0:
             return self.intercept + self.slope * memories[2]
         # "+ 7" for ceil

diff --git a/uplc/machine.py b/uplc/machine.py
@@ -22,12 +22,13 @@ def budget_cost_of_op_on_model(
     model: Union[BuiltinCostModel, CekMachineCostModel],
     op: Union[BuiltInFun, CekOp],
     *args: int,
+    values=[],
 ):
     if op not in model.cpu or op not in model.memory:
         return Budget(0, 0)
     return Budget(
-        cpu=model.cpu[op].cost(*args),
-        memory=model.memory[op].cost(*args),
+        cpu=model.cpu[op].cost(*args, values=values),
+        memory=model.memory[op].cost(*args, values=values),
     )
 
 
@@ -269,6 +270,7 @@ def apply_evaluate(self, context, function, argument):
                         self.builtin_cost_model,
                         function.builtin,
                         *(arg.ex_mem() for arg in arguments),
+                        values=arguments,
                     )
                     self.spend_budget(cost)
                     if function.builtin == BuiltInFun.Trace: