From 1949a6ec3bf3567b0d8b1d02caa7a4bf0a6f5ac5 Mon Sep 17 00:00:00 2001
From: Maximilian Algehed <m.algehed@gmail.com>
Date: Mon, 15 Apr 2024 20:47:00 +0200
Subject: [PATCH] more work

---
 libs/cardano-ledger-test/bench/Bench/Constrained/STS.hs | 5 +++--
 libs/constrained-generators/src/Constrained/Base.hs     | 8 ++++----
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/libs/cardano-ledger-test/bench/Bench/Constrained/STS.hs b/libs/cardano-ledger-test/bench/Bench/Constrained/STS.hs
index 33e2b1cf9fd..4f9d180fd9f 100644
--- a/libs/cardano-ledger-test/bench/Bench/Constrained/STS.hs
+++ b/libs/cardano-ledger-test/bench/Bench/Constrained/STS.hs
@@ -12,13 +12,14 @@ import Cardano.Ledger.Conway
 import Cardano.Ledger.Conway.Rules
 import Cardano.Ledger.Crypto
 import Constrained
+import Control.DeepSeq
 import Criterion
 import Test.Cardano.Ledger.Constrained.Conway
 
 govEnv :: GovEnv (ConwayEra StandardCrypto)
 govEnv = genFromSpecWithSeed 10 30 (govEnvSpec @ConwayFn)
 
-singleProposalTreeSpec :: Spec ConwayFn ProposalTree
+singleProposalTreeSpec :: Specification ConwayFn ProposalTree
 singleProposalTreeSpec = constrained $ \ppupTree ->
   [ wellFormedChildren (lit SNothing) ppupTree
   , allGASInTree ppupTree $ \gas ->
@@ -38,7 +39,7 @@ stsBenchmarks =
   where
     govPropSpec = govProposalsSpec @ConwayFn govEnv
 
-benchSpec :: (HasSpec fn a, NFData a) => Int -> Int -> String -> Spec fn a -> Benchmark
+benchSpec :: (HasSpec fn a, NFData a) => Int -> Int -> String -> Specification fn a -> Benchmark
 benchSpec seed size nm spec =
   bench (unlines [nm, show (genFromSpecWithSeed seed size spec)]) $
     nf (genFromSpecWithSeed seed size) spec
diff --git a/libs/constrained-generators/src/Constrained/Base.hs b/libs/constrained-generators/src/Constrained/Base.hs
index 4a0eb00c3d7..262c2d5bc44 100644
--- a/libs/constrained-generators/src/Constrained/Base.hs
+++ b/libs/constrained-generators/src/Constrained/Base.hs
@@ -2340,9 +2340,9 @@ genNumList elemSIn foldSIn = do
   normFoldS <- normalize foldSIn
   explain
     [ "Can't generate list of ints with fold constraint"
-    , show $ "  elemSpec =" <+> pretty elemSIn
-    , show $ "  normElemSpec =" <+> pretty normElemS
-    , show $ "  foldSpec =" <+> pretty foldSIn
+    , "  elemSpec = " ++ show elemSIn
+    , "  normElemSpec = " ++ show normElemS
+    , "  foldSpec = " ++ show foldSIn
     ]
     $ gen (narrowFoldSpecs (normElemS, normFoldS)) 50 [] >>= pureGen . shuffle
   where
@@ -2376,7 +2376,7 @@ genNumList elemSIn foldSIn = do
       , not $ 0 `conformsToSpec` foldS =
           genError
             [ "Ran out of fuel in genNumList"
-            , "  " ++ show ("elemSpec =" <+> pretty elemSIn)
+            , "  elemSpec =" ++ show elemSIn
             , "  foldSpec = " ++ show foldSIn
             , "  lst = " ++ show lst
             ]