clean: remove pytest cache and egg-info from HF

.pytest_cache/.gitignore

@@ -1,2 +0,0 @@
-# Created by pytest automatically.
-*

.pytest_cache/CACHEDIR.TAG

@@ -1,4 +0,0 @@
-Signature: 8a477f597d28d172789f06886806bc55
-# This file is a cache directory tag created by pytest.
-# For information about cache directory tags, see:
-#	https://bford.info/cachedir/spec.html

.pytest_cache/README.md

@@ -1,8 +0,0 @@
-# pytest cache directory #
-
-This directory contains data from the pytest's cache plugin,
-which provides the `--lf` and `--ff` options, as well as the `cache` fixture.
-
-**Do not** commit this to version control.
-
-See [the docs](https://docs.pytest.org/en/stable/how-to/cache.html) for more information.

.pytest_cache/v/cache/lastfailed

@@ -1 +0,0 @@
-{}

.pytest_cache/v/cache/nodeids

@@ -1,82 +0,0 @@
-[
-  "tests/test_acceptance.py::TestAcceptanceRule::test_all_accepted",
-  "tests/test_acceptance.py::TestAcceptanceRule::test_batch_independence",
-  "tests/test_acceptance.py::TestAcceptanceRule::test_bonus_token_exists",
-  "tests/test_acceptance.py::TestAcceptanceRule::test_empty_block",
-  "tests/test_acceptance.py::TestAcceptanceRule::test_first_rejected",
-  "tests/test_acceptance.py::TestAcceptanceRule::test_partial_acceptance",
-  "tests/test_acceptance.py::TestExpectedAcceptLength::test_completely_wrong",
-  "tests/test_acceptance.py::TestExpectedAcceptLength::test_perfect_match",
-  "tests/test_backbone.py::TestDFlashAttention::test_forward_shape",
-  "tests/test_backbone.py::TestDFlashAttention::test_gqa_forward",
-  "tests/test_backbone.py::TestDFlashAttention::test_gradient_flow",
-  "tests/test_backbone.py::TestDFlashAttention::test_with_mask",
-  "tests/test_backbone.py::TestDFlashBackbone::test_empty_draft",
-  "tests/test_backbone.py::TestDFlashBackbone::test_forward_shape",
-  "tests/test_backbone.py::TestDFlashBackbone::test_gradient_flow",
-  "tests/test_backbone.py::TestDFlashBackbone::test_many_layers",
-  "tests/test_backbone.py::TestDFlashBackbone::test_output_hidden_states",
-  "tests/test_backbone.py::TestDFlashBackbone::test_single_draft_token",
-  "tests/test_backbone.py::TestDFlashDecoderLayer::test_all_activations",
-  "tests/test_backbone.py::TestDFlashDecoderLayer::test_forward_shape",
-  "tests/test_backbone.py::TestDFlashDecoderLayer::test_residual_connection",
-  "tests/test_backbone.py::TestDSparkAttentionMask::test_context_attention",
-  "tests/test_backbone.py::TestDSparkAttentionMask::test_cross_block_no_attention",
-  "tests/test_backbone.py::TestDSparkAttentionMask::test_intra_block_attention",
-  "tests/test_backbone.py::TestDSparkAttentionMask::test_mask_shape",
-  "tests/test_markov_head.py::TestGatedMarkovHead::test_forward_shape",
-  "tests/test_markov_head.py::TestGatedMarkovHead::test_init",
-  "tests/test_markov_head.py::TestRNNHead::test_apply_block_logits",
-  "tests/test_markov_head.py::TestRNNHead::test_empty_block",
-  "tests/test_markov_head.py::TestRNNHead::test_init",
-  "tests/test_markov_head.py::TestRNNHead::test_step",
-  "tests/test_markov_head.py::TestVanillaMarkov::test_apply_block_logits",
-  "tests/test_markov_head.py::TestVanillaMarkov::test_apply_step_logits",
-  "tests/test_markov_head.py::TestVanillaMarkov::test_build_gated",
-  "tests/test_markov_head.py::TestVanillaMarkov::test_build_vanilla",
-  "tests/test_markov_head.py::TestVanillaMarkov::test_compute_step_bias",
-  "tests/test_markov_head.py::TestVanillaMarkov::test_get_prev_embeddings",
-  "tests/test_markov_head.py::TestVanillaMarkov::test_init",
-  "tests/test_markov_head.py::TestVanillaMarkov::test_sample_block_tokens_empty",
-  "tests/test_markov_head.py::TestVanillaMarkov::test_sample_block_tokens_greedy",
-  "tests/test_sampling.py::TestSampling::test_gather_token_probs",
-  "tests/test_sampling.py::TestSampling::test_logits_to_probs_greedy",
-  "tests/test_sampling.py::TestSampling::test_logits_to_probs_temperature",
-  "tests/test_sampling.py::TestSampling::test_sample_residual",
-  "tests/test_sampling.py::TestSampling::test_sample_residual_identical",
-  "tests/test_sampling.py::TestSampling::test_sample_tokens_2d",
-  "tests/test_sampling.py::TestSampling::test_sample_tokens_greedy",
-  "tests/test_sampling.py::TestSampling::test_sample_tokens_temperature",
-  "tests/test_scheduler.py::TestHardwareAwareScheduler::test_all_zero_confidence",
-  "tests/test_scheduler.py::TestHardwareAwareScheduler::test_monotonic_selection",
-  "tests/test_scheduler.py::TestHardwareAwareScheduler::test_multiple_requests",
-  "tests/test_scheduler.py::TestHardwareAwareScheduler::test_no_future_leakage",
-  "tests/test_scheduler.py::TestHardwareAwareScheduler::test_per_request_validity",
-  "tests/test_scheduler.py::TestHardwareAwareScheduler::test_single_request",
-  "tests/test_scheduler.py::TestStaticScheduler::test_fixed_length",
-  "tests/test_scheduler.py::TestStaticScheduler::test_fixed_length_zero",
-  "tests/test_scheduler.py::TestStaticScheduler::test_static_threshold",
-  "tests/test_scheduler.py::TestStaticScheduler::test_static_threshold_all_below",
-  "tests/test_scheduler.py::TestThroughputProfile::test_clamp_high",
-  "tests/test_scheduler.py::TestThroughputProfile::test_clamp_low",
-  "tests/test_scheduler.py::TestThroughputProfile::test_exact_lookup",
-  "tests/test_scheduler.py::TestThroughputProfile::test_interpolation",
-  "tests/test_shapes.py::TestConfidenceHeadShapes::test_accept_rate_shape",
-  "tests/test_shapes.py::TestConfidenceHeadShapes::test_confidence_head",
-  "tests/test_shapes.py::TestIntegration::test_draft_verify_cycle",
-  "tests/test_shapes.py::TestIntegration::test_train_step_shape",
-  "tests/test_shapes.py::TestLossShapes::test_loss_decay_weights",
-  "tests/test_shapes.py::TestLossShapes::test_loss_shapes",
-  "tests/test_shapes.py::TestLossShapes::test_loss_with_all_terms",
-  "tests/test_shapes.py::TestModelShapes::test_forward_no_confidence",
-  "tests/test_shapes.py::TestModelShapes::test_forward_shapes",
-  "tests/test_shapes.py::TestModelShapes::test_sample_block_shapes",
-  "tests/test_sts.py::TestExpectedCalibrationError::test_miscalibrated",
-  "tests/test_sts.py::TestExpectedCalibrationError::test_perfectly_calibrated",
-  "tests/test_sts.py::TestExpectedCalibrationError::test_shape",
-  "tests/test_sts.py::TestSTSCalibrator::test_fit_transform",
-  "tests/test_sts.py::TestSTSCalibrator::test_shape_preserved",
-  "tests/test_sts.py::TestSTSCalibrator::test_transform_unfitted",
-  "tests/test_sts.py::TestSequentialTemperatureScaling::test_sts_monotonic",
-  "tests/test_sts.py::TestSequentialTemperatureScaling::test_sts_returns_temperatures"
-]

src/uraionspec.egg-info/PKG-INFO

@@ -1,22 +0,0 @@
-Metadata-Version: 2.4
-Name: uraionspec
-Version: 0.1.0
-Summary: Faithful DSpark-style speculative decoding implementation by Uraion Labs
-Author-email: Uraion Labs <uraionlabs@gmail.com>
-License: MIT
-Requires-Python: >=3.10
-Description-Content-Type: text/markdown
-Requires-Dist: torch>=2.1.0
-Requires-Dist: transformers>=4.38.0
-Requires-Dist: datasets>=2.14.0
-Requires-Dist: accelerate>=0.25.0
-Requires-Dist: numpy>=1.24.0
-Requires-Dist: tqdm>=4.64.0
-Requires-Dist: sentencepiece>=0.1.99
-Requires-Dist: protobuf>=3.20
-Provides-Extra: dev
-Requires-Dist: pytest>=7.0; extra == "dev"
-Requires-Dist: ruff>=0.1.0; extra == "dev"
-Requires-Dist: pytest-cov>=4.1.0; extra == "dev"
-Provides-Extra: eval
-Requires-Dist: vllm>=0.4.0; extra == "eval"

src/uraionspec.egg-info/SOURCES.txt

@@ -1,35 +0,0 @@
-pyproject.toml
-src/uraionspec/__init__.py
-src/uraionspec.egg-info/PKG-INFO
-src/uraionspec.egg-info/SOURCES.txt
-src/uraionspec.egg-info/dependency_links.txt
-src/uraionspec.egg-info/requires.txt
-src/uraionspec.egg-info/top_level.txt
-src/uraionspec/calibration/__init__.py
-src/uraionspec/calibration/sts.py
-src/uraionspec/decoding/__init__.py
-src/uraionspec/decoding/acceptance.py
-src/uraionspec/decoding/scheduler.py
-src/uraionspec/decoding/speculative.py
-src/uraionspec/evaluation/__init__.py
-src/uraionspec/evaluation/benchmark_latency.py
-src/uraionspec/evaluation/eval_acceptance.py
-src/uraionspec/models/__init__.py
-src/uraionspec/models/confidence_head.py
-src/uraionspec/models/draft_model.py
-src/uraionspec/models/markov_head.py
-src/uraionspec/models/rnn_head.py
-src/uraionspec/training/__init__.py
-src/uraionspec/training/cache_targets.py
-src/uraionspec/training/dataset.py
-src/uraionspec/training/losses.py
-src/uraionspec/training/train_drafter.py
-src/uraionspec/utils/__init__.py
-src/uraionspec/utils/hf.py
-src/uraionspec/utils/logging.py
-src/uraionspec/utils/seed.py
-tests/test_acceptance.py
-tests/test_markov_head.py
-tests/test_scheduler.py
-tests/test_shapes.py
-tests/test_sts.py

src/uraionspec.egg-info/dependency_links.txt

@@ -1 +0,0 @@
-

src/uraionspec.egg-info/requires.txt

@@ -1,16 +0,0 @@
-torch>=2.1.0
-transformers>=4.38.0
-datasets>=2.14.0
-accelerate>=0.25.0
-numpy>=1.24.0
-tqdm>=4.64.0
-sentencepiece>=0.1.99
-protobuf>=3.20
-
-[dev]
-pytest>=7.0
-ruff>=0.1.0
-pytest-cov>=4.1.0
-
-[eval]
-vllm>=0.4.0

src/uraionspec.egg-info/top_level.txt

@@ -1 +0,0 @@
-uraionspec