popov/neuron-tai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test descriptions

Browse Source
This commit is contained in:
Dobromir Popov
2026-07-11 22:25:30 +03:00
parent 7d259d7c9b
commit 7cf8d9bcf3
43 changed files with 876 additions and 265 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test_safetensors_selection.py
View File

@@ -34,6 +34,7 @@ def _write_snapshot(tmp_path, *, config=None):
def test_selects_only_weight_shards_for_middle_layer_range(tmp_path):
"Selects only weight shards for middle layer range\n\nTags: general"
_write_snapshot(tmp_path)
files = select_safetensors_files_for_layers(tmp_path, 2, 3)
@@ -49,6 +50,7 @@ def test_selects_only_weight_shards_for_middle_layer_range(tmp_path):
def test_head_range_includes_embeddings(tmp_path):
"Head range includes embeddings\n\nTags: general"
_write_snapshot(tmp_path)
files = select_safetensors_files_for_layers(tmp_path, 0, 0)
@@ -58,6 +60,7 @@ def test_head_range_includes_embeddings(tmp_path):
def test_tail_range_includes_norm_and_lm_head_from_inferred_layer_count(tmp_path):
"Tail range includes norm and lm head from inferred layer count\n\nTags: general"
_write_snapshot(tmp_path, config={"text_config": {"num_hidden_layers": 5}})
files = select_safetensors_files_for_layers(tmp_path, 4, 4)
@@ -68,6 +71,7 @@ def test_tail_range_includes_norm_and_lm_head_from_inferred_layer_count(tmp_path
def test_tail_files_are_not_selected_without_total_layer_count(tmp_path):
"Tail files are not selected without total layer count\n\nTags: general"
_write_snapshot(tmp_path, config={"architectures": ["UnknownForTest"]})
files = select_safetensors_files_for_layers(tmp_path, 4, 4)
@@ -77,6 +81,7 @@ def test_tail_files_are_not_selected_without_total_layer_count(tmp_path):
def test_rejects_unsafe_weight_map_paths(tmp_path):
"Rejects unsafe weight map paths\n\nTags: general"
(tmp_path / "model.safetensors.index.json").write_text(
json.dumps({"weight_map": {"model.layers.0.weight": "../escape.safetensors"}}),
encoding="utf-8",