LICENSE
README.md
pyproject.toml
requirements.txt
src/flash_ansr/__init__.py
src/flash_ansr/__main__.py
src/flash_ansr/flash_ansr.py
src/flash_ansr/refine.py
src/flash_ansr/results.py
src/flash_ansr.egg-info/PKG-INFO
src/flash_ansr.egg-info/SOURCES.txt
src/flash_ansr.egg-info/dependency_links.txt
src/flash_ansr.egg-info/entry_points.txt
src/flash_ansr.egg-info/requires.txt
src/flash_ansr.egg-info/top_level.txt
src/flash_ansr/baselines/__init__.py
src/flash_ansr/baselines/brute_force_model.py
src/flash_ansr/baselines/skeleton_pool_model.py
src/flash_ansr/benchmarks/__init__.py
src/flash_ansr/benchmarks/fastsrb.py
src/flash_ansr/compat/__init__.py
src/flash_ansr/compat/convert_data.py
src/flash_ansr/compat/evaluation_nesymres.py
src/flash_ansr/compat/evaluation_pysr.py
src/flash_ansr/compat/nesymres.py
src/flash_ansr/data/__init__.py
src/flash_ansr/data/collate.py
src/flash_ansr/data/data.py
src/flash_ansr/data/streaming.py
src/flash_ansr/decoding/mcts.py
src/flash_ansr/eval/__init__.py
src/flash_ansr/eval/core.py
src/flash_ansr/eval/data_sources.py
src/flash_ansr/eval/engine.py
src/flash_ansr/eval/evaluation.py
src/flash_ansr/eval/evaluation_fastsrb.py
src/flash_ansr/eval/model_adapters.py
src/flash_ansr/eval/result_store.py
src/flash_ansr/eval/run_config.py
src/flash_ansr/eval/sample_metadata.py
src/flash_ansr/eval/metrics/__init__.py
src/flash_ansr/eval/metrics/bootstrap.py
src/flash_ansr/eval/metrics/token_prediction.py
src/flash_ansr/eval/metrics/zss.py
src/flash_ansr/expressions/__init__.py
src/flash_ansr/expressions/compilation.py
src/flash_ansr/expressions/distributions.py
src/flash_ansr/expressions/holdout.py
src/flash_ansr/expressions/normalization.py
src/flash_ansr/expressions/prior_factory.py
src/flash_ansr/expressions/skeleton_pool.py
src/flash_ansr/expressions/skeleton_sampling.py
src/flash_ansr/expressions/structure.py
src/flash_ansr/expressions/support_sampling.py
src/flash_ansr/expressions/token_ops.py
src/flash_ansr/generation/__init__.py
src/flash_ansr/generation/beam.py
src/flash_ansr/generation/mcts.py
src/flash_ansr/model/__init__.py
src/flash_ansr/model/factory.py
src/flash_ansr/model/flash_ansr_model.py
src/flash_ansr/model/manage.py
src/flash_ansr/model/pre_encoder.py
src/flash_ansr/model/tokenizer.py
src/flash_ansr/model/common/__init__.py
src/flash_ansr/model/common/components.py
src/flash_ansr/model/decoders/__init__.py
src/flash_ansr/model/decoders/components.py
src/flash_ansr/model/decoders/transformer.py
src/flash_ansr/model/encoders/__init__.py
src/flash_ansr/model/encoders/base.py
src/flash_ansr/model/encoders/set_transformer.py
src/flash_ansr/preprocessing/__init__.py
src/flash_ansr/preprocessing/feature_extractor.py
src/flash_ansr/preprocessing/pipeline.py
src/flash_ansr/preprocessing/prompt_serialization.py
src/flash_ansr/preprocessing/schemas.py
src/flash_ansr/train/__init__.py
src/flash_ansr/train/optimizers.py
src/flash_ansr/train/schedules.py
src/flash_ansr/train/train.py
src/flash_ansr/utils/__init__.py
src/flash_ansr/utils/config_io.py
src/flash_ansr/utils/generation.py
src/flash_ansr/utils/numeric.py
src/flash_ansr/utils/paths.py
src/flash_ansr/utils/tensor_ops.py
tests/test_data.py
tests/test_decoding.py
tests/test_flash_ansr_prompt.py
tests/test_inference.py
tests/test_numeric.py
tests/test_preprocess_features.py
tests/test_refine.py
tests/test_results_serialization.py