JinghuiLuAstronaut commited on
Commit
34f75b1
·
verified ·
1 Parent(s): f6ebfae

Add files using upload-large-folder tool

Browse files
Files changed (20) hide show
  1. 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
  2. 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
  3. 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
  4. 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
  5. 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
  6. 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
  7. LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/array_constructors.py +137 -0
  8. LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/array_like.py +41 -0
  9. LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/arrayterator.py +27 -0
  10. LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/comparisons.py +301 -0
  11. LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/flatiter.py +16 -0
  12. LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/fromnumeric.py +260 -0
  13. LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/index_tricks.py +64 -0
  14. LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/literal.py +47 -0
  15. LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/modules.py +42 -0
  16. LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/multiarray.py +76 -0
  17. LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/ndarray_conversion.py +94 -0
  18. LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/scalars.py +248 -0
  19. LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/ufunclike.py +46 -0
  20. 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)