Add files using upload-large-folder tool
Browse files- LTA_openwebtext_dualt/logs/owt_bert_absrope_time4_C1_to_1024_mask1_sameT_gumbel_sde_watch/infer_lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525_step_0020000.log +136 -0
- LTA_openwebtext_dualt/logs/owt_bert_absrope_time4_C1_to_1024_mask1_sameT_gumbel_sde_watch/infer_lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525_step_0050000.log +136 -0
- LTA_openwebtext_dualt/logs/owt_bert_absrope_time4_C1_to_1024_mask1_sameT_gumbel_sde_watch/infer_lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525_step_0060000.log +136 -0
- LTA_openwebtext_dualt/logs/owt_bert_absrope_time4_C1_to_1024_mask1_sameT_gumbel_sde_watch/processed_lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525_steps128_c1_1024_gumbel_t1p45_n128.txt +7 -0
- LTA_openwebtext_dualt/logs/owt_t5elf_absrope_time4_len1025_C1_to_1024_prebos_mask1_sameT_dualline_watch/infer_lta_owt_t5elf_absrope_time4_dirichlet_len1025_C1_to_1024_prebos_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260526_step_0010000.log +199 -0
- LTA_openwebtext_dualt/logs/owt_t5elf_absrope_time4_len1025_C1_to_1024_prebos_mask1_sameT_dualline_watch/processed_lta_owt_t5elf_absrope_time4_dirichlet_len1025_C1_to_1024_prebos_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260526_steps128_c1_1024_gumbel_t1p45_n128.txt +1 -0
- LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/array_constructors.py +137 -0
- LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/array_like.py +41 -0
- LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/arrayterator.py +27 -0
- LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/comparisons.py +301 -0
- LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/flatiter.py +16 -0
- LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/fromnumeric.py +260 -0
- LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/index_tricks.py +64 -0
- LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/literal.py +47 -0
- LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/modules.py +42 -0
- LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/multiarray.py +76 -0
- LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/ndarray_conversion.py +94 -0
- LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/scalars.py +248 -0
- LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/ufunclike.py +46 -0
- LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/warnings_and_errors.py +6 -0
LTA_openwebtext_dualt/logs/owt_bert_absrope_time4_C1_to_1024_mask1_sameT_gumbel_sde_watch/infer_lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525_step_0020000.log
ADDED
|
@@ -0,0 +1,136 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[watch-gumbel] 2026-05-26_02:03:21 infer runs/lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525/step_0020000.pt -> docs/lta_samples/metrics_20260525/owt_bert_absrope_time4_C1_to_1024_mask1_sameT_sde_gumbel_topp0p95_tau1p0_to_0p2_blend_c1_1024_n128/lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525/step_0020000
|
| 2 |
+
[load] runs/lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525/step_0020000.pt
|
| 3 |
+
[ckpt] step=20000
|
| 4 |
+
[sde] generated 2/128
|
| 5 |
+
[sde] generated 4/128
|
| 6 |
+
[sde] generated 6/128
|
| 7 |
+
[sde] generated 8/128
|
| 8 |
+
[sde] generated 10/128
|
| 9 |
+
[sde] generated 12/128
|
| 10 |
+
[sde] generated 14/128
|
| 11 |
+
[sde] generated 16/128
|
| 12 |
+
[sde] generated 18/128
|
| 13 |
+
[sde] generated 20/128
|
| 14 |
+
[sde] generated 22/128
|
| 15 |
+
[sde] generated 24/128
|
| 16 |
+
[sde] generated 26/128
|
| 17 |
+
[sde] generated 28/128
|
| 18 |
+
[sde] generated 30/128
|
| 19 |
+
[sde] generated 32/128
|
| 20 |
+
[sde] generated 34/128
|
| 21 |
+
[sde] generated 36/128
|
| 22 |
+
[sde] generated 38/128
|
| 23 |
+
[sde] generated 40/128
|
| 24 |
+
[sde] generated 42/128
|
| 25 |
+
[sde] generated 44/128
|
| 26 |
+
[sde] generated 46/128
|
| 27 |
+
[sde] generated 48/128
|
| 28 |
+
[sde] generated 50/128
|
| 29 |
+
[sde] generated 52/128
|
| 30 |
+
[sde] generated 54/128
|
| 31 |
+
[sde] generated 56/128
|
| 32 |
+
[sde] generated 58/128
|
| 33 |
+
[sde] generated 60/128
|
| 34 |
+
[sde] generated 62/128
|
| 35 |
+
[sde] generated 64/128
|
| 36 |
+
[sde] generated 66/128
|
| 37 |
+
[sde] generated 68/128
|
| 38 |
+
[sde] generated 70/128
|
| 39 |
+
[sde] generated 72/128
|
| 40 |
+
[sde] generated 74/128
|
| 41 |
+
[sde] generated 76/128
|
| 42 |
+
[sde] generated 78/128
|
| 43 |
+
[sde] generated 80/128
|
| 44 |
+
[sde] generated 82/128
|
| 45 |
+
[sde] generated 84/128
|
| 46 |
+
[sde] generated 86/128
|
| 47 |
+
[sde] generated 88/128
|
| 48 |
+
[sde] generated 90/128
|
| 49 |
+
[sde] generated 92/128
|
| 50 |
+
[sde] generated 94/128
|
| 51 |
+
[sde] generated 96/128
|
| 52 |
+
[sde] generated 98/128
|
| 53 |
+
[sde] generated 100/128
|
| 54 |
+
[sde] generated 102/128
|
| 55 |
+
[sde] generated 104/128
|
| 56 |
+
[sde] generated 106/128
|
| 57 |
+
[sde] generated 108/128
|
| 58 |
+
[sde] generated 110/128
|
| 59 |
+
[sde] generated 112/128
|
| 60 |
+
[sde] generated 114/128
|
| 61 |
+
[sde] generated 116/128
|
| 62 |
+
[sde] generated 118/128
|
| 63 |
+
[sde] generated 120/128
|
| 64 |
+
[sde] generated 122/128
|
| 65 |
+
[sde] generated 124/128
|
| 66 |
+
[sde] generated 126/128
|
| 67 |
+
[sde] generated 128/128
|
| 68 |
+
[score] loading scorer: /e2e-data/evad-tech-vla/wanghan58/models/flowtext_scorers/gpt2-large-standard
|
| 69 |
+
[summary] {
|
| 70 |
+
"type": "summary",
|
| 71 |
+
"checkpoint": "runs/lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525/step_0020000.pt",
|
| 72 |
+
"step": 20000,
|
| 73 |
+
"decode": {
|
| 74 |
+
"decode_rule": "dirichlet_resample_sde",
|
| 75 |
+
"steps": 128,
|
| 76 |
+
"model_t_mode": "support_t",
|
| 77 |
+
"mean_mode": "endpoint_only",
|
| 78 |
+
"anchor_gamma": 1.0,
|
| 79 |
+
"endpoint_floor": 0.0,
|
| 80 |
+
"concentration_min": 1.0,
|
| 81 |
+
"concentration_max": 1024.0,
|
| 82 |
+
"endpoint_temp": 1.45,
|
| 83 |
+
"endpoint_temp_start": null,
|
| 84 |
+
"endpoint_temp_end": null,
|
| 85 |
+
"endpoint_projection": "gumbel_softmax",
|
| 86 |
+
"endpoint_top_k": 0,
|
| 87 |
+
"endpoint_top_p": 0.95,
|
| 88 |
+
"gumbel_tau_start": 1.0,
|
| 89 |
+
"gumbel_tau_end": 0.2,
|
| 90 |
+
"gumbel_noise_scale_start": 1.0,
|
| 91 |
+
"gumbel_noise_scale_end": 1.0,
|
| 92 |
+
"ban_special_tokens": false,
|
| 93 |
+
"banned_endpoint_ids": [],
|
| 94 |
+
"support_power": 1.0,
|
| 95 |
+
"semantic_power": 1.0,
|
| 96 |
+
"noise_init": "dirichlet",
|
| 97 |
+
"noise_sigma": -1.0,
|
| 98 |
+
"noise_dirichlet_concentration": 1.0,
|
| 99 |
+
"sde_resample": "dirichlet",
|
| 100 |
+
"logistic_normal_sigma_min": 0.18,
|
| 101 |
+
"logistic_normal_sigma_max": 3.0,
|
| 102 |
+
"logistic_normal_tau_min": 0.65,
|
| 103 |
+
"logistic_normal_tau_max": 1.0,
|
| 104 |
+
"final_from": "blend_0.5",
|
| 105 |
+
"n_samples": 128,
|
| 106 |
+
"seed": 20260524
|
| 107 |
+
},
|
| 108 |
+
"raw_genppl": {
|
| 109 |
+
"ppl": 1.3402534767040193,
|
| 110 |
+
"nll_per_token": 0.2928587577934152,
|
| 111 |
+
"tokens": 121854,
|
| 112 |
+
"kept_samples": 128,
|
| 113 |
+
"total_samples": 128,
|
| 114 |
+
"empty_rate": 0.0,
|
| 115 |
+
"skipped_samples": 0
|
| 116 |
+
},
|
| 117 |
+
"stripped_genppl": {
|
| 118 |
+
"ppl": 1.329592652013063,
|
| 119 |
+
"nll_per_token": 0.28487261863947294,
|
| 120 |
+
"tokens": 121693,
|
| 121 |
+
"kept_samples": 128,
|
| 122 |
+
"total_samples": 128,
|
| 123 |
+
"empty_rate": 0.0,
|
| 124 |
+
"skipped_samples": 0
|
| 125 |
+
},
|
| 126 |
+
"diversity": {
|
| 127 |
+
"sample_entropy": 0.24651868226929907,
|
| 128 |
+
"unique_tokens": 262,
|
| 129 |
+
"token_count": 131072,
|
| 130 |
+
"distinct_1": 0.0019989013671875,
|
| 131 |
+
"distinct_2": 0.006392045454545455,
|
| 132 |
+
"top_token_mass": 0.7519149780273438
|
| 133 |
+
}
|
| 134 |
+
}
|
| 135 |
+
[done] docs/lta_samples/metrics_20260525/owt_bert_absrope_time4_C1_to_1024_mask1_sameT_sde_gumbel_topp0p95_tau1p0_to_0p2_blend_c1_1024_n128/lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525/step_0020000/sde_steps128_samples128_scored.jsonl
|
| 136 |
+
[watch-gumbel] 2026-05-26_02:10:15 done step_0020000
|
LTA_openwebtext_dualt/logs/owt_bert_absrope_time4_C1_to_1024_mask1_sameT_gumbel_sde_watch/infer_lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525_step_0050000.log
ADDED
|
@@ -0,0 +1,136 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[watch-gumbel] 2026-05-26_07:41:32 infer runs/lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525/step_0050000.pt -> docs/lta_samples/metrics_20260525/owt_bert_absrope_time4_C1_to_1024_mask1_sameT_sde_gumbel_topp0p95_tau1p0_to_0p2_blend_c1_1024_n128/lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525/step_0050000
|
| 2 |
+
[load] runs/lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525/step_0050000.pt
|
| 3 |
+
[ckpt] step=50000
|
| 4 |
+
[sde] generated 2/128
|
| 5 |
+
[sde] generated 4/128
|
| 6 |
+
[sde] generated 6/128
|
| 7 |
+
[sde] generated 8/128
|
| 8 |
+
[sde] generated 10/128
|
| 9 |
+
[sde] generated 12/128
|
| 10 |
+
[sde] generated 14/128
|
| 11 |
+
[sde] generated 16/128
|
| 12 |
+
[sde] generated 18/128
|
| 13 |
+
[sde] generated 20/128
|
| 14 |
+
[sde] generated 22/128
|
| 15 |
+
[sde] generated 24/128
|
| 16 |
+
[sde] generated 26/128
|
| 17 |
+
[sde] generated 28/128
|
| 18 |
+
[sde] generated 30/128
|
| 19 |
+
[sde] generated 32/128
|
| 20 |
+
[sde] generated 34/128
|
| 21 |
+
[sde] generated 36/128
|
| 22 |
+
[sde] generated 38/128
|
| 23 |
+
[sde] generated 40/128
|
| 24 |
+
[sde] generated 42/128
|
| 25 |
+
[sde] generated 44/128
|
| 26 |
+
[sde] generated 46/128
|
| 27 |
+
[sde] generated 48/128
|
| 28 |
+
[sde] generated 50/128
|
| 29 |
+
[sde] generated 52/128
|
| 30 |
+
[sde] generated 54/128
|
| 31 |
+
[sde] generated 56/128
|
| 32 |
+
[sde] generated 58/128
|
| 33 |
+
[sde] generated 60/128
|
| 34 |
+
[sde] generated 62/128
|
| 35 |
+
[sde] generated 64/128
|
| 36 |
+
[sde] generated 66/128
|
| 37 |
+
[sde] generated 68/128
|
| 38 |
+
[sde] generated 70/128
|
| 39 |
+
[sde] generated 72/128
|
| 40 |
+
[sde] generated 74/128
|
| 41 |
+
[sde] generated 76/128
|
| 42 |
+
[sde] generated 78/128
|
| 43 |
+
[sde] generated 80/128
|
| 44 |
+
[sde] generated 82/128
|
| 45 |
+
[sde] generated 84/128
|
| 46 |
+
[sde] generated 86/128
|
| 47 |
+
[sde] generated 88/128
|
| 48 |
+
[sde] generated 90/128
|
| 49 |
+
[sde] generated 92/128
|
| 50 |
+
[sde] generated 94/128
|
| 51 |
+
[sde] generated 96/128
|
| 52 |
+
[sde] generated 98/128
|
| 53 |
+
[sde] generated 100/128
|
| 54 |
+
[sde] generated 102/128
|
| 55 |
+
[sde] generated 104/128
|
| 56 |
+
[sde] generated 106/128
|
| 57 |
+
[sde] generated 108/128
|
| 58 |
+
[sde] generated 110/128
|
| 59 |
+
[sde] generated 112/128
|
| 60 |
+
[sde] generated 114/128
|
| 61 |
+
[sde] generated 116/128
|
| 62 |
+
[sde] generated 118/128
|
| 63 |
+
[sde] generated 120/128
|
| 64 |
+
[sde] generated 122/128
|
| 65 |
+
[sde] generated 124/128
|
| 66 |
+
[sde] generated 126/128
|
| 67 |
+
[sde] generated 128/128
|
| 68 |
+
[score] loading scorer: /e2e-data/evad-tech-vla/wanghan58/models/flowtext_scorers/gpt2-large-standard
|
| 69 |
+
[summary] {
|
| 70 |
+
"type": "summary",
|
| 71 |
+
"checkpoint": "runs/lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525/step_0050000.pt",
|
| 72 |
+
"step": 50000,
|
| 73 |
+
"decode": {
|
| 74 |
+
"decode_rule": "dirichlet_resample_sde",
|
| 75 |
+
"steps": 128,
|
| 76 |
+
"model_t_mode": "support_t",
|
| 77 |
+
"mean_mode": "endpoint_only",
|
| 78 |
+
"anchor_gamma": 1.0,
|
| 79 |
+
"endpoint_floor": 0.0,
|
| 80 |
+
"concentration_min": 1.0,
|
| 81 |
+
"concentration_max": 1024.0,
|
| 82 |
+
"endpoint_temp": 1.45,
|
| 83 |
+
"endpoint_temp_start": null,
|
| 84 |
+
"endpoint_temp_end": null,
|
| 85 |
+
"endpoint_projection": "gumbel_softmax",
|
| 86 |
+
"endpoint_top_k": 0,
|
| 87 |
+
"endpoint_top_p": 0.95,
|
| 88 |
+
"gumbel_tau_start": 1.0,
|
| 89 |
+
"gumbel_tau_end": 0.2,
|
| 90 |
+
"gumbel_noise_scale_start": 1.0,
|
| 91 |
+
"gumbel_noise_scale_end": 1.0,
|
| 92 |
+
"ban_special_tokens": false,
|
| 93 |
+
"banned_endpoint_ids": [],
|
| 94 |
+
"support_power": 1.0,
|
| 95 |
+
"semantic_power": 1.0,
|
| 96 |
+
"noise_init": "dirichlet",
|
| 97 |
+
"noise_sigma": -1.0,
|
| 98 |
+
"noise_dirichlet_concentration": 1.0,
|
| 99 |
+
"sde_resample": "dirichlet",
|
| 100 |
+
"logistic_normal_sigma_min": 0.18,
|
| 101 |
+
"logistic_normal_sigma_max": 3.0,
|
| 102 |
+
"logistic_normal_tau_min": 0.65,
|
| 103 |
+
"logistic_normal_tau_max": 1.0,
|
| 104 |
+
"final_from": "blend_0.5",
|
| 105 |
+
"n_samples": 128,
|
| 106 |
+
"seed": 20260524
|
| 107 |
+
},
|
| 108 |
+
"raw_genppl": {
|
| 109 |
+
"ppl": 1.0464811518513861,
|
| 110 |
+
"nll_per_token": 0.045433252088485226,
|
| 111 |
+
"tokens": 130944,
|
| 112 |
+
"kept_samples": 128,
|
| 113 |
+
"total_samples": 128,
|
| 114 |
+
"empty_rate": 0.0,
|
| 115 |
+
"skipped_samples": 0
|
| 116 |
+
},
|
| 117 |
+
"stripped_genppl": {
|
| 118 |
+
"ppl": 1.0318298872771858,
|
| 119 |
+
"nll_per_token": 0.03133381556240585,
|
| 120 |
+
"tokens": 130817,
|
| 121 |
+
"kept_samples": 128,
|
| 122 |
+
"total_samples": 128,
|
| 123 |
+
"empty_rate": 0.0,
|
| 124 |
+
"skipped_samples": 0
|
| 125 |
+
},
|
| 126 |
+
"diversity": {
|
| 127 |
+
"sample_entropy": 0.007866103229201246,
|
| 128 |
+
"unique_tokens": 4,
|
| 129 |
+
"token_count": 131072,
|
| 130 |
+
"distinct_1": 3.0517578125e-05,
|
| 131 |
+
"distinct_2": 4.5821114369501466e-05,
|
| 132 |
+
"top_token_mass": 0.9990081787109375
|
| 133 |
+
}
|
| 134 |
+
}
|
| 135 |
+
[done] docs/lta_samples/metrics_20260525/owt_bert_absrope_time4_C1_to_1024_mask1_sameT_sde_gumbel_topp0p95_tau1p0_to_0p2_blend_c1_1024_n128/lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525/step_0050000/sde_steps128_samples128_scored.jsonl
|
| 136 |
+
[watch-gumbel] 2026-05-26_07:48:25 done step_0050000
|
LTA_openwebtext_dualt/logs/owt_bert_absrope_time4_C1_to_1024_mask1_sameT_gumbel_sde_watch/infer_lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525_step_0060000.log
ADDED
|
@@ -0,0 +1,136 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[watch-gumbel] 2026-05-26_09:34:28 infer runs/lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525/step_0060000.pt -> docs/lta_samples/metrics_20260525/owt_bert_absrope_time4_C1_to_1024_mask1_sameT_sde_gumbel_topp0p95_tau1p0_to_0p2_blend_c1_1024_n128/lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525/step_0060000
|
| 2 |
+
[load] runs/lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525/step_0060000.pt
|
| 3 |
+
[ckpt] step=60000
|
| 4 |
+
[sde] generated 2/128
|
| 5 |
+
[sde] generated 4/128
|
| 6 |
+
[sde] generated 6/128
|
| 7 |
+
[sde] generated 8/128
|
| 8 |
+
[sde] generated 10/128
|
| 9 |
+
[sde] generated 12/128
|
| 10 |
+
[sde] generated 14/128
|
| 11 |
+
[sde] generated 16/128
|
| 12 |
+
[sde] generated 18/128
|
| 13 |
+
[sde] generated 20/128
|
| 14 |
+
[sde] generated 22/128
|
| 15 |
+
[sde] generated 24/128
|
| 16 |
+
[sde] generated 26/128
|
| 17 |
+
[sde] generated 28/128
|
| 18 |
+
[sde] generated 30/128
|
| 19 |
+
[sde] generated 32/128
|
| 20 |
+
[sde] generated 34/128
|
| 21 |
+
[sde] generated 36/128
|
| 22 |
+
[sde] generated 38/128
|
| 23 |
+
[sde] generated 40/128
|
| 24 |
+
[sde] generated 42/128
|
| 25 |
+
[sde] generated 44/128
|
| 26 |
+
[sde] generated 46/128
|
| 27 |
+
[sde] generated 48/128
|
| 28 |
+
[sde] generated 50/128
|
| 29 |
+
[sde] generated 52/128
|
| 30 |
+
[sde] generated 54/128
|
| 31 |
+
[sde] generated 56/128
|
| 32 |
+
[sde] generated 58/128
|
| 33 |
+
[sde] generated 60/128
|
| 34 |
+
[sde] generated 62/128
|
| 35 |
+
[sde] generated 64/128
|
| 36 |
+
[sde] generated 66/128
|
| 37 |
+
[sde] generated 68/128
|
| 38 |
+
[sde] generated 70/128
|
| 39 |
+
[sde] generated 72/128
|
| 40 |
+
[sde] generated 74/128
|
| 41 |
+
[sde] generated 76/128
|
| 42 |
+
[sde] generated 78/128
|
| 43 |
+
[sde] generated 80/128
|
| 44 |
+
[sde] generated 82/128
|
| 45 |
+
[sde] generated 84/128
|
| 46 |
+
[sde] generated 86/128
|
| 47 |
+
[sde] generated 88/128
|
| 48 |
+
[sde] generated 90/128
|
| 49 |
+
[sde] generated 92/128
|
| 50 |
+
[sde] generated 94/128
|
| 51 |
+
[sde] generated 96/128
|
| 52 |
+
[sde] generated 98/128
|
| 53 |
+
[sde] generated 100/128
|
| 54 |
+
[sde] generated 102/128
|
| 55 |
+
[sde] generated 104/128
|
| 56 |
+
[sde] generated 106/128
|
| 57 |
+
[sde] generated 108/128
|
| 58 |
+
[sde] generated 110/128
|
| 59 |
+
[sde] generated 112/128
|
| 60 |
+
[sde] generated 114/128
|
| 61 |
+
[sde] generated 116/128
|
| 62 |
+
[sde] generated 118/128
|
| 63 |
+
[sde] generated 120/128
|
| 64 |
+
[sde] generated 122/128
|
| 65 |
+
[sde] generated 124/128
|
| 66 |
+
[sde] generated 126/128
|
| 67 |
+
[sde] generated 128/128
|
| 68 |
+
[score] loading scorer: /e2e-data/evad-tech-vla/wanghan58/models/flowtext_scorers/gpt2-large-standard
|
| 69 |
+
[summary] {
|
| 70 |
+
"type": "summary",
|
| 71 |
+
"checkpoint": "runs/lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525/step_0060000.pt",
|
| 72 |
+
"step": 60000,
|
| 73 |
+
"decode": {
|
| 74 |
+
"decode_rule": "dirichlet_resample_sde",
|
| 75 |
+
"steps": 128,
|
| 76 |
+
"model_t_mode": "support_t",
|
| 77 |
+
"mean_mode": "endpoint_only",
|
| 78 |
+
"anchor_gamma": 1.0,
|
| 79 |
+
"endpoint_floor": 0.0,
|
| 80 |
+
"concentration_min": 1.0,
|
| 81 |
+
"concentration_max": 1024.0,
|
| 82 |
+
"endpoint_temp": 1.45,
|
| 83 |
+
"endpoint_temp_start": null,
|
| 84 |
+
"endpoint_temp_end": null,
|
| 85 |
+
"endpoint_projection": "gumbel_softmax",
|
| 86 |
+
"endpoint_top_k": 0,
|
| 87 |
+
"endpoint_top_p": 0.95,
|
| 88 |
+
"gumbel_tau_start": 1.0,
|
| 89 |
+
"gumbel_tau_end": 0.2,
|
| 90 |
+
"gumbel_noise_scale_start": 1.0,
|
| 91 |
+
"gumbel_noise_scale_end": 1.0,
|
| 92 |
+
"ban_special_tokens": false,
|
| 93 |
+
"banned_endpoint_ids": [],
|
| 94 |
+
"support_power": 1.0,
|
| 95 |
+
"semantic_power": 1.0,
|
| 96 |
+
"noise_init": "dirichlet",
|
| 97 |
+
"noise_sigma": -1.0,
|
| 98 |
+
"noise_dirichlet_concentration": 1.0,
|
| 99 |
+
"sde_resample": "dirichlet",
|
| 100 |
+
"logistic_normal_sigma_min": 0.18,
|
| 101 |
+
"logistic_normal_sigma_max": 3.0,
|
| 102 |
+
"logistic_normal_tau_min": 0.65,
|
| 103 |
+
"logistic_normal_tau_max": 1.0,
|
| 104 |
+
"final_from": "blend_0.5",
|
| 105 |
+
"n_samples": 128,
|
| 106 |
+
"seed": 20260524
|
| 107 |
+
},
|
| 108 |
+
"raw_genppl": {
|
| 109 |
+
"ppl": 1.0477807062851836,
|
| 110 |
+
"nll_per_token": 0.04667431427519331,
|
| 111 |
+
"tokens": 126936,
|
| 112 |
+
"kept_samples": 128,
|
| 113 |
+
"total_samples": 128,
|
| 114 |
+
"empty_rate": 0.0,
|
| 115 |
+
"skipped_samples": 0
|
| 116 |
+
},
|
| 117 |
+
"stripped_genppl": {
|
| 118 |
+
"ppl": 1.0321466838054298,
|
| 119 |
+
"nll_per_token": 0.031640792429459864,
|
| 120 |
+
"tokens": 126797,
|
| 121 |
+
"kept_samples": 128,
|
| 122 |
+
"total_samples": 128,
|
| 123 |
+
"empty_rate": 0.0,
|
| 124 |
+
"skipped_samples": 0
|
| 125 |
+
},
|
| 126 |
+
"diversity": {
|
| 127 |
+
"sample_entropy": 0.007866103229201246,
|
| 128 |
+
"unique_tokens": 6,
|
| 129 |
+
"token_count": 131072,
|
| 130 |
+
"distinct_1": 4.57763671875e-05,
|
| 131 |
+
"distinct_2": 7.636852394916911e-05,
|
| 132 |
+
"top_token_mass": 0.9365692138671875
|
| 133 |
+
}
|
| 134 |
+
}
|
| 135 |
+
[done] docs/lta_samples/metrics_20260525/owt_bert_absrope_time4_C1_to_1024_mask1_sameT_sde_gumbel_topp0p95_tau1p0_to_0p2_blend_c1_1024_n128/lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525/step_0060000/sde_steps128_samples128_scored.jsonl
|
| 136 |
+
[watch-gumbel] 2026-05-26_09:41:20 done step_0060000
|
LTA_openwebtext_dualt/logs/owt_bert_absrope_time4_C1_to_1024_mask1_sameT_gumbel_sde_watch/processed_lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525_steps128_c1_1024_gumbel_t1p45_n128.txt
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
runs/lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525/step_0010000.pt
|
| 2 |
+
runs/lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525/step_0020000.pt
|
| 3 |
+
runs/lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525/step_0030000.pt
|
| 4 |
+
runs/lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525/step_0040000.pt
|
| 5 |
+
runs/lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525/step_0050000.pt
|
| 6 |
+
runs/lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525/step_0060000.pt
|
| 7 |
+
runs/lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260525/step_0070000.pt
|
LTA_openwebtext_dualt/logs/owt_t5elf_absrope_time4_len1025_C1_to_1024_prebos_mask1_sameT_dualline_watch/infer_lta_owt_t5elf_absrope_time4_dirichlet_len1025_C1_to_1024_prebos_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260526_step_0010000.log
ADDED
|
@@ -0,0 +1,199 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[watch-gumbel] 2026-05-26_13:35:23 infer runs/lta_owt_t5elf_absrope_time4_dirichlet_len1025_C1_to_1024_prebos_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260526/step_0010000.pt -> docs/lta_samples/metrics_20260526/owt_t5elf_absrope_time4_len1025_C1_to_1024_prebos_mask1_sameT_dualline_dirres_c1_1024_n128/lta_owt_t5elf_absrope_time4_dirichlet_len1025_C1_to_1024_prebos_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260526/step_0010000
|
| 2 |
+
[decode] max_len=1025 generated=1/128
|
| 3 |
+
[decode] max_len=1025 generated=2/128
|
| 4 |
+
[decode] max_len=1025 generated=3/128
|
| 5 |
+
[decode] max_len=1025 generated=4/128
|
| 6 |
+
[decode] max_len=1025 generated=5/128
|
| 7 |
+
[decode] max_len=1025 generated=6/128
|
| 8 |
+
[decode] max_len=1025 generated=7/128
|
| 9 |
+
[decode] max_len=1025 generated=8/128
|
| 10 |
+
[decode] max_len=1025 generated=9/128
|
| 11 |
+
[decode] max_len=1025 generated=10/128
|
| 12 |
+
[decode] max_len=1025 generated=11/128
|
| 13 |
+
[decode] max_len=1025 generated=12/128
|
| 14 |
+
[decode] max_len=1025 generated=13/128
|
| 15 |
+
[decode] max_len=1025 generated=14/128
|
| 16 |
+
[decode] max_len=1025 generated=15/128
|
| 17 |
+
[decode] max_len=1025 generated=16/128
|
| 18 |
+
[decode] max_len=1025 generated=17/128
|
| 19 |
+
[decode] max_len=1025 generated=18/128
|
| 20 |
+
[decode] max_len=1025 generated=19/128
|
| 21 |
+
[decode] max_len=1025 generated=20/128
|
| 22 |
+
[decode] max_len=1025 generated=21/128
|
| 23 |
+
[decode] max_len=1025 generated=22/128
|
| 24 |
+
[decode] max_len=1025 generated=23/128
|
| 25 |
+
[decode] max_len=1025 generated=24/128
|
| 26 |
+
[decode] max_len=1025 generated=25/128
|
| 27 |
+
[decode] max_len=1025 generated=26/128
|
| 28 |
+
[decode] max_len=1025 generated=27/128
|
| 29 |
+
[decode] max_len=1025 generated=28/128
|
| 30 |
+
[decode] max_len=1025 generated=29/128
|
| 31 |
+
[decode] max_len=1025 generated=30/128
|
| 32 |
+
[decode] max_len=1025 generated=31/128
|
| 33 |
+
[decode] max_len=1025 generated=32/128
|
| 34 |
+
[decode] max_len=1025 generated=33/128
|
| 35 |
+
[decode] max_len=1025 generated=34/128
|
| 36 |
+
[decode] max_len=1025 generated=35/128
|
| 37 |
+
[decode] max_len=1025 generated=36/128
|
| 38 |
+
[decode] max_len=1025 generated=37/128
|
| 39 |
+
[decode] max_len=1025 generated=38/128
|
| 40 |
+
[decode] max_len=1025 generated=39/128
|
| 41 |
+
[decode] max_len=1025 generated=40/128
|
| 42 |
+
[decode] max_len=1025 generated=41/128
|
| 43 |
+
[decode] max_len=1025 generated=42/128
|
| 44 |
+
[decode] max_len=1025 generated=43/128
|
| 45 |
+
[decode] max_len=1025 generated=44/128
|
| 46 |
+
[decode] max_len=1025 generated=45/128
|
| 47 |
+
[decode] max_len=1025 generated=46/128
|
| 48 |
+
[decode] max_len=1025 generated=47/128
|
| 49 |
+
[decode] max_len=1025 generated=48/128
|
| 50 |
+
[decode] max_len=1025 generated=49/128
|
| 51 |
+
[decode] max_len=1025 generated=50/128
|
| 52 |
+
[decode] max_len=1025 generated=51/128
|
| 53 |
+
[decode] max_len=1025 generated=52/128
|
| 54 |
+
[decode] max_len=1025 generated=53/128
|
| 55 |
+
[decode] max_len=1025 generated=54/128
|
| 56 |
+
[decode] max_len=1025 generated=55/128
|
| 57 |
+
[decode] max_len=1025 generated=56/128
|
| 58 |
+
[decode] max_len=1025 generated=57/128
|
| 59 |
+
[decode] max_len=1025 generated=58/128
|
| 60 |
+
[decode] max_len=1025 generated=59/128
|
| 61 |
+
[decode] max_len=1025 generated=60/128
|
| 62 |
+
[decode] max_len=1025 generated=61/128
|
| 63 |
+
[decode] max_len=1025 generated=62/128
|
| 64 |
+
[decode] max_len=1025 generated=63/128
|
| 65 |
+
[decode] max_len=1025 generated=64/128
|
| 66 |
+
[decode] max_len=1025 generated=65/128
|
| 67 |
+
[decode] max_len=1025 generated=66/128
|
| 68 |
+
[decode] max_len=1025 generated=67/128
|
| 69 |
+
[decode] max_len=1025 generated=68/128
|
| 70 |
+
[decode] max_len=1025 generated=69/128
|
| 71 |
+
[decode] max_len=1025 generated=70/128
|
| 72 |
+
[decode] max_len=1025 generated=71/128
|
| 73 |
+
[decode] max_len=1025 generated=72/128
|
| 74 |
+
[decode] max_len=1025 generated=73/128
|
| 75 |
+
[decode] max_len=1025 generated=74/128
|
| 76 |
+
[decode] max_len=1025 generated=75/128
|
| 77 |
+
[decode] max_len=1025 generated=76/128
|
| 78 |
+
[decode] max_len=1025 generated=77/128
|
| 79 |
+
[decode] max_len=1025 generated=78/128
|
| 80 |
+
[decode] max_len=1025 generated=79/128
|
| 81 |
+
[decode] max_len=1025 generated=80/128
|
| 82 |
+
[decode] max_len=1025 generated=81/128
|
| 83 |
+
[decode] max_len=1025 generated=82/128
|
| 84 |
+
[decode] max_len=1025 generated=83/128
|
| 85 |
+
[decode] max_len=1025 generated=84/128
|
| 86 |
+
[decode] max_len=1025 generated=85/128
|
| 87 |
+
[decode] max_len=1025 generated=86/128
|
| 88 |
+
[decode] max_len=1025 generated=87/128
|
| 89 |
+
[decode] max_len=1025 generated=88/128
|
| 90 |
+
[decode] max_len=1025 generated=89/128
|
| 91 |
+
[decode] max_len=1025 generated=90/128
|
| 92 |
+
[decode] max_len=1025 generated=91/128
|
| 93 |
+
[decode] max_len=1025 generated=92/128
|
| 94 |
+
[decode] max_len=1025 generated=93/128
|
| 95 |
+
[decode] max_len=1025 generated=94/128
|
| 96 |
+
[decode] max_len=1025 generated=95/128
|
| 97 |
+
[decode] max_len=1025 generated=96/128
|
| 98 |
+
[decode] max_len=1025 generated=97/128
|
| 99 |
+
[decode] max_len=1025 generated=98/128
|
| 100 |
+
[decode] max_len=1025 generated=99/128
|
| 101 |
+
[decode] max_len=1025 generated=100/128
|
| 102 |
+
[decode] max_len=1025 generated=101/128
|
| 103 |
+
[decode] max_len=1025 generated=102/128
|
| 104 |
+
[decode] max_len=1025 generated=103/128
|
| 105 |
+
[decode] max_len=1025 generated=104/128
|
| 106 |
+
[decode] max_len=1025 generated=105/128
|
| 107 |
+
[decode] max_len=1025 generated=106/128
|
| 108 |
+
[decode] max_len=1025 generated=107/128
|
| 109 |
+
[decode] max_len=1025 generated=108/128
|
| 110 |
+
[decode] max_len=1025 generated=109/128
|
| 111 |
+
[decode] max_len=1025 generated=110/128
|
| 112 |
+
[decode] max_len=1025 generated=111/128
|
| 113 |
+
[decode] max_len=1025 generated=112/128
|
| 114 |
+
[decode] max_len=1025 generated=113/128
|
| 115 |
+
[decode] max_len=1025 generated=114/128
|
| 116 |
+
[decode] max_len=1025 generated=115/128
|
| 117 |
+
[decode] max_len=1025 generated=116/128
|
| 118 |
+
[decode] max_len=1025 generated=117/128
|
| 119 |
+
[decode] max_len=1025 generated=118/128
|
| 120 |
+
[decode] max_len=1025 generated=119/128
|
| 121 |
+
[decode] max_len=1025 generated=120/128
|
| 122 |
+
[decode] max_len=1025 generated=121/128
|
| 123 |
+
[decode] max_len=1025 generated=122/128
|
| 124 |
+
[decode] max_len=1025 generated=123/128
|
| 125 |
+
[decode] max_len=1025 generated=124/128
|
| 126 |
+
[decode] max_len=1025 generated=125/128
|
| 127 |
+
[decode] max_len=1025 generated=126/128
|
| 128 |
+
[decode] max_len=1025 generated=127/128
|
| 129 |
+
[decode] max_len=1025 generated=128/128
|
| 130 |
+
[
|
| 131 |
+
{
|
| 132 |
+
"checkpoint": "runs/lta_owt_t5elf_absrope_time4_dirichlet_len1025_C1_to_1024_prebos_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260526/step_0010000.pt",
|
| 133 |
+
"ckpt_step": 10000,
|
| 134 |
+
"max_len": 1025,
|
| 135 |
+
"decode_rule": "dual_line_resample",
|
| 136 |
+
"support_power": 1.0,
|
| 137 |
+
"semantic_power": 1.5,
|
| 138 |
+
"steps": 128,
|
| 139 |
+
"c_min": 1.0,
|
| 140 |
+
"c_max": 1024.0,
|
| 141 |
+
"anchor_mode": "state",
|
| 142 |
+
"model_t_mode": "flow",
|
| 143 |
+
"time_schedule": "uniform",
|
| 144 |
+
"time_logit_mean": -1.5,
|
| 145 |
+
"time_logit_std": 0.8,
|
| 146 |
+
"time_power": 2.0,
|
| 147 |
+
"input_noise_scale": 0.0,
|
| 148 |
+
"input_noise_until": 1.0,
|
| 149 |
+
"input_noise_dirichlet_concentration": 1.0,
|
| 150 |
+
"endpoint_softening": "none",
|
| 151 |
+
"endpoint_soft_power": 2.0,
|
| 152 |
+
"endpoint_soft_min_conf": 0.0,
|
| 153 |
+
"endpoint_soft_max_conf": 1.0,
|
| 154 |
+
"soft_target_decode_mode": "off",
|
| 155 |
+
"soft_target_power": 1.0,
|
| 156 |
+
"soft_target_min_conf": 0.0,
|
| 157 |
+
"soft_target_max_conf": 1.0,
|
| 158 |
+
"soft_target_debias_start": 0.7,
|
| 159 |
+
"final_from": "blend",
|
| 160 |
+
"final_decode": "argmax",
|
| 161 |
+
"final_sample_temp": 1.0,
|
| 162 |
+
"final_top_k": 0,
|
| 163 |
+
"final_top_p": 1.0,
|
| 164 |
+
"commit_mode": "off",
|
| 165 |
+
"commit_conf_threshold": 0.0,
|
| 166 |
+
"commit_margin_threshold": 0.0,
|
| 167 |
+
"commit_start": 0.0,
|
| 168 |
+
"commit_min_ratio": 0.0,
|
| 169 |
+
"commit_max_ratio": 1.0,
|
| 170 |
+
"commit_power": 2.0,
|
| 171 |
+
"commit_freq_max_frac": 0.08,
|
| 172 |
+
"early_temp": 2.8,
|
| 173 |
+
"late_temp": 1.45,
|
| 174 |
+
"temp_end": 0.55,
|
| 175 |
+
"temp_power": 1.5,
|
| 176 |
+
"pos_extend": "repeat",
|
| 177 |
+
"fixed_first_token_id": null,
|
| 178 |
+
"fixed_first_token_text": "",
|
| 179 |
+
"fixed_first_initial_argmax": false,
|
| 180 |
+
"use_ema": false,
|
| 181 |
+
"n_samples": 128,
|
| 182 |
+
"sample_entropy": 1.5424398482114823,
|
| 183 |
+
"unique_tokens": 808,
|
| 184 |
+
"token_count": 131200,
|
| 185 |
+
"distinct_1": 0.006158536585365854,
|
| 186 |
+
"distinct_2": 0.04183197021484375,
|
| 187 |
+
"top_token_mass": 0.1704496951219512,
|
| 188 |
+
"texts_preview": [
|
| 189 |
+
"</s> , , 2016 ( e the year of the s- the part of the s, The volume of the quarter of the year, observed in the year of the s ., c,e the 2012 s ) The distribution of the total reduction of the quarter, the s- the figures of the quarter- the s- the s of the s . s,, the year, in the top- the fourth quarter of the total of the year of 2016,- the volume of the s,- the period, ,- the increase of the year- s , on the fourth quarter,- the the quarter of the total of the year of the year, in the s . The year of the 2015 , the year of the year of the period, the year of 2016,- the top quarter , 2013 . The increase of the s ,- the 2015, from the year of the quarter,- the year of 2016,- a large part of the 2015 ,- the- year , in the s, 2012 , as well as the cause of the s,- the year, from the fourth year, in the the--,, in- the, ,e,, e 2009,,ee, 2012,,,e, year , thee52 year ,,,eee 2015,,, ,,,e 2012 -,e thee part 2015-- 2012,,, the year year year,,, ,, ,,888 the- year year, - the year year,, , ,,,, , ,203888 the year year year the 2008,,, the year year year ,,,e year year the 2012,,, ,,, the year yeard the year year ,,, part , 2009,, , part, 2015,,, thee year , 2012, thee, year, 2012,,,, ,,, the year of, 2012 ,,, ,, ,,,e thee year 2008 2012,, , ,,,, thee year,, year ,,,s i...,,, ,,e the year- the- ,,e, the 2008 year,, ,, 2012 year,,, the year ,,, ,, ,- ,,, the year, year,, 2012 , ,,,, ,,,, the, 2008 , , ,-,, ,, , , the 2009 , the,e year, ,,,,,,,,,,, part,, 2008 2012 -,,- ,- ,,, ,-,, ,,, year, the the quarter year , -,,,,, year232, the 2012, , the year, year,,, ,, ,232, the year year year 2008,, year year ,,,,,e the year year the year, 2009,,, 2008 , year, the year- ,,, , the e quarter-,, year, the ,,, the, the year the 2006 ,- year, year-,, 2015, the 2012- year year -, - year the ,-,, , , the part, the ,,, -,,,- ,, ,,,, ,, , , the 2009e,,, the-, 2009, ,,,,--- the - period, year, ,a , , -,s 2009, from ,,, ,, , e isi. ., -,-,- 2012 2009,,,e part, the 2006, ,,, , , ,e, year -,, 2012e,- , ,, , ,,,, ,e ,, e,e,, , ,, e ,,, , ,- e, ,e ,,,eee the 2006,-,- - year,, 2009-, 2012- ,, ,,,e year,,?-- ,- - 2015,,,",
|
| 190 |
+
"</s> “<unk>, you <unk>,’ or , you have?,<unk>,’, you . ’’<unk>,’ you know it? . . ., . <unk>,’, you . . . . ., in a ,’ you know that, and you must forget it... <unk>,’ pym , you know it. ... you don’t put it, you with a , said “ , I have a , is it ?” “?<unk>,’, you , ? <unk> ” “<unk> ? What? you know you is be’ if <unk> wasn’t that . . . <unk> h,’ if you want , you could do with, I’d be with <unk>.’, you said , you get discussion of the experiences, and what you get that? <unk> , you may have that,’ you know, you get on a . . ., a?’ , you?, <unk>, . , you s<unk> if you get this report,’<unk> you <unk> <unk>’, you<unk> <unk><unk>. ..,’ you, , <unk>, you <unk> <unk><unk>’, you <unk>, <unk><unk><unk><unk><unk><unk> ?<unk><unk>?<unk><unk><unk> ? <unk>,<unk>’.,, you. ’, you<unk> <unk><unk>, you<unk>?<unk> <unk>,’ at you ’,.’, you,,<unk> <unk> <unk>. you’. know’ that you,<unk>’’ said’’e?<unk><unk> ’,, you’ you,’ ,<unk>’’’’’ you’, <unk>. , said, said’’<unk><unk> <unk> <unk> <unk>’ <unk><unk> <unk><unk><unk>,’ youd, ’, .<unk> ,’ you<unk>,<unk>, ?’ ,<unk>, ’’<unk>’ .. ’. you <unk><unk><unk>’’’,’’?’’,’’’172,,’’’199 , ’’,’d’196196196172’,d ’146’888,,’’196165196172,1/2,’ D’’’’.’7771/2-172, ’’<unk>172d,172-188,’ ’’ <unk>?<unk><unk>,<unk><unk>? <unk> <unk><unk> <unk>?<unk> ?<unk>186<unk>,,<unk> <unk>,<unk>’<unk>?<unk>? ?<unk><unk><unk> .’ ’’?d’186,211<unk>,<unk>,<unk> , ’, <unk>?<unk><unk><unk>? <unk> ?<unk> ?<unk><unk>, <unk>. 186.,<unk>., ,<unk>.777,,480’<unk> ,1721/2,1/2 .<unk> 186,,,150<unk>?196<unk>172232?,, ., ., .’172ACS-196172,180,<unk>s’ ... <unk> <unk>152<unk>1/2,430D ,430-D<unk>,<unk>,<unk>’<unk>,’’D’’187 -172172s1/2 .,. <unk>,<unk>,186<unk>333<unk>? ?.<unk><unk> ,186<unk> .’.’’’’<unk>,’’’’’<unk>’, , ,,96<unk>430<unk> , , <unk><unk> ,<unk> <unk> <unk>’,’’’ ’<unk>’ <unk>’’’’’’’’’ ’,<unk>’ ’<unk>’ ’’,<unk>,’,<unk> <unk>?<unk> <unk>’,,’’’<unk> ,.<unk><unk>., ?, <unk><unk> <unk><unk><unk>?, <unk> <unk>’<unk>, at , ,.<unk><unk> .. .<unk>.’ . , .’,,?, ?’<unk>, .’’ ’ ., ’?,,’<unk><unk>, .. <unk> .’,<unk><unk>,<unk>’,,?<unk>’ ’<unk>?<unk>?’’’’’<unk>?<unk><unk>’,’’’m? <unk> <unk>, <unk>, ,, ., .’, .’ ,,<unk>.? ,,",
|
| 191 |
+
"</s> <unk> SSD ' <unk> SSD ' ' ' <unk> SSD ' ' ' ' ' ' ' ' ' ' ' ' ' <unk> SSD ' ' ' <unk> SSD '' <unk> SSD ' ' <unk> SSD ' <unk> SSD ' ' ' <unk> SSD ' <unk> ' ' <unk> SSD ' ' ' ' ' ' ' ' ' <unk> SSD ' ' ' ' <unk> ' ' <unk> SSD ' ' <unk> SSD ' <unk> SSD ' <unk> <unk> <unk> SSD ' <unk> <unk> SSD ' <unk> SSD ' <unk> SSD <unk> SSD ' ' ' ' <unk> ' <unk> <unk> <unk> <unk> <unk> SSD <unk> SSD ' ' <unk> SSD ' ' <unk> SSD ' <unk> <unk> SSD <unk> SSD ' <unk> ' <unk> <unk> <unk> ' ' ' ' <unk> SSD ' ' <unk> <unk> <unk> 'tt;' SSD with00x0xxx0xx00x0xxxxxxxxxxxxx0xxxxxxxxxxxxxxxxxxxxx0xx0xxxxxxxxxxx0xx00xxxx0xxxxxxxxxxxxx0xxxxxxxxxxxxx0xxx000xxxxxxx0xxxxxxxxxxxxxxxxxxxxxx00xx0x0xxxx0x0xx0xx0xxxxxxxx000x0xx0xxx000xx00000xx0xx0xx0x00xxxxxx0xxxxx0x0xxxxx000x00xx0x0xx0x0xx0x0000xxx00x000xx00x00x0xx00000x000xxxxx0xxx000x0x0xx0x00xx0000x0xxx000x00000000000000x00000000000x0x00x000000000x0x00x0000x00xx00x000x000000;0;;;;;x00xx0000000x00000x00xxx00xxxx0x000x0x000000x0x0x0000xx000xx000000xxx0x0x0x0xxx000000x0x00x0xxx000x00000000xx00000000x0x000x00x000xxxxx000x0x000x00xx00x000x0x0000xxx00000xx0x0x00000x0000x0xxxx00xxxxxx000000x0x00x0x000xx0xxxx00xxxxxx00xxxx0xx0xxxx0x00xxxxxx00xxxxxxx00xxx0000xx0x0x00xx0xxxxxxxxxxxx0x00xxx;;;;;;;;;0;x0x0xx;x;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;",
|
| 192 |
+
"</s> The <unk> is a <unk>- g , a t <unk> is a -e , a sc <unk> / .,t know what is a dy -e , a , ad ,e , ad ,t getz what is exactly what is om, what, dv ( <unk>): I. what. . d , m -e , a , ,? dd , s d , ? om's ad , what's do o <unk> , daw, e, -e , l<unk> oc d) <unk>e <unk>a -e . a s s d , a i.e. dm, dt<unk> o- pn ,e ms- , a- dl gyzn , a s- <unk> ew -<unk>, arh pn-)<unk>in <unk> , <unk>, <unk>,<unk><unk> /o<unk> ( ()n-d-n, d- den - , d'g, ,'' ( a)<unk> 's,', ''. ''d', is,',''' (' -- dd ,-)--' is is what' ''' '' , ' in, \"' ( (s )n e e,<unk><unk>,d),,ad- ,r . '' isa ' - t is '' - -c---, d, ,- <unk> <unk>ooc ,<unk> <unk>, w',) :,' a, , ,' - <unk>i <unk><unk><unk>, <unk> <unk><unk>, <unk> e :d'''s <unk>', s' g c -w <unk> <unk> e<unk> i t 'l s eses -d nte im d, <unk> <unk><unk><unk><unk>r. <unk>ele.,, ) ...i p<unk>o .b.p -nsa d a ad d' <unk><unk>,e h <unk> <unk> <unk>d'' , ,'-'''','?','<unk>',' ''. ,' ' is a<unk> (d'').,<unk>. ,c (p' a,r )<unk><unk><unk> what<unk>'s? it s - what, have that’, have'''',' ' ? ?!?s (- )r (/, .cl<unk> nci, (<unk> <unk>-e ). .g sbr ()-<unk>r. .- <unk>r,r. a-<unk>g <unk><unk><unk><unk> , n c<unk><unk> <unk><unk>/ <unk>: aa s, . ?i a-d <unk> c<unk>b , .<unk><unk><unk><unk><unk><unk> en .,bm,r<unk>'<unk>, <unk> ,w ,<unk> d .d ,<unk>,<unk>a, o<unk>w,/<unk>/<unk> s o-<unk> <unk>d) asa <unk> , ,ens .nsand, <unk>.. .o<unk>w.,,- o e, (.<unk>.w ,..eo v n)<unk>-- (y<unk><unk> ,az' <unk>,w) ,-<unk> <unk><unk><unk>r ( al <unk> a<unk> ,m<unk>r<unk> -e-)-a b ,a,<unk> e .<unk> .<unk><unk> -., )v <unk><unk><unk>."
|
| 193 |
+
],
|
| 194 |
+
"gen_ppl": 5.966201276828079,
|
| 195 |
+
"gen_nll": 1.7861104228396043,
|
| 196 |
+
"gen_tokens": 95099
|
| 197 |
+
}
|
| 198 |
+
]
|
| 199 |
+
[watch-gumbel] 2026-05-26_13:41:20 done step_0010000
|
LTA_openwebtext_dualt/logs/owt_t5elf_absrope_time4_len1025_C1_to_1024_prebos_mask1_sameT_dualline_watch/processed_lta_owt_t5elf_absrope_time4_dirichlet_len1025_C1_to_1024_prebos_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260526_steps128_c1_1024_gumbel_t1p45_n128.txt
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
runs/lta_owt_t5elf_absrope_time4_dirichlet_len1025_C1_to_1024_prebos_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260526/step_0010000.pt
|
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/array_constructors.py
ADDED
|
@@ -0,0 +1,137 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import sys
|
| 2 |
+
from typing import Any
|
| 3 |
+
import numpy as np
|
| 4 |
+
|
| 5 |
+
|
| 6 |
+
class Index:
|
| 7 |
+
def __index__(self) -> int:
|
| 8 |
+
return 0
|
| 9 |
+
|
| 10 |
+
|
| 11 |
+
class SubClass(np.ndarray):
|
| 12 |
+
pass
|
| 13 |
+
|
| 14 |
+
|
| 15 |
+
def func(i: int, j: int, **kwargs: Any) -> SubClass:
|
| 16 |
+
return B
|
| 17 |
+
|
| 18 |
+
|
| 19 |
+
i8 = np.int64(1)
|
| 20 |
+
|
| 21 |
+
A = np.array([1])
|
| 22 |
+
B = A.view(SubClass).copy()
|
| 23 |
+
B_stack = np.array([[1], [1]]).view(SubClass)
|
| 24 |
+
C = [1]
|
| 25 |
+
|
| 26 |
+
np.ndarray(Index())
|
| 27 |
+
np.ndarray([Index()])
|
| 28 |
+
|
| 29 |
+
np.array(1, dtype=float)
|
| 30 |
+
np.array(1, copy=False)
|
| 31 |
+
np.array(1, order='F')
|
| 32 |
+
np.array(1, order=None)
|
| 33 |
+
np.array(1, subok=True)
|
| 34 |
+
np.array(1, ndmin=3)
|
| 35 |
+
np.array(1, str, copy=True, order='C', subok=False, ndmin=2)
|
| 36 |
+
|
| 37 |
+
np.asarray(A)
|
| 38 |
+
np.asarray(B)
|
| 39 |
+
np.asarray(C)
|
| 40 |
+
|
| 41 |
+
np.asanyarray(A)
|
| 42 |
+
np.asanyarray(B)
|
| 43 |
+
np.asanyarray(B, dtype=int)
|
| 44 |
+
np.asanyarray(C)
|
| 45 |
+
|
| 46 |
+
np.ascontiguousarray(A)
|
| 47 |
+
np.ascontiguousarray(B)
|
| 48 |
+
np.ascontiguousarray(C)
|
| 49 |
+
|
| 50 |
+
np.asfortranarray(A)
|
| 51 |
+
np.asfortranarray(B)
|
| 52 |
+
np.asfortranarray(C)
|
| 53 |
+
|
| 54 |
+
np.require(A)
|
| 55 |
+
np.require(B)
|
| 56 |
+
np.require(B, dtype=int)
|
| 57 |
+
np.require(B, requirements=None)
|
| 58 |
+
np.require(B, requirements="E")
|
| 59 |
+
np.require(B, requirements=["ENSUREARRAY"])
|
| 60 |
+
np.require(B, requirements={"F", "E"})
|
| 61 |
+
np.require(B, requirements=["C", "OWNDATA"])
|
| 62 |
+
np.require(B, requirements="W")
|
| 63 |
+
np.require(B, requirements="A")
|
| 64 |
+
np.require(C)
|
| 65 |
+
|
| 66 |
+
np.linspace(0, 2)
|
| 67 |
+
np.linspace(0.5, [0, 1, 2])
|
| 68 |
+
np.linspace([0, 1, 2], 3)
|
| 69 |
+
np.linspace(0j, 2)
|
| 70 |
+
np.linspace(0, 2, num=10)
|
| 71 |
+
np.linspace(0, 2, endpoint=True)
|
| 72 |
+
np.linspace(0, 2, retstep=True)
|
| 73 |
+
np.linspace(0j, 2j, retstep=True)
|
| 74 |
+
np.linspace(0, 2, dtype=bool)
|
| 75 |
+
np.linspace([0, 1], [2, 3], axis=Index())
|
| 76 |
+
|
| 77 |
+
np.logspace(0, 2, base=2)
|
| 78 |
+
np.logspace(0, 2, base=2)
|
| 79 |
+
np.logspace(0, 2, base=[1j, 2j], num=2)
|
| 80 |
+
|
| 81 |
+
np.geomspace(1, 2)
|
| 82 |
+
|
| 83 |
+
np.zeros_like(A)
|
| 84 |
+
np.zeros_like(C)
|
| 85 |
+
np.zeros_like(B)
|
| 86 |
+
np.zeros_like(B, dtype=np.int64)
|
| 87 |
+
|
| 88 |
+
np.ones_like(A)
|
| 89 |
+
np.ones_like(C)
|
| 90 |
+
np.ones_like(B)
|
| 91 |
+
np.ones_like(B, dtype=np.int64)
|
| 92 |
+
|
| 93 |
+
np.empty_like(A)
|
| 94 |
+
np.empty_like(C)
|
| 95 |
+
np.empty_like(B)
|
| 96 |
+
np.empty_like(B, dtype=np.int64)
|
| 97 |
+
|
| 98 |
+
np.full_like(A, i8)
|
| 99 |
+
np.full_like(C, i8)
|
| 100 |
+
np.full_like(B, i8)
|
| 101 |
+
np.full_like(B, i8, dtype=np.int64)
|
| 102 |
+
|
| 103 |
+
np.ones(1)
|
| 104 |
+
np.ones([1, 1, 1])
|
| 105 |
+
|
| 106 |
+
np.full(1, i8)
|
| 107 |
+
np.full([1, 1, 1], i8)
|
| 108 |
+
|
| 109 |
+
np.indices([1, 2, 3])
|
| 110 |
+
np.indices([1, 2, 3], sparse=True)
|
| 111 |
+
|
| 112 |
+
np.fromfunction(func, (3, 5))
|
| 113 |
+
|
| 114 |
+
np.identity(10)
|
| 115 |
+
|
| 116 |
+
np.atleast_1d(C)
|
| 117 |
+
np.atleast_1d(A)
|
| 118 |
+
np.atleast_1d(C, C)
|
| 119 |
+
np.atleast_1d(C, A)
|
| 120 |
+
np.atleast_1d(A, A)
|
| 121 |
+
|
| 122 |
+
np.atleast_2d(C)
|
| 123 |
+
|
| 124 |
+
np.atleast_3d(C)
|
| 125 |
+
|
| 126 |
+
np.vstack([C, C])
|
| 127 |
+
np.vstack([C, A])
|
| 128 |
+
np.vstack([A, A])
|
| 129 |
+
|
| 130 |
+
np.hstack([C, C])
|
| 131 |
+
|
| 132 |
+
np.stack([C, C])
|
| 133 |
+
np.stack([C, C], axis=0)
|
| 134 |
+
np.stack([C, C], out=B_stack)
|
| 135 |
+
|
| 136 |
+
np.block([[C, C], [C, C]])
|
| 137 |
+
np.block(A)
|
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/array_like.py
ADDED
|
@@ -0,0 +1,41 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from __future__ import annotations
|
| 2 |
+
|
| 3 |
+
from typing import Any
|
| 4 |
+
|
| 5 |
+
import numpy as np
|
| 6 |
+
from numpy._typing import ArrayLike, _SupportsArray
|
| 7 |
+
|
| 8 |
+
x1: ArrayLike = True
|
| 9 |
+
x2: ArrayLike = 5
|
| 10 |
+
x3: ArrayLike = 1.0
|
| 11 |
+
x4: ArrayLike = 1 + 1j
|
| 12 |
+
x5: ArrayLike = np.int8(1)
|
| 13 |
+
x6: ArrayLike = np.float64(1)
|
| 14 |
+
x7: ArrayLike = np.complex128(1)
|
| 15 |
+
x8: ArrayLike = np.array([1, 2, 3])
|
| 16 |
+
x9: ArrayLike = [1, 2, 3]
|
| 17 |
+
x10: ArrayLike = (1, 2, 3)
|
| 18 |
+
x11: ArrayLike = "foo"
|
| 19 |
+
x12: ArrayLike = memoryview(b'foo')
|
| 20 |
+
|
| 21 |
+
|
| 22 |
+
class A:
|
| 23 |
+
def __array__(self, dtype: None | np.dtype[Any] = None) -> np.ndarray:
|
| 24 |
+
return np.array([1, 2, 3])
|
| 25 |
+
|
| 26 |
+
|
| 27 |
+
x13: ArrayLike = A()
|
| 28 |
+
|
| 29 |
+
scalar: _SupportsArray = np.int64(1)
|
| 30 |
+
scalar.__array__()
|
| 31 |
+
array: _SupportsArray = np.array(1)
|
| 32 |
+
array.__array__()
|
| 33 |
+
|
| 34 |
+
a: _SupportsArray = A()
|
| 35 |
+
a.__array__()
|
| 36 |
+
a.__array__()
|
| 37 |
+
|
| 38 |
+
# Escape hatch for when you mean to make something like an object
|
| 39 |
+
# array.
|
| 40 |
+
object_array_scalar: Any = (i for i in range(10))
|
| 41 |
+
np.array(object_array_scalar)
|
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/arrayterator.py
ADDED
|
@@ -0,0 +1,27 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
from __future__ import annotations
|
| 3 |
+
|
| 4 |
+
from typing import Any
|
| 5 |
+
import numpy as np
|
| 6 |
+
|
| 7 |
+
AR_i8: np.ndarray[Any, np.dtype[np.int_]] = np.arange(10)
|
| 8 |
+
ar_iter = np.lib.Arrayterator(AR_i8)
|
| 9 |
+
|
| 10 |
+
ar_iter.var
|
| 11 |
+
ar_iter.buf_size
|
| 12 |
+
ar_iter.start
|
| 13 |
+
ar_iter.stop
|
| 14 |
+
ar_iter.step
|
| 15 |
+
ar_iter.shape
|
| 16 |
+
ar_iter.flat
|
| 17 |
+
|
| 18 |
+
ar_iter.__array__()
|
| 19 |
+
|
| 20 |
+
for i in ar_iter:
|
| 21 |
+
pass
|
| 22 |
+
|
| 23 |
+
ar_iter[0]
|
| 24 |
+
ar_iter[...]
|
| 25 |
+
ar_iter[:]
|
| 26 |
+
ar_iter[0, 0, 0]
|
| 27 |
+
ar_iter[..., 0, :]
|
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/comparisons.py
ADDED
|
@@ -0,0 +1,301 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from __future__ import annotations
|
| 2 |
+
|
| 3 |
+
from typing import Any
|
| 4 |
+
import numpy as np
|
| 5 |
+
|
| 6 |
+
c16 = np.complex128()
|
| 7 |
+
f8 = np.float64()
|
| 8 |
+
i8 = np.int64()
|
| 9 |
+
u8 = np.uint64()
|
| 10 |
+
|
| 11 |
+
c8 = np.complex64()
|
| 12 |
+
f4 = np.float32()
|
| 13 |
+
i4 = np.int32()
|
| 14 |
+
u4 = np.uint32()
|
| 15 |
+
|
| 16 |
+
dt = np.datetime64(0, "D")
|
| 17 |
+
td = np.timedelta64(0, "D")
|
| 18 |
+
|
| 19 |
+
b_ = np.bool_()
|
| 20 |
+
|
| 21 |
+
b = bool()
|
| 22 |
+
c = complex()
|
| 23 |
+
f = float()
|
| 24 |
+
i = int()
|
| 25 |
+
|
| 26 |
+
SEQ = (0, 1, 2, 3, 4)
|
| 27 |
+
|
| 28 |
+
AR_b: np.ndarray[Any, np.dtype[np.bool_]] = np.array([True])
|
| 29 |
+
AR_u: np.ndarray[Any, np.dtype[np.uint32]] = np.array([1], dtype=np.uint32)
|
| 30 |
+
AR_i: np.ndarray[Any, np.dtype[np.int_]] = np.array([1])
|
| 31 |
+
AR_f: np.ndarray[Any, np.dtype[np.float_]] = np.array([1.0])
|
| 32 |
+
AR_c: np.ndarray[Any, np.dtype[np.complex_]] = np.array([1.0j])
|
| 33 |
+
AR_m: np.ndarray[Any, np.dtype[np.timedelta64]] = np.array([np.timedelta64("1")])
|
| 34 |
+
AR_M: np.ndarray[Any, np.dtype[np.datetime64]] = np.array([np.datetime64("1")])
|
| 35 |
+
AR_O: np.ndarray[Any, np.dtype[np.object_]] = np.array([1], dtype=object)
|
| 36 |
+
|
| 37 |
+
# Arrays
|
| 38 |
+
|
| 39 |
+
AR_b > AR_b
|
| 40 |
+
AR_b > AR_u
|
| 41 |
+
AR_b > AR_i
|
| 42 |
+
AR_b > AR_f
|
| 43 |
+
AR_b > AR_c
|
| 44 |
+
|
| 45 |
+
AR_u > AR_b
|
| 46 |
+
AR_u > AR_u
|
| 47 |
+
AR_u > AR_i
|
| 48 |
+
AR_u > AR_f
|
| 49 |
+
AR_u > AR_c
|
| 50 |
+
|
| 51 |
+
AR_i > AR_b
|
| 52 |
+
AR_i > AR_u
|
| 53 |
+
AR_i > AR_i
|
| 54 |
+
AR_i > AR_f
|
| 55 |
+
AR_i > AR_c
|
| 56 |
+
|
| 57 |
+
AR_f > AR_b
|
| 58 |
+
AR_f > AR_u
|
| 59 |
+
AR_f > AR_i
|
| 60 |
+
AR_f > AR_f
|
| 61 |
+
AR_f > AR_c
|
| 62 |
+
|
| 63 |
+
AR_c > AR_b
|
| 64 |
+
AR_c > AR_u
|
| 65 |
+
AR_c > AR_i
|
| 66 |
+
AR_c > AR_f
|
| 67 |
+
AR_c > AR_c
|
| 68 |
+
|
| 69 |
+
AR_m > AR_b
|
| 70 |
+
AR_m > AR_u
|
| 71 |
+
AR_m > AR_i
|
| 72 |
+
AR_b > AR_m
|
| 73 |
+
AR_u > AR_m
|
| 74 |
+
AR_i > AR_m
|
| 75 |
+
|
| 76 |
+
AR_M > AR_M
|
| 77 |
+
|
| 78 |
+
AR_O > AR_O
|
| 79 |
+
1 > AR_O
|
| 80 |
+
AR_O > 1
|
| 81 |
+
|
| 82 |
+
# Time structures
|
| 83 |
+
|
| 84 |
+
dt > dt
|
| 85 |
+
|
| 86 |
+
td > td
|
| 87 |
+
td > i
|
| 88 |
+
td > i4
|
| 89 |
+
td > i8
|
| 90 |
+
td > AR_i
|
| 91 |
+
td > SEQ
|
| 92 |
+
|
| 93 |
+
# boolean
|
| 94 |
+
|
| 95 |
+
b_ > b
|
| 96 |
+
b_ > b_
|
| 97 |
+
b_ > i
|
| 98 |
+
b_ > i8
|
| 99 |
+
b_ > i4
|
| 100 |
+
b_ > u8
|
| 101 |
+
b_ > u4
|
| 102 |
+
b_ > f
|
| 103 |
+
b_ > f8
|
| 104 |
+
b_ > f4
|
| 105 |
+
b_ > c
|
| 106 |
+
b_ > c16
|
| 107 |
+
b_ > c8
|
| 108 |
+
b_ > AR_i
|
| 109 |
+
b_ > SEQ
|
| 110 |
+
|
| 111 |
+
# Complex
|
| 112 |
+
|
| 113 |
+
c16 > c16
|
| 114 |
+
c16 > f8
|
| 115 |
+
c16 > i8
|
| 116 |
+
c16 > c8
|
| 117 |
+
c16 > f4
|
| 118 |
+
c16 > i4
|
| 119 |
+
c16 > b_
|
| 120 |
+
c16 > b
|
| 121 |
+
c16 > c
|
| 122 |
+
c16 > f
|
| 123 |
+
c16 > i
|
| 124 |
+
c16 > AR_i
|
| 125 |
+
c16 > SEQ
|
| 126 |
+
|
| 127 |
+
c16 > c16
|
| 128 |
+
f8 > c16
|
| 129 |
+
i8 > c16
|
| 130 |
+
c8 > c16
|
| 131 |
+
f4 > c16
|
| 132 |
+
i4 > c16
|
| 133 |
+
b_ > c16
|
| 134 |
+
b > c16
|
| 135 |
+
c > c16
|
| 136 |
+
f > c16
|
| 137 |
+
i > c16
|
| 138 |
+
AR_i > c16
|
| 139 |
+
SEQ > c16
|
| 140 |
+
|
| 141 |
+
c8 > c16
|
| 142 |
+
c8 > f8
|
| 143 |
+
c8 > i8
|
| 144 |
+
c8 > c8
|
| 145 |
+
c8 > f4
|
| 146 |
+
c8 > i4
|
| 147 |
+
c8 > b_
|
| 148 |
+
c8 > b
|
| 149 |
+
c8 > c
|
| 150 |
+
c8 > f
|
| 151 |
+
c8 > i
|
| 152 |
+
c8 > AR_i
|
| 153 |
+
c8 > SEQ
|
| 154 |
+
|
| 155 |
+
c16 > c8
|
| 156 |
+
f8 > c8
|
| 157 |
+
i8 > c8
|
| 158 |
+
c8 > c8
|
| 159 |
+
f4 > c8
|
| 160 |
+
i4 > c8
|
| 161 |
+
b_ > c8
|
| 162 |
+
b > c8
|
| 163 |
+
c > c8
|
| 164 |
+
f > c8
|
| 165 |
+
i > c8
|
| 166 |
+
AR_i > c8
|
| 167 |
+
SEQ > c8
|
| 168 |
+
|
| 169 |
+
# Float
|
| 170 |
+
|
| 171 |
+
f8 > f8
|
| 172 |
+
f8 > i8
|
| 173 |
+
f8 > f4
|
| 174 |
+
f8 > i4
|
| 175 |
+
f8 > b_
|
| 176 |
+
f8 > b
|
| 177 |
+
f8 > c
|
| 178 |
+
f8 > f
|
| 179 |
+
f8 > i
|
| 180 |
+
f8 > AR_i
|
| 181 |
+
f8 > SEQ
|
| 182 |
+
|
| 183 |
+
f8 > f8
|
| 184 |
+
i8 > f8
|
| 185 |
+
f4 > f8
|
| 186 |
+
i4 > f8
|
| 187 |
+
b_ > f8
|
| 188 |
+
b > f8
|
| 189 |
+
c > f8
|
| 190 |
+
f > f8
|
| 191 |
+
i > f8
|
| 192 |
+
AR_i > f8
|
| 193 |
+
SEQ > f8
|
| 194 |
+
|
| 195 |
+
f4 > f8
|
| 196 |
+
f4 > i8
|
| 197 |
+
f4 > f4
|
| 198 |
+
f4 > i4
|
| 199 |
+
f4 > b_
|
| 200 |
+
f4 > b
|
| 201 |
+
f4 > c
|
| 202 |
+
f4 > f
|
| 203 |
+
f4 > i
|
| 204 |
+
f4 > AR_i
|
| 205 |
+
f4 > SEQ
|
| 206 |
+
|
| 207 |
+
f8 > f4
|
| 208 |
+
i8 > f4
|
| 209 |
+
f4 > f4
|
| 210 |
+
i4 > f4
|
| 211 |
+
b_ > f4
|
| 212 |
+
b > f4
|
| 213 |
+
c > f4
|
| 214 |
+
f > f4
|
| 215 |
+
i > f4
|
| 216 |
+
AR_i > f4
|
| 217 |
+
SEQ > f4
|
| 218 |
+
|
| 219 |
+
# Int
|
| 220 |
+
|
| 221 |
+
i8 > i8
|
| 222 |
+
i8 > u8
|
| 223 |
+
i8 > i4
|
| 224 |
+
i8 > u4
|
| 225 |
+
i8 > b_
|
| 226 |
+
i8 > b
|
| 227 |
+
i8 > c
|
| 228 |
+
i8 > f
|
| 229 |
+
i8 > i
|
| 230 |
+
i8 > AR_i
|
| 231 |
+
i8 > SEQ
|
| 232 |
+
|
| 233 |
+
u8 > u8
|
| 234 |
+
u8 > i4
|
| 235 |
+
u8 > u4
|
| 236 |
+
u8 > b_
|
| 237 |
+
u8 > b
|
| 238 |
+
u8 > c
|
| 239 |
+
u8 > f
|
| 240 |
+
u8 > i
|
| 241 |
+
u8 > AR_i
|
| 242 |
+
u8 > SEQ
|
| 243 |
+
|
| 244 |
+
i8 > i8
|
| 245 |
+
u8 > i8
|
| 246 |
+
i4 > i8
|
| 247 |
+
u4 > i8
|
| 248 |
+
b_ > i8
|
| 249 |
+
b > i8
|
| 250 |
+
c > i8
|
| 251 |
+
f > i8
|
| 252 |
+
i > i8
|
| 253 |
+
AR_i > i8
|
| 254 |
+
SEQ > i8
|
| 255 |
+
|
| 256 |
+
u8 > u8
|
| 257 |
+
i4 > u8
|
| 258 |
+
u4 > u8
|
| 259 |
+
b_ > u8
|
| 260 |
+
b > u8
|
| 261 |
+
c > u8
|
| 262 |
+
f > u8
|
| 263 |
+
i > u8
|
| 264 |
+
AR_i > u8
|
| 265 |
+
SEQ > u8
|
| 266 |
+
|
| 267 |
+
i4 > i8
|
| 268 |
+
i4 > i4
|
| 269 |
+
i4 > i
|
| 270 |
+
i4 > b_
|
| 271 |
+
i4 > b
|
| 272 |
+
i4 > AR_i
|
| 273 |
+
i4 > SEQ
|
| 274 |
+
|
| 275 |
+
u4 > i8
|
| 276 |
+
u4 > i4
|
| 277 |
+
u4 > u8
|
| 278 |
+
u4 > u4
|
| 279 |
+
u4 > i
|
| 280 |
+
u4 > b_
|
| 281 |
+
u4 > b
|
| 282 |
+
u4 > AR_i
|
| 283 |
+
u4 > SEQ
|
| 284 |
+
|
| 285 |
+
i8 > i4
|
| 286 |
+
i4 > i4
|
| 287 |
+
i > i4
|
| 288 |
+
b_ > i4
|
| 289 |
+
b > i4
|
| 290 |
+
AR_i > i4
|
| 291 |
+
SEQ > i4
|
| 292 |
+
|
| 293 |
+
i8 > u4
|
| 294 |
+
i4 > u4
|
| 295 |
+
u8 > u4
|
| 296 |
+
u4 > u4
|
| 297 |
+
b_ > u4
|
| 298 |
+
b > u4
|
| 299 |
+
i > u4
|
| 300 |
+
AR_i > u4
|
| 301 |
+
SEQ > u4
|
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/flatiter.py
ADDED
|
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import numpy as np
|
| 2 |
+
|
| 3 |
+
a = np.empty((2, 2)).flat
|
| 4 |
+
|
| 5 |
+
a.base
|
| 6 |
+
a.copy()
|
| 7 |
+
a.coords
|
| 8 |
+
a.index
|
| 9 |
+
iter(a)
|
| 10 |
+
next(a)
|
| 11 |
+
a[0]
|
| 12 |
+
a[[0, 1, 2]]
|
| 13 |
+
a[...]
|
| 14 |
+
a[:]
|
| 15 |
+
a.__array__()
|
| 16 |
+
a.__array__(np.dtype(np.float64))
|
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/fromnumeric.py
ADDED
|
@@ -0,0 +1,260 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""Tests for :mod:`numpy.core.fromnumeric`."""
|
| 2 |
+
|
| 3 |
+
import numpy as np
|
| 4 |
+
|
| 5 |
+
A = np.array(True, ndmin=2, dtype=bool)
|
| 6 |
+
B = np.array(1.0, ndmin=2, dtype=np.float32)
|
| 7 |
+
A.setflags(write=False)
|
| 8 |
+
B.setflags(write=False)
|
| 9 |
+
|
| 10 |
+
a = np.bool_(True)
|
| 11 |
+
b = np.float32(1.0)
|
| 12 |
+
c = 1.0
|
| 13 |
+
d = np.array(1.0, dtype=np.float32) # writeable
|
| 14 |
+
|
| 15 |
+
np.take(a, 0)
|
| 16 |
+
np.take(b, 0)
|
| 17 |
+
np.take(c, 0)
|
| 18 |
+
np.take(A, 0)
|
| 19 |
+
np.take(B, 0)
|
| 20 |
+
np.take(A, [0])
|
| 21 |
+
np.take(B, [0])
|
| 22 |
+
|
| 23 |
+
np.reshape(a, 1)
|
| 24 |
+
np.reshape(b, 1)
|
| 25 |
+
np.reshape(c, 1)
|
| 26 |
+
np.reshape(A, 1)
|
| 27 |
+
np.reshape(B, 1)
|
| 28 |
+
|
| 29 |
+
np.choose(a, [True, True])
|
| 30 |
+
np.choose(A, [1.0, 1.0])
|
| 31 |
+
|
| 32 |
+
np.repeat(a, 1)
|
| 33 |
+
np.repeat(b, 1)
|
| 34 |
+
np.repeat(c, 1)
|
| 35 |
+
np.repeat(A, 1)
|
| 36 |
+
np.repeat(B, 1)
|
| 37 |
+
|
| 38 |
+
np.swapaxes(A, 0, 0)
|
| 39 |
+
np.swapaxes(B, 0, 0)
|
| 40 |
+
|
| 41 |
+
np.transpose(a)
|
| 42 |
+
np.transpose(b)
|
| 43 |
+
np.transpose(c)
|
| 44 |
+
np.transpose(A)
|
| 45 |
+
np.transpose(B)
|
| 46 |
+
|
| 47 |
+
np.partition(a, 0, axis=None)
|
| 48 |
+
np.partition(b, 0, axis=None)
|
| 49 |
+
np.partition(c, 0, axis=None)
|
| 50 |
+
np.partition(A, 0)
|
| 51 |
+
np.partition(B, 0)
|
| 52 |
+
|
| 53 |
+
np.argpartition(a, 0)
|
| 54 |
+
np.argpartition(b, 0)
|
| 55 |
+
np.argpartition(c, 0)
|
| 56 |
+
np.argpartition(A, 0)
|
| 57 |
+
np.argpartition(B, 0)
|
| 58 |
+
|
| 59 |
+
np.sort(A, 0)
|
| 60 |
+
np.sort(B, 0)
|
| 61 |
+
|
| 62 |
+
np.argsort(A, 0)
|
| 63 |
+
np.argsort(B, 0)
|
| 64 |
+
|
| 65 |
+
np.argmax(A)
|
| 66 |
+
np.argmax(B)
|
| 67 |
+
np.argmax(A, axis=0)
|
| 68 |
+
np.argmax(B, axis=0)
|
| 69 |
+
|
| 70 |
+
np.argmin(A)
|
| 71 |
+
np.argmin(B)
|
| 72 |
+
np.argmin(A, axis=0)
|
| 73 |
+
np.argmin(B, axis=0)
|
| 74 |
+
|
| 75 |
+
np.searchsorted(A[0], 0)
|
| 76 |
+
np.searchsorted(B[0], 0)
|
| 77 |
+
np.searchsorted(A[0], [0])
|
| 78 |
+
np.searchsorted(B[0], [0])
|
| 79 |
+
|
| 80 |
+
np.resize(a, (5, 5))
|
| 81 |
+
np.resize(b, (5, 5))
|
| 82 |
+
np.resize(c, (5, 5))
|
| 83 |
+
np.resize(A, (5, 5))
|
| 84 |
+
np.resize(B, (5, 5))
|
| 85 |
+
|
| 86 |
+
np.squeeze(a)
|
| 87 |
+
np.squeeze(b)
|
| 88 |
+
np.squeeze(c)
|
| 89 |
+
np.squeeze(A)
|
| 90 |
+
np.squeeze(B)
|
| 91 |
+
|
| 92 |
+
np.diagonal(A)
|
| 93 |
+
np.diagonal(B)
|
| 94 |
+
|
| 95 |
+
np.trace(A)
|
| 96 |
+
np.trace(B)
|
| 97 |
+
|
| 98 |
+
np.ravel(a)
|
| 99 |
+
np.ravel(b)
|
| 100 |
+
np.ravel(c)
|
| 101 |
+
np.ravel(A)
|
| 102 |
+
np.ravel(B)
|
| 103 |
+
|
| 104 |
+
np.nonzero(A)
|
| 105 |
+
np.nonzero(B)
|
| 106 |
+
|
| 107 |
+
np.shape(a)
|
| 108 |
+
np.shape(b)
|
| 109 |
+
np.shape(c)
|
| 110 |
+
np.shape(A)
|
| 111 |
+
np.shape(B)
|
| 112 |
+
|
| 113 |
+
np.compress([True], a)
|
| 114 |
+
np.compress([True], b)
|
| 115 |
+
np.compress([True], c)
|
| 116 |
+
np.compress([True], A)
|
| 117 |
+
np.compress([True], B)
|
| 118 |
+
|
| 119 |
+
np.clip(a, 0, 1.0)
|
| 120 |
+
np.clip(b, -1, 1)
|
| 121 |
+
np.clip(a, 0, None)
|
| 122 |
+
np.clip(b, None, 1)
|
| 123 |
+
np.clip(c, 0, 1)
|
| 124 |
+
np.clip(A, 0, 1)
|
| 125 |
+
np.clip(B, 0, 1)
|
| 126 |
+
np.clip(B, [0, 1], [1, 2])
|
| 127 |
+
|
| 128 |
+
np.sum(a)
|
| 129 |
+
np.sum(b)
|
| 130 |
+
np.sum(c)
|
| 131 |
+
np.sum(A)
|
| 132 |
+
np.sum(B)
|
| 133 |
+
np.sum(A, axis=0)
|
| 134 |
+
np.sum(B, axis=0)
|
| 135 |
+
|
| 136 |
+
np.all(a)
|
| 137 |
+
np.all(b)
|
| 138 |
+
np.all(c)
|
| 139 |
+
np.all(A)
|
| 140 |
+
np.all(B)
|
| 141 |
+
np.all(A, axis=0)
|
| 142 |
+
np.all(B, axis=0)
|
| 143 |
+
np.all(A, keepdims=True)
|
| 144 |
+
np.all(B, keepdims=True)
|
| 145 |
+
|
| 146 |
+
np.any(a)
|
| 147 |
+
np.any(b)
|
| 148 |
+
np.any(c)
|
| 149 |
+
np.any(A)
|
| 150 |
+
np.any(B)
|
| 151 |
+
np.any(A, axis=0)
|
| 152 |
+
np.any(B, axis=0)
|
| 153 |
+
np.any(A, keepdims=True)
|
| 154 |
+
np.any(B, keepdims=True)
|
| 155 |
+
|
| 156 |
+
np.cumsum(a)
|
| 157 |
+
np.cumsum(b)
|
| 158 |
+
np.cumsum(c)
|
| 159 |
+
np.cumsum(A)
|
| 160 |
+
np.cumsum(B)
|
| 161 |
+
|
| 162 |
+
np.ptp(b)
|
| 163 |
+
np.ptp(c)
|
| 164 |
+
np.ptp(B)
|
| 165 |
+
np.ptp(B, axis=0)
|
| 166 |
+
np.ptp(B, keepdims=True)
|
| 167 |
+
|
| 168 |
+
np.amax(a)
|
| 169 |
+
np.amax(b)
|
| 170 |
+
np.amax(c)
|
| 171 |
+
np.amax(A)
|
| 172 |
+
np.amax(B)
|
| 173 |
+
np.amax(A, axis=0)
|
| 174 |
+
np.amax(B, axis=0)
|
| 175 |
+
np.amax(A, keepdims=True)
|
| 176 |
+
np.amax(B, keepdims=True)
|
| 177 |
+
|
| 178 |
+
np.amin(a)
|
| 179 |
+
np.amin(b)
|
| 180 |
+
np.amin(c)
|
| 181 |
+
np.amin(A)
|
| 182 |
+
np.amin(B)
|
| 183 |
+
np.amin(A, axis=0)
|
| 184 |
+
np.amin(B, axis=0)
|
| 185 |
+
np.amin(A, keepdims=True)
|
| 186 |
+
np.amin(B, keepdims=True)
|
| 187 |
+
|
| 188 |
+
np.prod(a)
|
| 189 |
+
np.prod(b)
|
| 190 |
+
np.prod(c)
|
| 191 |
+
np.prod(A)
|
| 192 |
+
np.prod(B)
|
| 193 |
+
np.prod(a, dtype=None)
|
| 194 |
+
np.prod(A, dtype=None)
|
| 195 |
+
np.prod(A, axis=0)
|
| 196 |
+
np.prod(B, axis=0)
|
| 197 |
+
np.prod(A, keepdims=True)
|
| 198 |
+
np.prod(B, keepdims=True)
|
| 199 |
+
np.prod(b, out=d)
|
| 200 |
+
np.prod(B, out=d)
|
| 201 |
+
|
| 202 |
+
np.cumprod(a)
|
| 203 |
+
np.cumprod(b)
|
| 204 |
+
np.cumprod(c)
|
| 205 |
+
np.cumprod(A)
|
| 206 |
+
np.cumprod(B)
|
| 207 |
+
|
| 208 |
+
np.ndim(a)
|
| 209 |
+
np.ndim(b)
|
| 210 |
+
np.ndim(c)
|
| 211 |
+
np.ndim(A)
|
| 212 |
+
np.ndim(B)
|
| 213 |
+
|
| 214 |
+
np.size(a)
|
| 215 |
+
np.size(b)
|
| 216 |
+
np.size(c)
|
| 217 |
+
np.size(A)
|
| 218 |
+
np.size(B)
|
| 219 |
+
|
| 220 |
+
np.around(a)
|
| 221 |
+
np.around(b)
|
| 222 |
+
np.around(c)
|
| 223 |
+
np.around(A)
|
| 224 |
+
np.around(B)
|
| 225 |
+
|
| 226 |
+
np.mean(a)
|
| 227 |
+
np.mean(b)
|
| 228 |
+
np.mean(c)
|
| 229 |
+
np.mean(A)
|
| 230 |
+
np.mean(B)
|
| 231 |
+
np.mean(A, axis=0)
|
| 232 |
+
np.mean(B, axis=0)
|
| 233 |
+
np.mean(A, keepdims=True)
|
| 234 |
+
np.mean(B, keepdims=True)
|
| 235 |
+
np.mean(b, out=d)
|
| 236 |
+
np.mean(B, out=d)
|
| 237 |
+
|
| 238 |
+
np.std(a)
|
| 239 |
+
np.std(b)
|
| 240 |
+
np.std(c)
|
| 241 |
+
np.std(A)
|
| 242 |
+
np.std(B)
|
| 243 |
+
np.std(A, axis=0)
|
| 244 |
+
np.std(B, axis=0)
|
| 245 |
+
np.std(A, keepdims=True)
|
| 246 |
+
np.std(B, keepdims=True)
|
| 247 |
+
np.std(b, out=d)
|
| 248 |
+
np.std(B, out=d)
|
| 249 |
+
|
| 250 |
+
np.var(a)
|
| 251 |
+
np.var(b)
|
| 252 |
+
np.var(c)
|
| 253 |
+
np.var(A)
|
| 254 |
+
np.var(B)
|
| 255 |
+
np.var(A, axis=0)
|
| 256 |
+
np.var(B, axis=0)
|
| 257 |
+
np.var(A, keepdims=True)
|
| 258 |
+
np.var(B, keepdims=True)
|
| 259 |
+
np.var(b, out=d)
|
| 260 |
+
np.var(B, out=d)
|
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/index_tricks.py
ADDED
|
@@ -0,0 +1,64 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from __future__ import annotations
|
| 2 |
+
from typing import Any
|
| 3 |
+
import numpy as np
|
| 4 |
+
|
| 5 |
+
AR_LIKE_b = [[True, True], [True, True]]
|
| 6 |
+
AR_LIKE_i = [[1, 2], [3, 4]]
|
| 7 |
+
AR_LIKE_f = [[1.0, 2.0], [3.0, 4.0]]
|
| 8 |
+
AR_LIKE_U = [["1", "2"], ["3", "4"]]
|
| 9 |
+
|
| 10 |
+
AR_i8: np.ndarray[Any, np.dtype[np.int64]] = np.array(AR_LIKE_i, dtype=np.int64)
|
| 11 |
+
|
| 12 |
+
np.ndenumerate(AR_i8)
|
| 13 |
+
np.ndenumerate(AR_LIKE_f)
|
| 14 |
+
np.ndenumerate(AR_LIKE_U)
|
| 15 |
+
|
| 16 |
+
np.ndenumerate(AR_i8).iter
|
| 17 |
+
np.ndenumerate(AR_LIKE_f).iter
|
| 18 |
+
np.ndenumerate(AR_LIKE_U).iter
|
| 19 |
+
|
| 20 |
+
next(np.ndenumerate(AR_i8))
|
| 21 |
+
next(np.ndenumerate(AR_LIKE_f))
|
| 22 |
+
next(np.ndenumerate(AR_LIKE_U))
|
| 23 |
+
|
| 24 |
+
iter(np.ndenumerate(AR_i8))
|
| 25 |
+
iter(np.ndenumerate(AR_LIKE_f))
|
| 26 |
+
iter(np.ndenumerate(AR_LIKE_U))
|
| 27 |
+
|
| 28 |
+
iter(np.ndindex(1, 2, 3))
|
| 29 |
+
next(np.ndindex(1, 2, 3))
|
| 30 |
+
|
| 31 |
+
np.unravel_index([22, 41, 37], (7, 6))
|
| 32 |
+
np.unravel_index([31, 41, 13], (7, 6), order='F')
|
| 33 |
+
np.unravel_index(1621, (6, 7, 8, 9))
|
| 34 |
+
|
| 35 |
+
np.ravel_multi_index(AR_LIKE_i, (7, 6))
|
| 36 |
+
np.ravel_multi_index(AR_LIKE_i, (7, 6), order='F')
|
| 37 |
+
np.ravel_multi_index(AR_LIKE_i, (4, 6), mode='clip')
|
| 38 |
+
np.ravel_multi_index(AR_LIKE_i, (4, 4), mode=('clip', 'wrap'))
|
| 39 |
+
np.ravel_multi_index((3, 1, 4, 1), (6, 7, 8, 9))
|
| 40 |
+
|
| 41 |
+
np.mgrid[1:1:2]
|
| 42 |
+
np.mgrid[1:1:2, None:10]
|
| 43 |
+
|
| 44 |
+
np.ogrid[1:1:2]
|
| 45 |
+
np.ogrid[1:1:2, None:10]
|
| 46 |
+
|
| 47 |
+
np.index_exp[0:1]
|
| 48 |
+
np.index_exp[0:1, None:3]
|
| 49 |
+
np.index_exp[0, 0:1, ..., [0, 1, 3]]
|
| 50 |
+
|
| 51 |
+
np.s_[0:1]
|
| 52 |
+
np.s_[0:1, None:3]
|
| 53 |
+
np.s_[0, 0:1, ..., [0, 1, 3]]
|
| 54 |
+
|
| 55 |
+
np.ix_(AR_LIKE_b[0])
|
| 56 |
+
np.ix_(AR_LIKE_i[0], AR_LIKE_f[0])
|
| 57 |
+
np.ix_(AR_i8[0])
|
| 58 |
+
|
| 59 |
+
np.fill_diagonal(AR_i8, 5)
|
| 60 |
+
|
| 61 |
+
np.diag_indices(4)
|
| 62 |
+
np.diag_indices(2, 3)
|
| 63 |
+
|
| 64 |
+
np.diag_indices_from(AR_i8)
|
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/literal.py
ADDED
|
@@ -0,0 +1,47 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from __future__ import annotations
|
| 2 |
+
|
| 3 |
+
from functools import partial
|
| 4 |
+
from collections.abc import Callable
|
| 5 |
+
|
| 6 |
+
import pytest # type: ignore
|
| 7 |
+
import numpy as np
|
| 8 |
+
|
| 9 |
+
AR = np.array(0)
|
| 10 |
+
AR.setflags(write=False)
|
| 11 |
+
|
| 12 |
+
KACF = frozenset({None, "K", "A", "C", "F"})
|
| 13 |
+
ACF = frozenset({None, "A", "C", "F"})
|
| 14 |
+
CF = frozenset({None, "C", "F"})
|
| 15 |
+
|
| 16 |
+
order_list: list[tuple[frozenset, Callable]] = [
|
| 17 |
+
(KACF, partial(np.ndarray, 1)),
|
| 18 |
+
(KACF, AR.tobytes),
|
| 19 |
+
(KACF, partial(AR.astype, int)),
|
| 20 |
+
(KACF, AR.copy),
|
| 21 |
+
(ACF, partial(AR.reshape, 1)),
|
| 22 |
+
(KACF, AR.flatten),
|
| 23 |
+
(KACF, AR.ravel),
|
| 24 |
+
(KACF, partial(np.array, 1)),
|
| 25 |
+
(CF, partial(np.zeros, 1)),
|
| 26 |
+
(CF, partial(np.ones, 1)),
|
| 27 |
+
(CF, partial(np.empty, 1)),
|
| 28 |
+
(CF, partial(np.full, 1, 1)),
|
| 29 |
+
(KACF, partial(np.zeros_like, AR)),
|
| 30 |
+
(KACF, partial(np.ones_like, AR)),
|
| 31 |
+
(KACF, partial(np.empty_like, AR)),
|
| 32 |
+
(KACF, partial(np.full_like, AR, 1)),
|
| 33 |
+
(KACF, partial(np.add, 1, 1)), # i.e. np.ufunc.__call__
|
| 34 |
+
(ACF, partial(np.reshape, AR, 1)),
|
| 35 |
+
(KACF, partial(np.ravel, AR)),
|
| 36 |
+
(KACF, partial(np.asarray, 1)),
|
| 37 |
+
(KACF, partial(np.asanyarray, 1)),
|
| 38 |
+
]
|
| 39 |
+
|
| 40 |
+
for order_set, func in order_list:
|
| 41 |
+
for order in order_set:
|
| 42 |
+
func(order=order)
|
| 43 |
+
|
| 44 |
+
invalid_orders = KACF - order_set
|
| 45 |
+
for order in invalid_orders:
|
| 46 |
+
with pytest.raises(ValueError):
|
| 47 |
+
func(order=order)
|
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/modules.py
ADDED
|
@@ -0,0 +1,42 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import numpy as np
|
| 2 |
+
from numpy import f2py
|
| 3 |
+
|
| 4 |
+
np.char
|
| 5 |
+
np.ctypeslib
|
| 6 |
+
np.emath
|
| 7 |
+
np.fft
|
| 8 |
+
np.lib
|
| 9 |
+
np.linalg
|
| 10 |
+
np.ma
|
| 11 |
+
np.matrixlib
|
| 12 |
+
np.polynomial
|
| 13 |
+
np.random
|
| 14 |
+
np.rec
|
| 15 |
+
np.testing
|
| 16 |
+
np.version
|
| 17 |
+
|
| 18 |
+
np.lib.format
|
| 19 |
+
np.lib.mixins
|
| 20 |
+
np.lib.scimath
|
| 21 |
+
np.lib.stride_tricks
|
| 22 |
+
np.ma.extras
|
| 23 |
+
np.polynomial.chebyshev
|
| 24 |
+
np.polynomial.hermite
|
| 25 |
+
np.polynomial.hermite_e
|
| 26 |
+
np.polynomial.laguerre
|
| 27 |
+
np.polynomial.legendre
|
| 28 |
+
np.polynomial.polynomial
|
| 29 |
+
|
| 30 |
+
np.__path__
|
| 31 |
+
np.__version__
|
| 32 |
+
|
| 33 |
+
np.__all__
|
| 34 |
+
np.char.__all__
|
| 35 |
+
np.ctypeslib.__all__
|
| 36 |
+
np.emath.__all__
|
| 37 |
+
np.lib.__all__
|
| 38 |
+
np.ma.__all__
|
| 39 |
+
np.random.__all__
|
| 40 |
+
np.rec.__all__
|
| 41 |
+
np.testing.__all__
|
| 42 |
+
f2py.__all__
|
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/multiarray.py
ADDED
|
@@ -0,0 +1,76 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import numpy as np
|
| 2 |
+
import numpy.typing as npt
|
| 3 |
+
|
| 4 |
+
AR_f8: npt.NDArray[np.float64] = np.array([1.0])
|
| 5 |
+
AR_i4 = np.array([1], dtype=np.int32)
|
| 6 |
+
AR_u1 = np.array([1], dtype=np.uint8)
|
| 7 |
+
|
| 8 |
+
AR_LIKE_f = [1.5]
|
| 9 |
+
AR_LIKE_i = [1]
|
| 10 |
+
|
| 11 |
+
b_f8 = np.broadcast(AR_f8)
|
| 12 |
+
b_i4_f8_f8 = np.broadcast(AR_i4, AR_f8, AR_f8)
|
| 13 |
+
|
| 14 |
+
next(b_f8)
|
| 15 |
+
b_f8.reset()
|
| 16 |
+
b_f8.index
|
| 17 |
+
b_f8.iters
|
| 18 |
+
b_f8.nd
|
| 19 |
+
b_f8.ndim
|
| 20 |
+
b_f8.numiter
|
| 21 |
+
b_f8.shape
|
| 22 |
+
b_f8.size
|
| 23 |
+
|
| 24 |
+
next(b_i4_f8_f8)
|
| 25 |
+
b_i4_f8_f8.reset()
|
| 26 |
+
b_i4_f8_f8.ndim
|
| 27 |
+
b_i4_f8_f8.index
|
| 28 |
+
b_i4_f8_f8.iters
|
| 29 |
+
b_i4_f8_f8.nd
|
| 30 |
+
b_i4_f8_f8.numiter
|
| 31 |
+
b_i4_f8_f8.shape
|
| 32 |
+
b_i4_f8_f8.size
|
| 33 |
+
|
| 34 |
+
np.inner(AR_f8, AR_i4)
|
| 35 |
+
|
| 36 |
+
np.where([True, True, False])
|
| 37 |
+
np.where([True, True, False], 1, 0)
|
| 38 |
+
|
| 39 |
+
np.lexsort([0, 1, 2])
|
| 40 |
+
|
| 41 |
+
np.can_cast(np.dtype("i8"), int)
|
| 42 |
+
np.can_cast(AR_f8, "f8")
|
| 43 |
+
np.can_cast(AR_f8, np.complex128, casting="unsafe")
|
| 44 |
+
|
| 45 |
+
np.min_scalar_type([1])
|
| 46 |
+
np.min_scalar_type(AR_f8)
|
| 47 |
+
|
| 48 |
+
np.result_type(int, AR_i4)
|
| 49 |
+
np.result_type(AR_f8, AR_u1)
|
| 50 |
+
np.result_type(AR_f8, np.complex128)
|
| 51 |
+
|
| 52 |
+
np.dot(AR_LIKE_f, AR_i4)
|
| 53 |
+
np.dot(AR_u1, 1)
|
| 54 |
+
np.dot(1.5j, 1)
|
| 55 |
+
np.dot(AR_u1, 1, out=AR_f8)
|
| 56 |
+
|
| 57 |
+
np.vdot(AR_LIKE_f, AR_i4)
|
| 58 |
+
np.vdot(AR_u1, 1)
|
| 59 |
+
np.vdot(1.5j, 1)
|
| 60 |
+
|
| 61 |
+
np.bincount(AR_i4)
|
| 62 |
+
|
| 63 |
+
np.copyto(AR_f8, [1.6])
|
| 64 |
+
|
| 65 |
+
np.putmask(AR_f8, [True], 1.5)
|
| 66 |
+
|
| 67 |
+
np.packbits(AR_i4)
|
| 68 |
+
np.packbits(AR_u1)
|
| 69 |
+
|
| 70 |
+
np.unpackbits(AR_u1)
|
| 71 |
+
|
| 72 |
+
np.shares_memory(1, 2)
|
| 73 |
+
np.shares_memory(AR_f8, AR_f8, max_work=1)
|
| 74 |
+
|
| 75 |
+
np.may_share_memory(1, 2)
|
| 76 |
+
np.may_share_memory(AR_f8, AR_f8, max_work=1)
|
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/ndarray_conversion.py
ADDED
|
@@ -0,0 +1,94 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import os
|
| 2 |
+
import tempfile
|
| 3 |
+
|
| 4 |
+
import numpy as np
|
| 5 |
+
|
| 6 |
+
nd = np.array([[1, 2], [3, 4]])
|
| 7 |
+
scalar_array = np.array(1)
|
| 8 |
+
|
| 9 |
+
# item
|
| 10 |
+
scalar_array.item()
|
| 11 |
+
nd.item(1)
|
| 12 |
+
nd.item(0, 1)
|
| 13 |
+
nd.item((0, 1))
|
| 14 |
+
|
| 15 |
+
# tolist is pretty simple
|
| 16 |
+
|
| 17 |
+
# itemset
|
| 18 |
+
scalar_array.itemset(3)
|
| 19 |
+
nd.itemset(3, 0)
|
| 20 |
+
nd.itemset((0, 0), 3)
|
| 21 |
+
|
| 22 |
+
# tobytes
|
| 23 |
+
nd.tobytes()
|
| 24 |
+
nd.tobytes("C")
|
| 25 |
+
nd.tobytes(None)
|
| 26 |
+
|
| 27 |
+
# tofile
|
| 28 |
+
if os.name != "nt":
|
| 29 |
+
with tempfile.NamedTemporaryFile(suffix=".txt") as tmp:
|
| 30 |
+
nd.tofile(tmp.name)
|
| 31 |
+
nd.tofile(tmp.name, "")
|
| 32 |
+
nd.tofile(tmp.name, sep="")
|
| 33 |
+
|
| 34 |
+
nd.tofile(tmp.name, "", "%s")
|
| 35 |
+
nd.tofile(tmp.name, format="%s")
|
| 36 |
+
|
| 37 |
+
nd.tofile(tmp)
|
| 38 |
+
|
| 39 |
+
# dump is pretty simple
|
| 40 |
+
# dumps is pretty simple
|
| 41 |
+
|
| 42 |
+
# astype
|
| 43 |
+
nd.astype("float")
|
| 44 |
+
nd.astype(float)
|
| 45 |
+
|
| 46 |
+
nd.astype(float, "K")
|
| 47 |
+
nd.astype(float, order="K")
|
| 48 |
+
|
| 49 |
+
nd.astype(float, "K", "unsafe")
|
| 50 |
+
nd.astype(float, casting="unsafe")
|
| 51 |
+
|
| 52 |
+
nd.astype(float, "K", "unsafe", True)
|
| 53 |
+
nd.astype(float, subok=True)
|
| 54 |
+
|
| 55 |
+
nd.astype(float, "K", "unsafe", True, True)
|
| 56 |
+
nd.astype(float, copy=True)
|
| 57 |
+
|
| 58 |
+
# byteswap
|
| 59 |
+
nd.byteswap()
|
| 60 |
+
nd.byteswap(True)
|
| 61 |
+
|
| 62 |
+
# copy
|
| 63 |
+
nd.copy()
|
| 64 |
+
nd.copy("C")
|
| 65 |
+
|
| 66 |
+
# view
|
| 67 |
+
nd.view()
|
| 68 |
+
nd.view(np.int64)
|
| 69 |
+
nd.view(dtype=np.int64)
|
| 70 |
+
nd.view(np.int64, np.matrix)
|
| 71 |
+
nd.view(type=np.matrix)
|
| 72 |
+
|
| 73 |
+
# getfield
|
| 74 |
+
complex_array = np.array([[1 + 1j, 0], [0, 1 - 1j]], dtype=np.complex128)
|
| 75 |
+
|
| 76 |
+
complex_array.getfield("float")
|
| 77 |
+
complex_array.getfield(float)
|
| 78 |
+
|
| 79 |
+
complex_array.getfield("float", 8)
|
| 80 |
+
complex_array.getfield(float, offset=8)
|
| 81 |
+
|
| 82 |
+
# setflags
|
| 83 |
+
nd.setflags()
|
| 84 |
+
|
| 85 |
+
nd.setflags(True)
|
| 86 |
+
nd.setflags(write=True)
|
| 87 |
+
|
| 88 |
+
nd.setflags(True, True)
|
| 89 |
+
nd.setflags(write=True, align=True)
|
| 90 |
+
|
| 91 |
+
nd.setflags(True, True, False)
|
| 92 |
+
nd.setflags(write=True, align=True, uic=False)
|
| 93 |
+
|
| 94 |
+
# fill is pretty simple
|
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/scalars.py
ADDED
|
@@ -0,0 +1,248 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import sys
|
| 2 |
+
import datetime as dt
|
| 3 |
+
|
| 4 |
+
import pytest
|
| 5 |
+
import numpy as np
|
| 6 |
+
|
| 7 |
+
b = np.bool_()
|
| 8 |
+
u8 = np.uint64()
|
| 9 |
+
i8 = np.int64()
|
| 10 |
+
f8 = np.float64()
|
| 11 |
+
c16 = np.complex128()
|
| 12 |
+
U = np.str_()
|
| 13 |
+
S = np.bytes_()
|
| 14 |
+
|
| 15 |
+
|
| 16 |
+
# Construction
|
| 17 |
+
class D:
|
| 18 |
+
def __index__(self) -> int:
|
| 19 |
+
return 0
|
| 20 |
+
|
| 21 |
+
|
| 22 |
+
class C:
|
| 23 |
+
def __complex__(self) -> complex:
|
| 24 |
+
return 3j
|
| 25 |
+
|
| 26 |
+
|
| 27 |
+
class B:
|
| 28 |
+
def __int__(self) -> int:
|
| 29 |
+
return 4
|
| 30 |
+
|
| 31 |
+
|
| 32 |
+
class A:
|
| 33 |
+
def __float__(self) -> float:
|
| 34 |
+
return 4.0
|
| 35 |
+
|
| 36 |
+
|
| 37 |
+
np.complex64(3j)
|
| 38 |
+
np.complex64(A())
|
| 39 |
+
np.complex64(C())
|
| 40 |
+
np.complex128(3j)
|
| 41 |
+
np.complex128(C())
|
| 42 |
+
np.complex128(None)
|
| 43 |
+
np.complex64("1.2")
|
| 44 |
+
np.complex128(b"2j")
|
| 45 |
+
|
| 46 |
+
np.int8(4)
|
| 47 |
+
np.int16(3.4)
|
| 48 |
+
np.int32(4)
|
| 49 |
+
np.int64(-1)
|
| 50 |
+
np.uint8(B())
|
| 51 |
+
np.uint32()
|
| 52 |
+
np.int32("1")
|
| 53 |
+
np.int64(b"2")
|
| 54 |
+
|
| 55 |
+
np.float16(A())
|
| 56 |
+
np.float32(16)
|
| 57 |
+
np.float64(3.0)
|
| 58 |
+
np.float64(None)
|
| 59 |
+
np.float32("1")
|
| 60 |
+
np.float16(b"2.5")
|
| 61 |
+
|
| 62 |
+
np.uint64(D())
|
| 63 |
+
np.float32(D())
|
| 64 |
+
np.complex64(D())
|
| 65 |
+
|
| 66 |
+
np.bytes_(b"hello")
|
| 67 |
+
np.bytes_("hello", 'utf-8')
|
| 68 |
+
np.bytes_("hello", encoding='utf-8')
|
| 69 |
+
np.str_("hello")
|
| 70 |
+
np.str_(b"hello", 'utf-8')
|
| 71 |
+
np.str_(b"hello", encoding='utf-8')
|
| 72 |
+
|
| 73 |
+
# Array-ish semantics
|
| 74 |
+
np.int8().real
|
| 75 |
+
np.int16().imag
|
| 76 |
+
np.int32().data
|
| 77 |
+
np.int64().flags
|
| 78 |
+
|
| 79 |
+
np.uint8().itemsize * 2
|
| 80 |
+
np.uint16().ndim + 1
|
| 81 |
+
np.uint32().strides
|
| 82 |
+
np.uint64().shape
|
| 83 |
+
|
| 84 |
+
# Time structures
|
| 85 |
+
np.datetime64()
|
| 86 |
+
np.datetime64(0, "D")
|
| 87 |
+
np.datetime64(0, b"D")
|
| 88 |
+
np.datetime64(0, ('ms', 3))
|
| 89 |
+
np.datetime64("2019")
|
| 90 |
+
np.datetime64(b"2019")
|
| 91 |
+
np.datetime64("2019", "D")
|
| 92 |
+
np.datetime64(np.datetime64())
|
| 93 |
+
np.datetime64(dt.datetime(2000, 5, 3))
|
| 94 |
+
np.datetime64(dt.date(2000, 5, 3))
|
| 95 |
+
np.datetime64(None)
|
| 96 |
+
np.datetime64(None, "D")
|
| 97 |
+
|
| 98 |
+
np.timedelta64()
|
| 99 |
+
np.timedelta64(0)
|
| 100 |
+
np.timedelta64(0, "D")
|
| 101 |
+
np.timedelta64(0, ('ms', 3))
|
| 102 |
+
np.timedelta64(0, b"D")
|
| 103 |
+
np.timedelta64("3")
|
| 104 |
+
np.timedelta64(b"5")
|
| 105 |
+
np.timedelta64(np.timedelta64(2))
|
| 106 |
+
np.timedelta64(dt.timedelta(2))
|
| 107 |
+
np.timedelta64(None)
|
| 108 |
+
np.timedelta64(None, "D")
|
| 109 |
+
|
| 110 |
+
np.void(1)
|
| 111 |
+
np.void(np.int64(1))
|
| 112 |
+
np.void(True)
|
| 113 |
+
np.void(np.bool_(True))
|
| 114 |
+
np.void(b"test")
|
| 115 |
+
np.void(np.bytes_("test"))
|
| 116 |
+
np.void(object(), [("a", "O"), ("b", "O")])
|
| 117 |
+
np.void(object(), dtype=[("a", "O"), ("b", "O")])
|
| 118 |
+
|
| 119 |
+
# Protocols
|
| 120 |
+
i8 = np.int64()
|
| 121 |
+
u8 = np.uint64()
|
| 122 |
+
f8 = np.float64()
|
| 123 |
+
c16 = np.complex128()
|
| 124 |
+
b_ = np.bool_()
|
| 125 |
+
td = np.timedelta64()
|
| 126 |
+
U = np.str_("1")
|
| 127 |
+
S = np.bytes_("1")
|
| 128 |
+
AR = np.array(1, dtype=np.float64)
|
| 129 |
+
|
| 130 |
+
int(i8)
|
| 131 |
+
int(u8)
|
| 132 |
+
int(f8)
|
| 133 |
+
int(b_)
|
| 134 |
+
int(td)
|
| 135 |
+
int(U)
|
| 136 |
+
int(S)
|
| 137 |
+
int(AR)
|
| 138 |
+
with pytest.warns(np.ComplexWarning):
|
| 139 |
+
int(c16)
|
| 140 |
+
|
| 141 |
+
float(i8)
|
| 142 |
+
float(u8)
|
| 143 |
+
float(f8)
|
| 144 |
+
float(b_)
|
| 145 |
+
float(td)
|
| 146 |
+
float(U)
|
| 147 |
+
float(S)
|
| 148 |
+
float(AR)
|
| 149 |
+
with pytest.warns(np.ComplexWarning):
|
| 150 |
+
float(c16)
|
| 151 |
+
|
| 152 |
+
complex(i8)
|
| 153 |
+
complex(u8)
|
| 154 |
+
complex(f8)
|
| 155 |
+
complex(c16)
|
| 156 |
+
complex(b_)
|
| 157 |
+
complex(td)
|
| 158 |
+
complex(U)
|
| 159 |
+
complex(AR)
|
| 160 |
+
|
| 161 |
+
|
| 162 |
+
# Misc
|
| 163 |
+
c16.dtype
|
| 164 |
+
c16.real
|
| 165 |
+
c16.imag
|
| 166 |
+
c16.real.real
|
| 167 |
+
c16.real.imag
|
| 168 |
+
c16.ndim
|
| 169 |
+
c16.size
|
| 170 |
+
c16.itemsize
|
| 171 |
+
c16.shape
|
| 172 |
+
c16.strides
|
| 173 |
+
c16.squeeze()
|
| 174 |
+
c16.byteswap()
|
| 175 |
+
c16.transpose()
|
| 176 |
+
|
| 177 |
+
# Aliases
|
| 178 |
+
np.string_()
|
| 179 |
+
|
| 180 |
+
np.byte()
|
| 181 |
+
np.short()
|
| 182 |
+
np.intc()
|
| 183 |
+
np.intp()
|
| 184 |
+
np.int_()
|
| 185 |
+
np.longlong()
|
| 186 |
+
|
| 187 |
+
np.ubyte()
|
| 188 |
+
np.ushort()
|
| 189 |
+
np.uintc()
|
| 190 |
+
np.uintp()
|
| 191 |
+
np.uint()
|
| 192 |
+
np.ulonglong()
|
| 193 |
+
|
| 194 |
+
np.half()
|
| 195 |
+
np.single()
|
| 196 |
+
np.double()
|
| 197 |
+
np.float_()
|
| 198 |
+
np.longdouble()
|
| 199 |
+
np.longfloat()
|
| 200 |
+
|
| 201 |
+
np.csingle()
|
| 202 |
+
np.singlecomplex()
|
| 203 |
+
np.cdouble()
|
| 204 |
+
np.complex_()
|
| 205 |
+
np.cfloat()
|
| 206 |
+
np.clongdouble()
|
| 207 |
+
np.clongfloat()
|
| 208 |
+
np.longcomplex()
|
| 209 |
+
|
| 210 |
+
b.item()
|
| 211 |
+
i8.item()
|
| 212 |
+
u8.item()
|
| 213 |
+
f8.item()
|
| 214 |
+
c16.item()
|
| 215 |
+
U.item()
|
| 216 |
+
S.item()
|
| 217 |
+
|
| 218 |
+
b.tolist()
|
| 219 |
+
i8.tolist()
|
| 220 |
+
u8.tolist()
|
| 221 |
+
f8.tolist()
|
| 222 |
+
c16.tolist()
|
| 223 |
+
U.tolist()
|
| 224 |
+
S.tolist()
|
| 225 |
+
|
| 226 |
+
b.ravel()
|
| 227 |
+
i8.ravel()
|
| 228 |
+
u8.ravel()
|
| 229 |
+
f8.ravel()
|
| 230 |
+
c16.ravel()
|
| 231 |
+
U.ravel()
|
| 232 |
+
S.ravel()
|
| 233 |
+
|
| 234 |
+
b.flatten()
|
| 235 |
+
i8.flatten()
|
| 236 |
+
u8.flatten()
|
| 237 |
+
f8.flatten()
|
| 238 |
+
c16.flatten()
|
| 239 |
+
U.flatten()
|
| 240 |
+
S.flatten()
|
| 241 |
+
|
| 242 |
+
b.reshape(1)
|
| 243 |
+
i8.reshape(1)
|
| 244 |
+
u8.reshape(1)
|
| 245 |
+
f8.reshape(1)
|
| 246 |
+
c16.reshape(1)
|
| 247 |
+
U.reshape(1)
|
| 248 |
+
S.reshape(1)
|
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/ufunclike.py
ADDED
|
@@ -0,0 +1,46 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from __future__ import annotations
|
| 2 |
+
from typing import Any
|
| 3 |
+
import numpy as np
|
| 4 |
+
|
| 5 |
+
|
| 6 |
+
class Object:
|
| 7 |
+
def __ceil__(self) -> Object:
|
| 8 |
+
return self
|
| 9 |
+
|
| 10 |
+
def __floor__(self) -> Object:
|
| 11 |
+
return self
|
| 12 |
+
|
| 13 |
+
def __ge__(self, value: object) -> bool:
|
| 14 |
+
return True
|
| 15 |
+
|
| 16 |
+
def __array__(self) -> np.ndarray[Any, np.dtype[np.object_]]:
|
| 17 |
+
ret = np.empty((), dtype=object)
|
| 18 |
+
ret[()] = self
|
| 19 |
+
return ret
|
| 20 |
+
|
| 21 |
+
|
| 22 |
+
AR_LIKE_b = [True, True, False]
|
| 23 |
+
AR_LIKE_u = [np.uint32(1), np.uint32(2), np.uint32(3)]
|
| 24 |
+
AR_LIKE_i = [1, 2, 3]
|
| 25 |
+
AR_LIKE_f = [1.0, 2.0, 3.0]
|
| 26 |
+
AR_LIKE_O = [Object(), Object(), Object()]
|
| 27 |
+
AR_U: np.ndarray[Any, np.dtype[np.str_]] = np.zeros(3, dtype="U5")
|
| 28 |
+
|
| 29 |
+
np.fix(AR_LIKE_b)
|
| 30 |
+
np.fix(AR_LIKE_u)
|
| 31 |
+
np.fix(AR_LIKE_i)
|
| 32 |
+
np.fix(AR_LIKE_f)
|
| 33 |
+
np.fix(AR_LIKE_O)
|
| 34 |
+
np.fix(AR_LIKE_f, out=AR_U)
|
| 35 |
+
|
| 36 |
+
np.isposinf(AR_LIKE_b)
|
| 37 |
+
np.isposinf(AR_LIKE_u)
|
| 38 |
+
np.isposinf(AR_LIKE_i)
|
| 39 |
+
np.isposinf(AR_LIKE_f)
|
| 40 |
+
np.isposinf(AR_LIKE_f, out=AR_U)
|
| 41 |
+
|
| 42 |
+
np.isneginf(AR_LIKE_b)
|
| 43 |
+
np.isneginf(AR_LIKE_u)
|
| 44 |
+
np.isneginf(AR_LIKE_i)
|
| 45 |
+
np.isneginf(AR_LIKE_f)
|
| 46 |
+
np.isneginf(AR_LIKE_f, out=AR_U)
|
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/warnings_and_errors.py
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import numpy as np
|
| 2 |
+
|
| 3 |
+
np.AxisError("test")
|
| 4 |
+
np.AxisError(1, ndim=2)
|
| 5 |
+
np.AxisError(1, ndim=2, msg_prefix="error")
|
| 6 |
+
np.AxisError(1, ndim=2, msg_prefix=None)
|