File size: 2,221 Bytes
d7ecc62 230508d d7ecc62 87cdae5 4c4517f d7ecc62 f9aaa57 d7ecc62 230508d d7ecc62 87cdae5 4c4517f d7ecc62 f9aaa57 d7ecc62 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 | """ml-chess-engine-rs: High-performance chess engine for ML training pipelines.
All chess state management, move generation, and game simulation is handled
by the Rust engine. This package provides Python bindings via PyO3.
"""
from chess_engine._engine import (
# Core game generation
generate_training_batch,
generate_random_games,
generate_clm_batch,
generate_checkmate_games,
generate_checkmate_training_batch,
# Diagnostic sets
generate_diagnostic_sets_py as generate_diagnostic_sets,
# Edge case statistics
compute_edge_stats_per_ply_py as compute_edge_stats_per_ply,
compute_edge_stats_per_game_py as compute_edge_stats_per_game,
edge_case_bits,
# Game validation & analysis
compute_legal_move_masks,
compute_legal_token_masks,
compute_legal_token_masks_sparse,
extract_board_states,
validate_games_py as validate_games,
# PGN parsing
parse_pgn_file,
pgn_to_tokens,
parse_pgn_enriched,
count_pgn_games_in_date_range,
parse_pgn_sampled,
# UCI parsing
parse_uci_file,
uci_to_tokens,
pgn_to_uci,
# Engine self-play generation
generate_engine_games_py as generate_engine_games,
# Vocabulary
export_move_vocabulary,
# Interactive game state (for RL)
PyGameState,
PyBatchRLEnv,
# Accuracy ceiling
compute_accuracy_ceiling_py as compute_accuracy_ceiling,
# Utilities
hello,
)
__all__ = [
"generate_training_batch",
"generate_random_games",
"generate_clm_batch",
"generate_checkmate_games",
"generate_checkmate_training_batch",
"generate_diagnostic_sets",
"compute_edge_stats_per_ply",
"compute_edge_stats_per_game",
"edge_case_bits",
"compute_legal_move_masks",
"compute_legal_token_masks",
"compute_legal_token_masks_sparse",
"extract_board_states",
"validate_games",
"parse_pgn_file",
"pgn_to_tokens",
"parse_pgn_enriched",
"count_pgn_games_in_date_range",
"parse_pgn_sampled",
"parse_uci_file",
"uci_to_tokens",
"pgn_to_uci",
"generate_engine_games",
"export_move_vocabulary",
"PyGameState",
"PyBatchRLEnv",
"compute_accuracy_ceiling",
"hello",
]
|