From f5ed230cdc65bbd5559d998114b37868bf4817f8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jann=20M=C3=BCller?= <j-mueller@users.noreply.github.com>
Date: Sun, 26 Jan 2025 11:21:02 +0100
Subject: [PATCH] Log finished ex units map

---
 src/coin-selection/lib/Convex/CoinSelection.hs | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/coin-selection/lib/Convex/CoinSelection.hs b/src/coin-selection/lib/Convex/CoinSelection.hs
index 34ab4c6..7cb267d 100644
--- a/src/coin-selection/lib/Convex/CoinSelection.hs
+++ b/src/coin-selection/lib/Convex/CoinSelection.hs
@@ -228,8 +228,8 @@ data TxBalancingMessage
     PrepareInputs {availableBalance :: C.Value, transactionBalance :: C.Value}
   | -- | Balancing a transaction body
     StartBalancing {numInputs :: !Int, numOutputs :: !Int}
-  | -- | Execution units of the transaction, or error message in case of script error
-    ExUnitsMap {exUnits :: [(C.ScriptWitnessIndex, Either String C.ExecutionUnits)]}
+  | -- | Execution units of the transaction
+    ExUnitsMap {exUnits :: [(C.ScriptWitnessIndex, C.ExecutionUnits)]}
   | -- | The transaction fee
     Txfee {fee :: C.Quantity}
   | -- | The remaining balance (after paying the fee)
@@ -292,13 +292,14 @@ balanceTransactionBody
           csiUtxo
           txbody0
 
-    traceWith tracer $ ExUnitsMap $ fmap (second (first (C.docToString . C.prettyError) . fmap snd)) $ Map.toList exUnitsMap
-
     exUnitsMap' <- balancingError $
       case Map.mapEither id exUnitsMap of
         (failures, exUnitsMap') ->
           handleExUnitsErrors (C.txScriptValidityToScriptValidity $ C.txScriptValidity csiTxBody) failures $
             fmap snd exUnitsMap'
+
+    traceWith tracer $ ExUnitsMap $ Map.toList exUnitsMap'
+
     txbodycontent1 <- balancingError $ substituteExecutionUnits exUnitsMap' csiTxBody
     let txbodycontent1' = txbodycontent1 & set L.txFee (Coin (2 ^ (32 :: Integer) - 1)) & over L.txOuts (|> changeOutputLarge)