From 96fd8ecf7c88c878e75fc36c7f24e8cf5263b6ee Mon Sep 17 00:00:00 2001
From: Kyle Altendorf <sda@fstab.net>
Date: Fri, 24 Jan 2025 10:01:07 -0500
Subject: [PATCH] more

---
 chia/_tests/core/data_layer/test_merkle_blob.py | 6 +++---
 chia/data_layer/util/merkle_blob.py             | 2 +-
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/chia/_tests/core/data_layer/test_merkle_blob.py b/chia/_tests/core/data_layer/test_merkle_blob.py
index b7cecd3bf1a8..0229a1ee1ddd 100644
--- a/chia/_tests/core/data_layer/test_merkle_blob.py
+++ b/chia/_tests/core/data_layer/test_merkle_blob.py
@@ -234,10 +234,10 @@ def generate_kvid(seed: int) -> tuple[KVId, KVId]:
     return kv_ids[0], kv_ids[1]
 
 
-def generate_hash(seed: int) -> bytes:
+def generate_hash(seed: int) -> bytes32:
     seed_bytes = seed.to_bytes(8, byteorder="big", signed=True)
     hash_obj = hashlib.sha256(seed_bytes)
-    return hash_obj.digest()
+    return bytes32(hash_obj.digest())
 
 
 def test_insert_delete_loads_all_keys() -> None:
@@ -337,7 +337,7 @@ def test_proof_of_inclusion_merkle_blob() -> None:
         num_deletes = 1 + repeats * 10
 
         kv_ids: list[tuple[KVId, KVId]] = []
-        hashes: list[bytes] = []
+        hashes: list[bytes32] = []
         for _ in range(num_inserts):
             seed += 1
             key, value = generate_kvid(seed)
diff --git a/chia/data_layer/util/merkle_blob.py b/chia/data_layer/util/merkle_blob.py
index 29a5950edef2..02745d364277 100644
--- a/chia/data_layer/util/merkle_blob.py
+++ b/chia/data_layer/util/merkle_blob.py
@@ -560,7 +560,7 @@ def get_nodes_with_indexes(self, index: TreeIndex = TreeIndex(0)) -> list[tuple[
 
         return this + left_nodes + right_nodes
 
-    def batch_insert(self, keys_values: list[tuple[KVId, KVId]], hashes: list[bytes]) -> None:
+    def batch_insert(self, keys_values: list[tuple[KVId, KVId]], hashes: list[bytes32]) -> None:
         indexes: list[TreeIndex] = []
 
         if len(self.key_to_index) <= 1: