chia-blockchain 2.5.0rc1__py3-none-any.whl → 2.5.1__py3-none-any.whl

chia/full_node/weight_proof.py

@@ -8,7 +8,7 @@ import random
 import tempfile
 from concurrent.futures.process import ProcessPoolExecutor
 from multiprocessing.context import BaseContext
-from typing import IO, Dict, List, Optional, Tuple
+from typing import IO, Optional
 
 from chia.consensus.block_header_validation import validate_finished_header_block
 from chia.consensus.block_record import BlockRecord
@@ -31,6 +31,7 @@ from chia.types.blockchain_format.sub_epoch_summary import SubEpochSummary
 from chia.types.blockchain_format.vdf import VDFInfo, VDFProof, validate_vdf
 from chia.types.end_of_slot_bundle import EndOfSubSlotBundle
 from chia.types.header_block import HeaderBlock
+from chia.types.validation_state import ValidationState
 from chia.types.weight_proof import (
     RecentChainData,
     SubEpochChallengeSegment,
@@ -44,6 +45,7 @@ from chia.util.block_cache import BlockCache
 from chia.util.hash import std_hash
 from chia.util.ints import uint8, uint32, uint64, uint128
 from chia.util.setproctitle import getproctitle, setproctitle
+from chia.util.task_referencer import create_referenced_task
 
 log = logging.getLogger(__name__)
 
@@ -92,8 +94,8 @@ class WeightProofHandler:
             self.tip = tip
             return wp
 
-    def get_sub_epoch_data(self, tip_height: uint32, summary_heights: List[uint32]) -> List[SubEpochData]:
-        sub_epoch_data: List[SubEpochData] = []
+    def get_sub_epoch_data(self, tip_height: uint32, summary_heights: list[uint32]) -> list[SubEpochData]:
+        sub_epoch_data: list[SubEpochData] = []
         for sub_epoch_n, ses_height in enumerate(summary_heights):
             if ses_height > tip_height:
                 break
@@ -107,7 +109,7 @@ class WeightProofHandler:
         Creates a weight proof object
         """
         assert self.blockchain is not None
-        sub_epoch_segments: List[SubEpochChallengeSegment] = []
+        sub_epoch_segments: list[SubEpochChallengeSegment] = []
         tip_rec = self.blockchain.try_block_record(tip)
         if tip_rec is None:
             log.error("failed not tip in cache")
@@ -164,7 +166,7 @@ class WeightProofHandler:
         log.debug(f"sub_epochs: {len(sub_epoch_data)}")
         return WeightProof(sub_epoch_data, sub_epoch_segments, recent_chain)
 
-    def get_seed_for_proof(self, summary_heights: List[uint32], tip_height: uint32) -> bytes32:
+    def get_seed_for_proof(self, summary_heights: list[uint32], tip_height: uint32) -> bytes32:
         count = 0
         ses = None
         for sub_epoch_n, ses_height in enumerate(reversed(summary_heights)):
@@ -177,8 +179,8 @@ class WeightProofHandler:
         seed = ses.get_hash()
         return seed
 
-    async def _get_recent_chain(self, tip_height: uint32) -> Optional[List[HeaderBlock]]:
-        recent_chain: List[HeaderBlock] = []
+    async def _get_recent_chain(self, tip_height: uint32) -> Optional[list[HeaderBlock]]:
+        recent_chain: list[HeaderBlock] = []
         ses_heights = self.blockchain.get_ses_heights()
         min_height = 0
         count_ses = 0
@@ -288,8 +290,8 @@ class WeightProofHandler:
 
     async def __create_sub_epoch_segments(
         self, ses_block: BlockRecord, se_start: BlockRecord, sub_epoch_n: uint32
-    ) -> Optional[List[SubEpochChallengeSegment]]:
-        segments: List[SubEpochChallengeSegment] = []
+    ) -> Optional[list[SubEpochChallengeSegment]]:
+        segments: list[SubEpochChallengeSegment] = []
         start_height = await self.get_prev_two_slots_height(se_start)
 
         blocks = await self.blockchain.get_block_records_in_range(
@@ -345,12 +347,12 @@ class WeightProofHandler:
         self,
         header_block: HeaderBlock,
         sub_epoch_n: uint32,
-        header_blocks: Dict[bytes32, HeaderBlock],
-        blocks: Dict[bytes32, BlockRecord],
+        header_blocks: dict[bytes32, HeaderBlock],
+        blocks: dict[bytes32, BlockRecord],
         first_segment_in_sub_epoch: bool,
-    ) -> Tuple[Optional[SubEpochChallengeSegment], uint32]:
+    ) -> tuple[Optional[SubEpochChallengeSegment], uint32]:
         assert self.blockchain is not None
-        sub_slots: List[SubSlotData] = []
+        sub_slots: list[SubSlotData] = []
         log.debug(f"create challenge segment block {header_block.header_hash} block height {header_block.height} ")
         # VDFs from sub slots before challenge block
         first_sub_slots, first_rc_end_of_slot_vdf = await self.__first_sub_slot_vdfs(
@@ -392,10 +394,10 @@ class WeightProofHandler:
     async def __first_sub_slot_vdfs(
         self,
         header_block: HeaderBlock,
-        header_blocks: Dict[bytes32, HeaderBlock],
-        blocks: Dict[bytes32, BlockRecord],
+        header_blocks: dict[bytes32, HeaderBlock],
+        blocks: dict[bytes32, BlockRecord],
         first_in_sub_epoch: bool,
-    ) -> Tuple[Optional[List[SubSlotData]], Optional[VDFInfo]]:
+    ) -> tuple[Optional[list[SubSlotData]], Optional[VDFInfo]]:
         # combine cc vdfs of all reward blocks from the start of the sub slot to end
         header_block_sub_rec = blocks[header_block.header_hash]
         # find slot start
@@ -418,8 +420,8 @@ class WeightProofHandler:
                     curr_sub_rec = blocks[curr_sub_rec.prev_hash]
 
         curr = header_blocks[curr_sub_rec.header_hash]
-        sub_slots_data: List[SubSlotData] = []
-        tmp_sub_slots_data: List[SubSlotData] = []
+        sub_slots_data: list[SubSlotData] = []
+        tmp_sub_slots_data: list[SubSlotData] = []
         while curr.height < header_block.height:
             if curr is None:
                 log.error("failed fetching block")
@@ -469,8 +471,8 @@ class WeightProofHandler:
     def first_rc_end_of_slot_vdf(
         self,
         header_block: HeaderBlock,
-        blocks: Dict[bytes32, BlockRecord],
-        header_blocks: Dict[bytes32, HeaderBlock],
+        blocks: dict[bytes32, BlockRecord],
+        header_blocks: dict[bytes32, HeaderBlock],
     ) -> Optional[VDFInfo]:
         curr = blocks[header_block.header_hash]
         while curr.height > 0 and not curr.sub_epoch_summary_included:
@@ -478,16 +480,16 @@ class WeightProofHandler:
         return header_blocks[curr.header_hash].finished_sub_slots[-1].reward_chain.end_of_slot_vdf
 
     async def __slot_end_vdf(
-        self, start_height: uint32, header_blocks: Dict[bytes32, HeaderBlock], blocks: Dict[bytes32, BlockRecord]
-    ) -> Tuple[Optional[List[SubSlotData]], uint32]:
+        self, start_height: uint32, header_blocks: dict[bytes32, HeaderBlock], blocks: dict[bytes32, BlockRecord]
+    ) -> tuple[Optional[list[SubSlotData]], uint32]:
         # gets all vdfs first sub slot after challenge block to last sub slot
         log.debug(f"slot end vdf start height {start_height}")
         header_hash = self.blockchain.height_to_hash(start_height)
         assert header_hash is not None
         curr = header_blocks[header_hash]
         curr_header_hash = curr.header_hash
-        sub_slots_data: List[SubSlotData] = []
-        tmp_sub_slots_data: List[SubSlotData] = []
+        sub_slots_data: list[SubSlotData] = []
+        tmp_sub_slots_data: list[SubSlotData] = []
         while not blocks[curr_header_hash].is_challenge_block(self.constants):
             if curr.first_in_sub_slot:
                 sub_slots_data.extend(tmp_sub_slots_data)
@@ -512,7 +514,7 @@ class WeightProofHandler:
         log.debug(f"slot end vdf end height {curr.height} slots {len(sub_slots_data)} ")
         return sub_slots_data, curr.height
 
-    def handle_block_vdfs(self, curr: HeaderBlock, blocks: Dict[bytes32, BlockRecord]) -> SubSlotData:
+    def handle_block_vdfs(self, curr: HeaderBlock, blocks: dict[bytes32, BlockRecord]) -> SubSlotData:
         cc_sp_proof = None
         icc_ip_proof = None
         cc_sp_info = None
@@ -558,7 +560,7 @@ class WeightProofHandler:
             curr.total_iters,
         )
 
-    def validate_weight_proof_single_proc(self, weight_proof: WeightProof) -> Tuple[bool, uint32]:
+    def validate_weight_proof_single_proc(self, weight_proof: WeightProof) -> tuple[bool, uint32]:
         assert self.blockchain is not None
         assert len(weight_proof.sub_epochs) > 0
         if len(weight_proof.sub_epochs) == 0:
@@ -588,7 +590,7 @@ class WeightProofHandler:
         fork_point, _ = self.get_fork_point(summaries)
         return True, fork_point
 
-    async def validate_weight_proof(self, weight_proof: WeightProof) -> Tuple[bool, uint32, List[SubEpochSummary]]:
+    async def validate_weight_proof(self, weight_proof: WeightProof) -> tuple[bool, uint32, list[SubEpochSummary]]:
         assert self.blockchain is not None
         if len(weight_proof.sub_epochs) == 0:
             return False, uint32(0), []
@@ -613,7 +615,7 @@ class WeightProofHandler:
             # The shutdown file manager must be inside of the executor manager so that
             # we request the workers close prior to waiting for them to close.
             with _create_shutdown_file() as shutdown_file:
-                task = asyncio.create_task(
+                task = create_referenced_task(
                     validate_weight_proof_inner(
                         self.constants,
                         executor,
@@ -629,7 +631,7 @@ class WeightProofHandler:
                 valid, _ = await task
         return valid, fork_point, summaries
 
-    def get_fork_point(self, received_summaries: List[SubEpochSummary]) -> Tuple[uint32, int]:
+    def get_fork_point(self, received_summaries: list[SubEpochSummary]) -> tuple[uint32, int]:
         # returns the fork height and ses index
         # iterate through sub epoch summaries to find fork point
         fork_point_index = 0
@@ -653,8 +655,8 @@ class WeightProofHandler:
 
 
 def _get_weights_for_sampling(
-    rng: random.Random, total_weight: uint128, recent_chain: List[HeaderBlock]
-) -> Optional[List[uint128]]:
+    rng: random.Random, total_weight: uint128, recent_chain: list[HeaderBlock]
+) -> Optional[list[uint128]]:
     weight_to_check = []
     last_l_weight = recent_chain[-1].reward_chain_block.weight - recent_chain[0].reward_chain_block.weight
     delta = last_l_weight / total_weight
@@ -675,10 +677,10 @@ def _get_weights_for_sampling(
 def _sample_sub_epoch(
     start_of_epoch_weight: uint128,
     end_of_epoch_weight: uint128,
-    weight_to_check: Optional[List[uint128]],
+    weight_to_check: Optional[list[uint128]],
 ) -> bool:
     """
-    weight_to_check: List[uint128] is expected to be sorted
+    weight_to_check: list[uint128] is expected to be sorted
     """
     if weight_to_check is None:
         return True
@@ -717,7 +719,7 @@ async def _challenge_block_vdfs(
     constants: ConsensusConstants,
     header_block: HeaderBlock,
     block_rec: BlockRecord,
-    sub_blocks: Dict[bytes32, BlockRecord],
+    sub_blocks: dict[bytes32, BlockRecord],
 ) -> SubSlotData:
     (_, _, _, _, cc_vdf_iters, _) = get_signage_point_vdf_info(
         constants,
@@ -828,7 +830,7 @@ def handle_end_of_slot(
 def _validate_sub_epoch_summaries(
     constants: ConsensusConstants,
     weight_proof: WeightProof,
-) -> Tuple[Optional[List[SubEpochSummary]], Optional[List[uint128]]]:
+) -> tuple[Optional[list[SubEpochSummary]], Optional[list[uint128]]]:
     last_ses_hash, last_ses_sub_height = _get_last_ses_hash(constants, weight_proof.recent_chain_data)
     if last_ses_hash is None:
         log.warning("could not find last ses block")
@@ -861,12 +863,12 @@ def _validate_sub_epoch_summaries(
 def _map_sub_epoch_summaries(
     sub_blocks_for_se: uint32,
     ses_hash: bytes32,
-    sub_epoch_data: List[SubEpochData],
+    sub_epoch_data: list[SubEpochData],
     curr_difficulty: uint64,
-) -> Tuple[List[SubEpochSummary], uint128, List[uint128]]:
+) -> tuple[list[SubEpochSummary], uint128, list[uint128]]:
     total_weight: uint128 = uint128(0)
-    summaries: List[SubEpochSummary] = []
-    sub_epoch_weight_list: List[uint128] = []
+    summaries: list[SubEpochSummary] = []
+    sub_epoch_weight_list: list[uint128] = []
     for idx, data in enumerate(sub_epoch_data):
         ses = SubEpochSummary(
             ses_hash,
@@ -879,8 +881,8 @@ def _map_sub_epoch_summaries(
         if idx < len(sub_epoch_data) - 1:
             delta = 0
             if idx > 0:
-                delta = sub_epoch_data[idx].num_blocks_overflow
-            log.debug(f"sub epoch {idx} start weight is {total_weight+curr_difficulty} ")
+                delta = data.num_blocks_overflow
+            log.debug(f"sub epoch {idx} start weight is {total_weight + curr_difficulty} ")
             sub_epoch_weight_list.append(uint128(total_weight + curr_difficulty))
             total_weight = uint128(
                 total_weight
@@ -902,7 +904,7 @@ def _map_sub_epoch_summaries(
 def _validate_summaries_weight(
     constants: ConsensusConstants,
     sub_epoch_data_weight: uint128,
-    summaries: List[SubEpochSummary],
+    summaries: list[SubEpochSummary],
     weight_proof: WeightProof,
 ) -> bool:
     num_over = summaries[-1].num_blocks_overflow
@@ -921,10 +923,10 @@ def _validate_sub_epoch_segments(
     constants: ConsensusConstants,
     rng: random.Random,
     weight_proof_bytes: bytes,
-    summaries_bytes: List[bytes],
+    summaries_bytes: list[bytes],
     height: uint32,
     validate_from: int = 0,
-) -> Optional[List[Tuple[VDFProof, ClassgroupElement, VDFInfo]]]:
+) -> Optional[list[tuple[VDFProof, ClassgroupElement, VDFInfo]]]:
     summaries = summaries_from_bytes(summaries_bytes)
     sub_epoch_segments: SubEpochSegments = SubEpochSegments.from_bytes(weight_proof_bytes)
     rc_sub_slot_hash = constants.GENESIS_CHALLENGE
@@ -987,7 +989,7 @@ def _validate_segment(
     first_segment_in_se: bool,
     sampled: bool,
     height: uint32,
-) -> Tuple[bool, int, int, int, List[Tuple[VDFProof, ClassgroupElement, VDFInfo]]]:
+) -> tuple[bool, int, int, int, list[tuple[VDFProof, ClassgroupElement, VDFInfo]]]:
     ip_iters, slot_iters, slots = 0, 0, 0
     after_challenge = False
     to_validate = []
@@ -1000,9 +1002,7 @@ def _validate_segment(
             if required_iters is None:
                 return False, uint64(0), uint64(0), uint64(0), []
             assert sub_slot_data.signage_point_index is not None
-            ip_iters = ip_iters + calculate_ip_iters(
-                constants, curr_ssi, sub_slot_data.signage_point_index, required_iters
-            )
+            ip_iters += calculate_ip_iters(constants, curr_ssi, sub_slot_data.signage_point_index, required_iters)
             vdf_list = _get_challenge_block_vdfs(constants, idx, segment.sub_slots, curr_ssi)
             to_validate.extend(vdf_list)
         elif sampled and after_challenge:
@@ -1011,17 +1011,17 @@ def _validate_segment(
                 log.error(f"failed to validate sub slot data {idx} vdfs")
                 return False, uint64(0), uint64(0), uint64(0), []
             to_validate.extend(vdf_list)
-        slot_iters = slot_iters + curr_ssi
-        slots = slots + uint64(1)
+        slot_iters += curr_ssi
+        slots += uint64(1)
     return True, ip_iters, slot_iters, slots, to_validate
 
 
 def _get_challenge_block_vdfs(
     constants: ConsensusConstants,
     sub_slot_idx: int,
-    sub_slots: List[SubSlotData],
+    sub_slots: list[SubSlotData],
     ssi: uint64,
-) -> List[Tuple[VDFProof, ClassgroupElement, VDFInfo]]:
+) -> list[tuple[VDFProof, ClassgroupElement, VDFInfo]]:
     to_validate = []
     sub_slot_data = sub_slots[sub_slot_idx]
     if sub_slot_data.cc_signage_point is not None and sub_slot_data.cc_sp_vdf_info:
@@ -1058,9 +1058,9 @@ def _get_challenge_block_vdfs(
 def _validate_sub_slot_data(
     constants: ConsensusConstants,
     sub_slot_idx: int,
-    sub_slots: List[SubSlotData],
+    sub_slots: list[SubSlotData],
     ssi: uint64,
-) -> Tuple[bool, List[Tuple[VDFProof, ClassgroupElement, VDFInfo]]]:
+) -> tuple[bool, list[tuple[VDFProof, ClassgroupElement, VDFInfo]]]:
     sub_slot_data = sub_slots[sub_slot_idx]
     assert sub_slot_idx > 0
     prev_ssd = sub_slots[sub_slot_idx - 1]
@@ -1134,7 +1134,7 @@ def sub_slot_data_vdf_input(
     constants: ConsensusConstants,
     sub_slot_data: SubSlotData,
     sub_slot_idx: int,
-    sub_slots: List[SubSlotData],
+    sub_slots: list[SubSlotData],
     is_overflow: bool,
     new_sub_slot: bool,
     ssi: uint64,
@@ -1198,9 +1198,9 @@ def sub_slot_data_vdf_input(
 def validate_recent_blocks(
     constants: ConsensusConstants,
     recent_chain_bytes: bytes,
-    summaries_bytes: List[bytes],
+    summaries_bytes: list[bytes],
     shutdown_file_path: Optional[pathlib.Path] = None,
-) -> Tuple[bool, List[bytes]]:
+) -> tuple[bool, list[bytes]]:
     recent_chain: RecentChainData = RecentChainData.from_bytes(recent_chain_bytes)
     summaries = summaries_from_bytes(summaries_bytes)
     sub_blocks = BlockCache({})
@@ -1253,8 +1253,9 @@ def validate_recent_blocks(
                 adjusted = True
             deficit = get_deficit(constants, deficit, prev_block_record, overflow, len(block.finished_sub_slots))
             if sub_slots > 2 and transaction_blocks > 11 and (tip_height - block.height < last_blocks_to_validate):
+                expected_vs = ValidationState(ssi, diff, None)
                 caluclated_required_iters, error = validate_finished_header_block(
-                    constants, sub_blocks, block, False, diff, ssi, ses_blocks > 2
+                    constants, sub_blocks, block, False, expected_vs, ses_blocks > 2
                 )
                 if error is not None:
                     log.error(f"block {block.header_hash} failed validation {error}")
@@ -1266,7 +1267,7 @@ def validate_recent_blocks(
                 if ret is None:
                     return False, []
                 required_iters = ret
-            validated_block_count = validated_block_count + 1
+            validated_block_count += 1
 
         curr_block_ses = None if not ses else summaries[ses_idx - 1]
         block_record = header_block_to_sub_block_record(
@@ -1384,7 +1385,7 @@ def __validate_pospace(
 def __get_rc_sub_slot(
     constants: ConsensusConstants,
     segment: SubEpochChallengeSegment,
-    summaries: List[SubEpochSummary],
+    summaries: list[SubEpochSummary],
     curr_ssi: uint64,
 ) -> RewardChainSubSlot:
     ses = summaries[uint32(segment.sub_epoch_n - 1)]
@@ -1465,7 +1466,7 @@ def __get_rc_sub_slot(
     return rc_sub_slot
 
 
-def __get_cc_sub_slot(sub_slots: List[SubSlotData], idx: int, ses: Optional[SubEpochSummary]) -> ChallengeChainSubSlot:
+def __get_cc_sub_slot(sub_slots: list[SubSlotData], idx: int, ses: Optional[SubEpochSummary]) -> ChallengeChainSubSlot:
     sub_slot: Optional[SubSlotData] = None
     for i in reversed(range(0, idx)):
         sub_slot = sub_slots[i]
@@ -1491,8 +1492,8 @@ def __get_cc_sub_slot(sub_slots: List[SubSlotData], idx: int, ses: Optional[SubE
 
 
 def _get_curr_diff_ssi(
-    constants: ConsensusConstants, idx: int, summaries: List[SubEpochSummary]
-) -> Tuple[uint64, uint64]:
+    constants: ConsensusConstants, idx: int, summaries: list[SubEpochSummary]
+) -> tuple[uint64, uint64]:
     curr_difficulty = constants.DIFFICULTY_STARTING
     curr_ssi = constants.SUB_SLOT_ITERS_STARTING
     for ses in reversed(summaries[0:idx]):
@@ -1505,7 +1506,7 @@ def _get_curr_diff_ssi(
     return curr_difficulty, curr_ssi
 
 
-def vars_to_bytes(summaries: List[SubEpochSummary], weight_proof: WeightProof) -> Tuple[List[bytes], bytes, bytes]:
+def vars_to_bytes(summaries: list[SubEpochSummary], weight_proof: WeightProof) -> tuple[list[bytes], bytes, bytes]:
     wp_recent_chain_bytes = bytes(RecentChainData(weight_proof.recent_chain_data))
     wp_segment_bytes = bytes(SubEpochSegments(weight_proof.sub_epoch_segments))
     summary_bytes = []
@@ -1514,7 +1515,7 @@ def vars_to_bytes(summaries: List[SubEpochSummary], weight_proof: WeightProof) -
     return summary_bytes, wp_segment_bytes, wp_recent_chain_bytes
 
 
-def summaries_from_bytes(summaries_bytes: List[bytes]) -> List[SubEpochSummary]:
+def summaries_from_bytes(summaries_bytes: list[bytes]) -> list[SubEpochSummary]:
     summaries = []
     for summary in summaries_bytes:
         summaries.append(SubEpochSummary.from_bytes(summary))
@@ -1522,8 +1523,8 @@ def summaries_from_bytes(summaries_bytes: List[bytes]) -> List[SubEpochSummary]:
 
 
 def _get_last_ses_hash(
-    constants: ConsensusConstants, recent_reward_chain: List[HeaderBlock]
-) -> Tuple[Optional[bytes32], uint32]:
+    constants: ConsensusConstants, recent_reward_chain: list[HeaderBlock]
+) -> tuple[Optional[bytes32], uint32]:
     for idx, block in enumerate(reversed(recent_reward_chain)):
         if (block.reward_chain_block.height % constants.SUB_EPOCH_BLOCKS) == 0:
             idx = len(recent_reward_chain) - 1 - idx  # reverse
@@ -1541,8 +1542,8 @@ def _get_last_ses_hash(
     return None, uint32(0)
 
 
-def _get_ses_idx(recent_reward_chain: List[HeaderBlock]) -> List[int]:
-    idxs: List[int] = []
+def _get_ses_idx(recent_reward_chain: list[HeaderBlock]) -> list[int]:
+    idxs: list[int] = []
     for idx, curr in enumerate(recent_reward_chain):
         if len(curr.finished_sub_slots) > 0:
             for slot in curr.finished_sub_slots:
@@ -1591,11 +1592,11 @@ def blue_boxed_end_of_slot(sub_slot: EndOfSubSlotBundle) -> bool:
 
 
 def validate_sub_epoch_sampling(
-    rng: random.Random, sub_epoch_weight_list: List[uint128], weight_proof: WeightProof
+    rng: random.Random, sub_epoch_weight_list: list[uint128], weight_proof: WeightProof
 ) -> bool:
     tip = weight_proof.recent_chain_data[-1]
     weight_to_check = _get_weights_for_sampling(rng, tip.weight, weight_proof.recent_chain_data)
-    sampled_sub_epochs: Dict[int, bool] = {}
+    sampled_sub_epochs: dict[int, bool] = {}
     for idx in range(1, len(sub_epoch_weight_list)):
         if _sample_sub_epoch(sub_epoch_weight_list[idx - 1], sub_epoch_weight_list[idx], weight_to_check):
             sampled_sub_epochs[idx - 1] = True
@@ -1613,9 +1614,9 @@ def validate_sub_epoch_sampling(
 
 
 def map_segments_by_sub_epoch(
-    sub_epoch_segments: List[SubEpochChallengeSegment],
-) -> Dict[int, List[SubEpochChallengeSegment]]:
-    segments: Dict[int, List[SubEpochChallengeSegment]] = {}
+    sub_epoch_segments: list[SubEpochChallengeSegment],
+) -> dict[int, list[SubEpochChallengeSegment]]:
+    segments: dict[int, list[SubEpochChallengeSegment]] = {}
     curr_sub_epoch_n = -1
     for idx, segment in enumerate(sub_epoch_segments):
         if curr_sub_epoch_n < segment.sub_epoch_n:
@@ -1627,7 +1628,7 @@ def map_segments_by_sub_epoch(
 
 def _validate_vdf_batch(
     constants: ConsensusConstants,
-    vdf_list: List[Tuple[bytes, bytes, bytes]],
+    vdf_list: list[tuple[bytes, bytes, bytes]],
     shutdown_file_path: Optional[pathlib.Path] = None,
 ) -> bool:
     for vdf_proof_bytes, class_group_bytes, info in vdf_list:
@@ -1650,11 +1651,11 @@ async def validate_weight_proof_inner(
     shutdown_file_name: str,
     num_processes: int,
     weight_proof: WeightProof,
-    summaries: List[SubEpochSummary],
-    sub_epoch_weight_list: List[uint128],
+    summaries: list[SubEpochSummary],
+    sub_epoch_weight_list: list[uint128],
     skip_segment_validation: bool,
     validate_from: int,
-) -> Tuple[bool, List[BlockRecord]]:
+) -> tuple[bool, list[BlockRecord]]:
     assert len(weight_proof.sub_epochs) > 0
     if len(weight_proof.sub_epochs) == 0:
         return False, []

chia/harvester/harvester.py

@@ -5,9 +5,10 @@ import concurrent
 import contextlib
 import dataclasses
 import logging
+from collections.abc import AsyncIterator
 from concurrent.futures.thread import ThreadPoolExecutor
 from pathlib import Path
-from typing import TYPE_CHECKING, Any, AsyncIterator, ClassVar, Dict, List, Optional, Tuple, cast
+from typing import TYPE_CHECKING, Any, ClassVar, Optional, cast
 
 from typing_extensions import Literal
 
@@ -71,7 +72,7 @@ class Harvester:
 
         return self._server
 
-    def __init__(self, root_path: Path, config: Dict[str, Any], constants: ConsensusConstants):
+    def __init__(self, root_path: Path, config: dict[str, Any], constants: ConsensusConstants):
         self.log = log
         self.root_path = root_path
         # TODO, remove checks below later after some versions / time
@@ -145,7 +146,7 @@ class Harvester:
 
             await self.plot_sync_sender.await_closed()
 
-    def get_connections(self, request_node_type: Optional[NodeType]) -> List[Dict[str, Any]]:
+    def get_connections(self, request_node_type: Optional[NodeType]) -> list[dict[str, Any]]:
         return default_get_connections(server=self.server, request_node_type=request_node_type)
 
     async def on_connect(self, connection: WSChiaConnection) -> None:
@@ -154,7 +155,7 @@ class Harvester:
     def _set_state_changed_callback(self, callback: StateChangedProtocol) -> None:
         self.state_changed_callback = callback
 
-    def state_changed(self, change: str, change_data: Optional[Dict[str, Any]] = None) -> None:
+    def state_changed(self, change: str, change_data: Optional[dict[str, Any]] = None) -> None:
         if self.state_changed_callback is not None:
             self.state_changed_callback(change, change_data)
 
@@ -181,9 +182,9 @@ class Harvester:
         asyncio.run_coroutine_threadsafe(self.plot_sync_sender.await_closed(), asyncio.get_running_loop())
         self.plot_manager.stop_refreshing()
 
-    def get_plots(self) -> Tuple[List[Dict[str, Any]], List[str], List[str]]:
+    def get_plots(self) -> tuple[list[dict[str, Any]], list[str], list[str]]:
         self.log.debug(f"get_plots prover items: {self.plot_manager.plot_count()}")
-        response_plots: List[Dict[str, Any]] = []
+        response_plots: list[dict[str, Any]] = []
         with self.plot_manager:
             for path, plot_info in self.plot_manager.plots.items():
                 prover = plot_info.prover
@@ -222,7 +223,7 @@ class Harvester:
         self.plot_manager.trigger_refresh()
         return True
 
-    async def get_plot_directories(self) -> List[str]:
+    async def get_plot_directories(self) -> list[str]:
         return get_plot_directories(self.root_path)
 
     async def remove_plot_directory(self, str_path: str) -> bool:
@@ -230,7 +231,7 @@ class Harvester:
         self.plot_manager.trigger_refresh()
         return True
 
-    async def get_harvester_config(self) -> Dict[str, Any]:
+    async def get_harvester_config(self) -> dict[str, Any]:
         return get_harvester_config(self.root_path)
 
     async def update_harvester_config(