JinghuiLuAstronaut commited on
Commit
f89dd0a
·
verified ·
1 Parent(s): 72264d1

Add files using upload-large-folder tool

Browse files
Files changed (20) hide show
  1. LTA_openwebtext_dualt/logs/lta_lm1b_classic_dirichlet_len256_gbs512_4gpu_10k_save1k_20260523_driver.log +0 -0
  2. LTA_openwebtext_dualt/logs/lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260526_watcher.sh +117 -0
  3. LTA_openwebtext_dualt/logs/lta_owt_gpt2cached_len1024_rollout1_p05_b64_mlpckpt2_bench4gpu_20260513_161437.log +150 -0
  4. LTA_openwebtext_dualt/logs/owt_lowk64plus_cleanbridge_2k_watcher.nohup +29 -0
  5. LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/arrayprint.py +37 -0
  6. LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/numeric.py +90 -0
  7. LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/random.py +1499 -0
  8. LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/simple.py +165 -0
  9. LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/reveal/arithmetic.pyi +516 -0
  10. LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/reveal/array_constructors.pyi +221 -0
  11. LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/reveal/datasource.pyi +29 -0
  12. LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/reveal/fromnumeric.pyi +305 -0
  13. LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/reveal/memmap.pyi +25 -0
  14. LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/reveal/mod.pyi +148 -0
  15. LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/reveal/ndarray_shape_manipulation.pyi +44 -0
  16. LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/reveal/nditer.pyi +55 -0
  17. LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/reveal/warnings_and_errors.pyi +16 -0
  18. LTA_openwebtext_dualt/mini_owt_logdirichlet/logs/decode1_4_uniform_online_20260610/lr2e3_online_latest_decode1_4_uniform.log +51 -0
  19. LTA_openwebtext_dualt/mini_owt_logdirichlet/logs/decode1_4_uniform_online_20260610/lr6e4_online_latest_decode1_4_uniform.log +51 -0
  20. LTA_openwebtext_dualt/mini_owt_logdirichlet/logs/infer_lr3e3_170k_logitnorm64_focused_20260612/gpu3.log +352 -0
LTA_openwebtext_dualt/logs/lta_lm1b_classic_dirichlet_len256_gbs512_4gpu_10k_save1k_20260523_driver.log ADDED
The diff for this file is too large to render. See raw diff
 
LTA_openwebtext_dualt/logs/lta_owt_bert_absrope_time4_dirichlet_len1024_C1_to_1024_mask1_sameT_gbs512_b32_8gpu_1m_save10k_20260526_watcher.sh ADDED
@@ -0,0 +1,117 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+ set -euo pipefail
3
+
4
+ cd /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt
5
+ export PYTHONPATH="$(pwd)${PYTHONPATH:+:$PYTHONPATH}"
6
+ export TOKENIZERS_PARALLELISM=false
7
+ export PYTHONUNBUFFERED=1
8
+
9
+ : "${RUN_DIR:?RUN_DIR is required}"
10
+ : "${OUT_BASE:?OUT_BASE is required}"
11
+ : "${LOG_DIR:?LOG_DIR is required}"
12
+ : "${TOKENIZER_PATH:?TOKENIZER_PATH is required}"
13
+ : "${SCORER:?SCORER is required}"
14
+
15
+ RUN_STEM="$(basename "${RUN_DIR}")"
16
+ TEMP_TAG="${ENDPOINT_TEMP//./p}"
17
+ PROCESSED_FILE="${LOG_DIR}/processed_${RUN_STEM}_steps${STEPS}_c${CMIN}_${CMAX}_gumbel_t${TEMP_TAG}_n${N_SAMPLES}.txt"
18
+
19
+ mkdir -p "${OUT_BASE}" "${LOG_DIR}"
20
+ touch "${PROCESSED_FILE}"
21
+
22
+ echo "[watch-gumbel] run_dir=${RUN_DIR}"
23
+ echo "[watch-gumbel] out_base=${OUT_BASE}"
24
+ echo "[watch-gumbel] interval=${STEP_INTERVAL} max_len=${MAX_LEN} steps=${STEPS} c=${CMIN}->${CMAX} decode_mode=${DECODE_MODE:-sde_gumbel} temp=${ENDPOINT_TEMP} top_p=${ENDPOINT_TOP_P} tau=${GUMBEL_TAU_START}->${GUMBEL_TAU_END} n=${N_SAMPLES}"
25
+
26
+ while true; do
27
+ shopt -s nullglob
28
+ ckpts=("${RUN_DIR}"/step_*.pt)
29
+ shopt -u nullglob
30
+
31
+ if (( ${#ckpts[@]} == 0 )); then
32
+ echo "[watch-gumbel] $(date +%F_%T) no ckpt yet"
33
+ sleep "${SLEEP_SECONDS}"
34
+ continue
35
+ fi
36
+
37
+ printf "%s\n" "${ckpts[@]}" | sort | while read -r ckpt; do
38
+ base="$(basename "${ckpt}")"
39
+ step="${base#step_}"
40
+ step="${step%.pt}"
41
+ step_num=$((10#${step}))
42
+ if (( step_num % STEP_INTERVAL != 0 )); then
43
+ continue
44
+ fi
45
+ if grep -Fxq "${ckpt}" "${PROCESSED_FILE}"; then
46
+ continue
47
+ fi
48
+
49
+ out_dir="${OUT_BASE}/step_${step}"
50
+ log_file="${LOG_DIR}/infer_${RUN_STEM}_step_${step}.log"
51
+ mkdir -p "${out_dir}"
52
+
53
+ echo "[watch-gumbel] $(date +%F_%T) infer ${ckpt} -> ${out_dir}" | tee -a "${log_file}"
54
+ if [[ "${DECODE_MODE:-sde_gumbel}" == "dual_line_probe" ]]; then
55
+ CUDA_VISIBLE_DEVICES="${WATCH_CUDA_VISIBLE_DEVICES}" python scripts/infer_softkl_decode_probe.py \
56
+ --checkpoint "${ckpt}" \
57
+ --tokenizer_path "${TOKENIZER_PATH}" \
58
+ --scorer "${SCORER}" \
59
+ --score \
60
+ --out_dir "${out_dir}" \
61
+ --max_lens "${MAX_LEN}" \
62
+ --n_samples "${N_SAMPLES}" \
63
+ --batch_size "${DECODE_BATCH}" \
64
+ --steps "${STEPS}" \
65
+ --decode_rule dual_line_resample \
66
+ --c_min "${CMIN}" \
67
+ --c_max "${CMAX}" \
68
+ --input_noise_dirichlet_concentration "${CMIN}" \
69
+ --anchor_mode state \
70
+ --model_t_mode flow \
71
+ --time_schedule uniform \
72
+ --support_power 1.0 \
73
+ --semantic_power "${DUAL_SEMANTIC_POWER}" \
74
+ --early_temp "${DUAL_EARLY_TEMP}" \
75
+ --late_temp "${DUAL_LATE_TEMP}" \
76
+ --temp_end "${DUAL_TEMP_END}" \
77
+ --temp_power "${DUAL_TEMP_POWER}" \
78
+ --final_from blend \
79
+ --final_decode argmax \
80
+ --seed 20260524 \
81
+ 2>&1 | tee -a "${log_file}"
82
+ else
83
+ CUDA_VISIBLE_DEVICES="${WATCH_CUDA_VISIBLE_DEVICES}" python scripts/eval_lm1b_c1024_fullycoupled_sde_genppl.py \
84
+ --checkpoint "${ckpt}" \
85
+ --tokenizer_path "${TOKENIZER_PATH}" \
86
+ --scorer "${SCORER}" \
87
+ --out_dir "${out_dir}" \
88
+ --n_samples "${N_SAMPLES}" \
89
+ --max_len "${MAX_LEN}" \
90
+ --steps "${STEPS}" \
91
+ --batch_size "${DECODE_BATCH}" \
92
+ --score_batch "${SCORE_BATCH}" \
93
+ --score_max_length "${SCORE_MAX_LENGTH}" \
94
+ --concentration_min "${CMIN}" \
95
+ --concentration_max "${CMAX}" \
96
+ --endpoint_temp "${ENDPOINT_TEMP}" \
97
+ --endpoint_projection gumbel_softmax \
98
+ --endpoint_top_p "${ENDPOINT_TOP_P}" \
99
+ --gumbel_tau_start "${GUMBEL_TAU_START}" \
100
+ --gumbel_tau_end "${GUMBEL_TAU_END}" \
101
+ --model_t_mode support_t \
102
+ --mean_mode endpoint_only \
103
+ --semantic_power 1.0 \
104
+ --noise_init dirichlet \
105
+ --noise_dirichlet_concentration "${CMIN}" \
106
+ --sde_resample dirichlet \
107
+ --final_from blend_0.5 \
108
+ --seed 20260524 \
109
+ 2>&1 | tee -a "${log_file}"
110
+ fi
111
+
112
+ echo "${ckpt}" >> "${PROCESSED_FILE}"
113
+ echo "[watch-gumbel] $(date +%F_%T) done step_${step}" | tee -a "${log_file}"
114
+ done
115
+
116
+ sleep "${SLEEP_SECONDS}"
117
+ done
LTA_openwebtext_dualt/logs/lta_owt_gpt2cached_len1024_rollout1_p05_b64_mlpckpt2_bench4gpu_20260513_161437.log ADDED
@@ -0,0 +1,150 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ NCCL version 2.25.1+cuda12.8
2
+ {
3
+ "device": "cuda:0",
4
+ "rank": 0,
5
+ "world_size": 4,
6
+ "samples": "owt_cached_chunks:8734897",
7
+ "vocab_size": 50257,
8
+ "tokenizer_vocab_size": 50257,
9
+ "save_dir": "runs/lta_owt_gpt2cached_len1024_rollout1_p05_b64_mlpckpt2_bench4gpu_20260513_161437",
10
+ "batch_size": 64,
11
+ "grad_accum": 1,
12
+ "effective_batch_size": 256,
13
+ "global_batch_size": 256,
14
+ "lr_schedule": "cosine",
15
+ "optimizer": "adamw",
16
+ "warmup_steps": 2,
17
+ "min_lr": 6e-05,
18
+ "weight_decay": 0.1,
19
+ "adamw_param_groups": "nanogpt",
20
+ "adam_beta1": 0.9,
21
+ "adam_beta2": 0.95,
22
+ "adam_eps": 1e-08,
23
+ "muon_momentum": 0.95,
24
+ "muon_ns_steps": 5,
25
+ "muon_update_scale": 1.0,
26
+ "ema_decay": 0.0,
27
+ "ema_start_step": 0,
28
+ "model_type": "ddit",
29
+ "dual_t": true,
30
+ "corrupt_t_mode": "same",
31
+ "corrupt_min_t": 0.0,
32
+ "corrupt_max_t": 1.0,
33
+ "prefix_block_prob": 0.0,
34
+ "prefix_block_len": 128,
35
+ "dirichlet_endpoint_mode": "categorical_dual_t",
36
+ "dirichlet_semantic_t_mode": "same",
37
+ "dirichlet_semantic_t_value": 0.0,
38
+ "categorical_wrong_from_full_vocab": true,
39
+ "categorical_wrong_from_batch_valid_tokens": false,
40
+ "mask_mixture_original_prob": 0.0,
41
+ "mask_mixture_lowk_prob": 0.0,
42
+ "mask_mixture_lowcorrupt_prob": 0.0,
43
+ "mask_mixture_block_prob": 0.0,
44
+ "mask_mixture_all_prob": 0.0,
45
+ "mask_mixture_lowk_clean_tokens": "1,2,4,8,16,32,64",
46
+ "mask_mixture_lowcorrupt_tokens": "1,2,4,8,16,32,64",
47
+ "mask_mixture_block_tokens": "64,128",
48
+ "simplex_bridge_sampler": "dirichlet",
49
+ "logistic_normal_sigma_min": 0.18,
50
+ "logistic_normal_sigma_max": 2.2,
51
+ "logistic_normal_tau_min": 0.65,
52
+ "logistic_normal_tau_max": 1.15,
53
+ "torch_compile": false,
54
+ "compile_mode": "max-autotune",
55
+ "state_format": "prob",
56
+ "target_loss": "hard_ce",
57
+ "meanflow_weight": 0.0,
58
+ "rollout_train_prob": 0.5,
59
+ "rollout_train_steps": 1,
60
+ "rollout_train_infer_steps": 64,
61
+ "rollout_train_temp": 1.45,
62
+ "rollout_train_max_gamma": 1.0,
63
+ "rollout_train_corrupt_only": true,
64
+ "rollout_train_samplewise": false,
65
+ "rollout_train_compute_always": false,
66
+ "bridge_noise_init": "logistic_normal",
67
+ "noise_sigma": -1.0,
68
+ "allow_tf32": true,
69
+ "activation_checkpointing": true,
70
+ "activation_checkpoint_interval": 1,
71
+ "activation_checkpoint_scope": "mlp",
72
+ "ddp_static_graph": false,
73
+ "ddp_gradient_as_bucket_view": true,
74
+ "blocking_data_transfer": false,
75
+ "dataloader_prefetch_factor": 4,
76
+ "full_train_stats": false,
77
+ "record_pad_truncate": false,
78
+ "record_add_eos": false,
79
+ "record_add_special_tokens": false,
80
+ "record_pad_token": "pad",
81
+ "record_shuffle_buffer": 10000,
82
+ "wrap": true,
83
+ "wrap_mode": "stream",
84
+ "wrap_record_buffer_size": 200,
85
+ "owt_cached_chunks": true,
86
+ "owt_chunk_cache_dir": "/e2e-data/evad-tech-vla/wanghan58/data/small_benchmarks/langflow_2604_11748/openwebtext_lta_cached_chunks/gpt2_len1024_train_minus_100k",
87
+ "owt_chunk_cache_rebuild": false,
88
+ "owt_chunk_cache_write_batch": 4096,
89
+ "owt_exact_repeat_per_chunk": 0,
90
+ "online_chunk_shuffle": false,
91
+ "online_chunk_shuffle_buffer": 10000,
92
+ "openwebtext_split": "train_minus_100k",
93
+ "detokenizer": "auto",
94
+ "resolved_detokenizer": null,
95
+ "num_workers": 8,
96
+ "latest_every": 0,
97
+ "resume_path": ""
98
+ }
99
+ [rank0]: Traceback (most recent call last):
100
+ [rank0]: File "/e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/train.py", line 1546, in <module>
101
+ [rank0]: main()
102
+ [rank0]: File "/e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/train.py", line 1465, in main
103
+ [rank0]: loss.backward()
104
+ [rank0]: File "/usr/local/lib/python3.12/dist-packages/torch/_tensor.py", line 648, in backward
105
+ [rank0]: torch.autograd.backward(
106
+ [rank0]: File "/usr/local/lib/python3.12/dist-packages/torch/autograd/__init__.py", line 347, in backward
107
+ [rank0]: _engine_run_backward(
108
+ [rank0]: File "/usr/local/lib/python3.12/dist-packages/torch/autograd/graph.py", line 823, in _engine_run_backward
109
+ [rank0]: return Variable._execution_engine.run_backward( # Calls into the C++ engine to run the backward pass
110
+ [rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
111
+ [rank0]: RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
112
+ [rank2]: Traceback (most recent call last):
113
+ [rank2]: File "/e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/train.py", line 1546, in <module>
114
+ [rank2]: main()
115
+ [rank2]: File "/e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/train.py", line 1465, in main
116
+ [rank2]: loss.backward()
117
+ [rank2]: File "/usr/local/lib/python3.12/dist-packages/torch/_tensor.py", line 648, in backward
118
+ [rank2]: torch.autograd.backward(
119
+ [rank2]: File "/usr/local/lib/python3.12/dist-packages/torch/autograd/__init__.py", line 347, in backward
120
+ [rank2]: _engine_run_backward(
121
+ [rank2]: File "/usr/local/lib/python3.12/dist-packages/torch/autograd/graph.py", line 823, in _engine_run_backward
122
+ [rank2]: return Variable._execution_engine.run_backward( # Calls into the C++ engine to run the backward pass
123
+ [rank2]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
124
+ [rank2]: RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
125
+ [rank3]: Traceback (most recent call last):
126
+ [rank3]: File "/e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/train.py", line 1546, in <module>
127
+ [rank3]: main()
128
+ [rank3]: File "/e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/train.py", line 1465, in main
129
+ [rank3]: loss.backward()
130
+ [rank3]: File "/usr/local/lib/python3.12/dist-packages/torch/_tensor.py", line 648, in backward
131
+ [rank3]: torch.autograd.backward(
132
+ [rank3]: File "/usr/local/lib/python3.12/dist-packages/torch/autograd/__init__.py", line 347, in backward
133
+ [rank3]: _engine_run_backward(
134
+ [rank3]: File "/usr/local/lib/python3.12/dist-packages/torch/autograd/graph.py", line 823, in _engine_run_backward
135
+ [rank3]: return Variable._execution_engine.run_backward( # Calls into the C++ engine to run the backward pass
136
+ [rank3]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
137
+ [rank3]: RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
138
+ [rank1]: Traceback (most recent call last):
139
+ [rank1]: File "/e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/train.py", line 1546, in <module>
140
+ [rank1]: main()
141
+ [rank1]: File "/e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/train.py", line 1465, in main
142
+ [rank1]: loss.backward()
143
+ [rank1]: File "/usr/local/lib/python3.12/dist-packages/torch/_tensor.py", line 648, in backward
144
+ [rank1]: torch.autograd.backward(
145
+ [rank1]: File "/usr/local/lib/python3.12/dist-packages/torch/autograd/__init__.py", line 347, in backward
146
+ [rank1]: _engine_run_backward(
147
+ [rank1]: File "/usr/local/lib/python3.12/dist-packages/torch/autograd/graph.py", line 823, in _engine_run_backward
148
+ [rank1]: return Variable._execution_engine.run_backward( # Calls into the C++ engine to run the backward pass
149
+ [rank1]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
150
+ [rank1]: RuntimeError: t() expects a tensor with <= 2 dimensions, but self is 3D
LTA_openwebtext_dualt/logs/owt_lowk64plus_cleanbridge_2k_watcher.nohup ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [watcher] start 2026-05-12T00:37:29+00:00 run_dir=runs/lta_owt_c1024_len1024_t0to1_lowk64plus_cleanbridge_buf1000_gbs128_4gpu_2k
2
+ [watcher] poll 2026-05-12T00:46:32+00:00 step=500 last=-1
3
+ [watcher] infer step=500 out=docs/lta_samples/metrics_20260512/owt_lowk64plus_cleanbridge_2k_rolling_sync_probe/step500_sync_probe_n16
4
+ [watcher] done step=500
5
+ [watcher] poll 2026-05-12T00:48:43+00:00 step=500 last=500
6
+ [watcher] poll 2026-05-12T00:50:16+00:00 step=500 last=500
7
+ [watcher] poll 2026-05-12T00:51:49+00:00 step=500 last=500
8
+ [watcher] poll 2026-05-12T00:53:23+00:00 step=500 last=500
9
+ [watcher] poll 2026-05-12T00:54:56+00:00 step=500 last=500
10
+ [watcher] poll 2026-05-12T00:56:29+00:00 step=1000 last=500
11
+ [watcher] infer step=1000 out=docs/lta_samples/metrics_20260512/owt_lowk64plus_cleanbridge_2k_rolling_sync_probe/step1000_sync_probe_n16
12
+ [watcher] done step=1000
13
+ [watcher] poll 2026-05-12T00:58:40+00:00 step=1000 last=1000
14
+ [watcher] poll 2026-05-12T01:00:13+00:00 step=1000 last=1000
15
+ [watcher] poll 2026-05-12T01:01:46+00:00 step=1000 last=1000
16
+ [watcher] poll 2026-05-12T01:03:20+00:00 step=1000 last=1000
17
+ [watcher] poll 2026-05-12T01:04:53+00:00 step=1500 last=1000
18
+ [watcher] infer step=1500 out=docs/lta_samples/metrics_20260512/owt_lowk64plus_cleanbridge_2k_rolling_sync_probe/step1500_sync_probe_n16
19
+ [watcher] done step=1500
20
+ [watcher] poll 2026-05-12T01:07:04+00:00 step=1500 last=1500
21
+ [watcher] poll 2026-05-12T01:08:37+00:00 step=1500 last=1500
22
+ [watcher] poll 2026-05-12T01:10:10+00:00 step=1500 last=1500
23
+ [watcher] poll 2026-05-12T01:11:44+00:00 step=1500 last=1500
24
+ [watcher] poll 2026-05-12T01:13:17+00:00 step=1500 last=1500
25
+ [watcher] poll 2026-05-12T01:14:50+00:00 step=1500 last=1500
26
+ [watcher] poll 2026-05-12T01:16:23+00:00 step=2000 last=1500
27
+ [watcher] infer step=2000 out=docs/lta_samples/metrics_20260512/owt_lowk64plus_cleanbridge_2k_rolling_sync_probe/step2000_sync_probe_n16
28
+ [watcher] done step=2000
29
+ [watcher] reached max step 2000; exiting
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/arrayprint.py ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import numpy as np
2
+
3
+ AR = np.arange(10)
4
+ AR.setflags(write=False)
5
+
6
+ with np.printoptions():
7
+ np.set_printoptions(
8
+ precision=1,
9
+ threshold=2,
10
+ edgeitems=3,
11
+ linewidth=4,
12
+ suppress=False,
13
+ nanstr="Bob",
14
+ infstr="Bill",
15
+ formatter={},
16
+ sign="+",
17
+ floatmode="unique",
18
+ )
19
+ np.get_printoptions()
20
+ str(AR)
21
+
22
+ np.array2string(
23
+ AR,
24
+ max_line_width=5,
25
+ precision=2,
26
+ suppress_small=True,
27
+ separator=";",
28
+ prefix="test",
29
+ threshold=5,
30
+ floatmode="fixed",
31
+ suffix="?",
32
+ legacy="1.13",
33
+ )
34
+ np.format_float_scientific(1, precision=5)
35
+ np.format_float_positional(1, trim="k")
36
+ np.array_repr(AR)
37
+ np.array_str(AR)
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/numeric.py ADDED
@@ -0,0 +1,90 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """
2
+ Tests for :mod:`numpy.core.numeric`.
3
+
4
+ Does not include tests which fall under ``array_constructors``.
5
+
6
+ """
7
+
8
+ from __future__ import annotations
9
+
10
+ import numpy as np
11
+
12
+ class SubClass(np.ndarray):
13
+ ...
14
+
15
+ i8 = np.int64(1)
16
+
17
+ A = np.arange(27).reshape(3, 3, 3)
18
+ B: list[list[list[int]]] = A.tolist()
19
+ C = np.empty((27, 27)).view(SubClass)
20
+
21
+ np.count_nonzero(i8)
22
+ np.count_nonzero(A)
23
+ np.count_nonzero(B)
24
+ np.count_nonzero(A, keepdims=True)
25
+ np.count_nonzero(A, axis=0)
26
+
27
+ np.isfortran(i8)
28
+ np.isfortran(A)
29
+
30
+ np.argwhere(i8)
31
+ np.argwhere(A)
32
+
33
+ np.flatnonzero(i8)
34
+ np.flatnonzero(A)
35
+
36
+ np.correlate(B[0][0], A.ravel(), mode="valid")
37
+ np.correlate(A.ravel(), A.ravel(), mode="same")
38
+
39
+ np.convolve(B[0][0], A.ravel(), mode="valid")
40
+ np.convolve(A.ravel(), A.ravel(), mode="same")
41
+
42
+ np.outer(i8, A)
43
+ np.outer(B, A)
44
+ np.outer(A, A)
45
+ np.outer(A, A, out=C)
46
+
47
+ np.tensordot(B, A)
48
+ np.tensordot(A, A)
49
+ np.tensordot(A, A, axes=0)
50
+ np.tensordot(A, A, axes=(0, 1))
51
+
52
+ np.isscalar(i8)
53
+ np.isscalar(A)
54
+ np.isscalar(B)
55
+
56
+ np.roll(A, 1)
57
+ np.roll(A, (1, 2))
58
+ np.roll(B, 1)
59
+
60
+ np.rollaxis(A, 0, 1)
61
+
62
+ np.moveaxis(A, 0, 1)
63
+ np.moveaxis(A, (0, 1), (1, 2))
64
+
65
+ np.cross(B, A)
66
+ np.cross(A, A)
67
+
68
+ np.indices([0, 1, 2])
69
+ np.indices([0, 1, 2], sparse=False)
70
+ np.indices([0, 1, 2], sparse=True)
71
+
72
+ np.binary_repr(1)
73
+
74
+ np.base_repr(1)
75
+
76
+ np.allclose(i8, A)
77
+ np.allclose(B, A)
78
+ np.allclose(A, A)
79
+
80
+ np.isclose(i8, A)
81
+ np.isclose(B, A)
82
+ np.isclose(A, A)
83
+
84
+ np.array_equal(i8, A)
85
+ np.array_equal(B, A)
86
+ np.array_equal(A, A)
87
+
88
+ np.array_equiv(i8, A)
89
+ np.array_equiv(B, A)
90
+ np.array_equiv(A, A)
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/random.py ADDED
@@ -0,0 +1,1499 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from __future__ import annotations
2
+
3
+ from typing import Any
4
+ import numpy as np
5
+
6
+ SEED_NONE = None
7
+ SEED_INT = 4579435749574957634658964293569
8
+ SEED_ARR: np.ndarray[Any, np.dtype[np.int64]] = np.array([1, 2, 3, 4], dtype=np.int64)
9
+ SEED_ARRLIKE: list[int] = [1, 2, 3, 4]
10
+ SEED_SEED_SEQ: np.random.SeedSequence = np.random.SeedSequence(0)
11
+ SEED_MT19937: np.random.MT19937 = np.random.MT19937(0)
12
+ SEED_PCG64: np.random.PCG64 = np.random.PCG64(0)
13
+ SEED_PHILOX: np.random.Philox = np.random.Philox(0)
14
+ SEED_SFC64: np.random.SFC64 = np.random.SFC64(0)
15
+
16
+ # default rng
17
+ np.random.default_rng()
18
+ np.random.default_rng(SEED_NONE)
19
+ np.random.default_rng(SEED_INT)
20
+ np.random.default_rng(SEED_ARR)
21
+ np.random.default_rng(SEED_ARRLIKE)
22
+ np.random.default_rng(SEED_SEED_SEQ)
23
+ np.random.default_rng(SEED_MT19937)
24
+ np.random.default_rng(SEED_PCG64)
25
+ np.random.default_rng(SEED_PHILOX)
26
+ np.random.default_rng(SEED_SFC64)
27
+
28
+ # Seed Sequence
29
+ np.random.SeedSequence(SEED_NONE)
30
+ np.random.SeedSequence(SEED_INT)
31
+ np.random.SeedSequence(SEED_ARR)
32
+ np.random.SeedSequence(SEED_ARRLIKE)
33
+
34
+ # Bit Generators
35
+ np.random.MT19937(SEED_NONE)
36
+ np.random.MT19937(SEED_INT)
37
+ np.random.MT19937(SEED_ARR)
38
+ np.random.MT19937(SEED_ARRLIKE)
39
+ np.random.MT19937(SEED_SEED_SEQ)
40
+
41
+ np.random.PCG64(SEED_NONE)
42
+ np.random.PCG64(SEED_INT)
43
+ np.random.PCG64(SEED_ARR)
44
+ np.random.PCG64(SEED_ARRLIKE)
45
+ np.random.PCG64(SEED_SEED_SEQ)
46
+
47
+ np.random.Philox(SEED_NONE)
48
+ np.random.Philox(SEED_INT)
49
+ np.random.Philox(SEED_ARR)
50
+ np.random.Philox(SEED_ARRLIKE)
51
+ np.random.Philox(SEED_SEED_SEQ)
52
+
53
+ np.random.SFC64(SEED_NONE)
54
+ np.random.SFC64(SEED_INT)
55
+ np.random.SFC64(SEED_ARR)
56
+ np.random.SFC64(SEED_ARRLIKE)
57
+ np.random.SFC64(SEED_SEED_SEQ)
58
+
59
+ seed_seq: np.random.bit_generator.SeedSequence = np.random.SeedSequence(SEED_NONE)
60
+ seed_seq.spawn(10)
61
+ seed_seq.generate_state(3)
62
+ seed_seq.generate_state(3, "u4")
63
+ seed_seq.generate_state(3, "uint32")
64
+ seed_seq.generate_state(3, "u8")
65
+ seed_seq.generate_state(3, "uint64")
66
+ seed_seq.generate_state(3, np.uint32)
67
+ seed_seq.generate_state(3, np.uint64)
68
+
69
+
70
+ def_gen: np.random.Generator = np.random.default_rng()
71
+
72
+ D_arr_0p1: np.ndarray[Any, np.dtype[np.float64]] = np.array([0.1])
73
+ D_arr_0p5: np.ndarray[Any, np.dtype[np.float64]] = np.array([0.5])
74
+ D_arr_0p9: np.ndarray[Any, np.dtype[np.float64]] = np.array([0.9])
75
+ D_arr_1p5: np.ndarray[Any, np.dtype[np.float64]] = np.array([1.5])
76
+ I_arr_10: np.ndarray[Any, np.dtype[np.int_]] = np.array([10], dtype=np.int_)
77
+ I_arr_20: np.ndarray[Any, np.dtype[np.int_]] = np.array([20], dtype=np.int_)
78
+ D_arr_like_0p1: list[float] = [0.1]
79
+ D_arr_like_0p5: list[float] = [0.5]
80
+ D_arr_like_0p9: list[float] = [0.9]
81
+ D_arr_like_1p5: list[float] = [1.5]
82
+ I_arr_like_10: list[int] = [10]
83
+ I_arr_like_20: list[int] = [20]
84
+ D_2D_like: list[list[float]] = [[1, 2], [2, 3], [3, 4], [4, 5.1]]
85
+ D_2D: np.ndarray[Any, np.dtype[np.float64]] = np.array(D_2D_like)
86
+
87
+ S_out: np.ndarray[Any, np.dtype[np.float32]] = np.empty(1, dtype=np.float32)
88
+ D_out: np.ndarray[Any, np.dtype[np.float64]] = np.empty(1)
89
+
90
+ def_gen.standard_normal()
91
+ def_gen.standard_normal(dtype=np.float32)
92
+ def_gen.standard_normal(dtype="float32")
93
+ def_gen.standard_normal(dtype="double")
94
+ def_gen.standard_normal(dtype=np.float64)
95
+ def_gen.standard_normal(size=None)
96
+ def_gen.standard_normal(size=1)
97
+ def_gen.standard_normal(size=1, dtype=np.float32)
98
+ def_gen.standard_normal(size=1, dtype="f4")
99
+ def_gen.standard_normal(size=1, dtype="float32", out=S_out)
100
+ def_gen.standard_normal(dtype=np.float32, out=S_out)
101
+ def_gen.standard_normal(size=1, dtype=np.float64)
102
+ def_gen.standard_normal(size=1, dtype="float64")
103
+ def_gen.standard_normal(size=1, dtype="f8")
104
+ def_gen.standard_normal(out=D_out)
105
+ def_gen.standard_normal(size=1, dtype="float64")
106
+ def_gen.standard_normal(size=1, dtype="float64", out=D_out)
107
+
108
+ def_gen.random()
109
+ def_gen.random(dtype=np.float32)
110
+ def_gen.random(dtype="float32")
111
+ def_gen.random(dtype="double")
112
+ def_gen.random(dtype=np.float64)
113
+ def_gen.random(size=None)
114
+ def_gen.random(size=1)
115
+ def_gen.random(size=1, dtype=np.float32)
116
+ def_gen.random(size=1, dtype="f4")
117
+ def_gen.random(size=1, dtype="float32", out=S_out)
118
+ def_gen.random(dtype=np.float32, out=S_out)
119
+ def_gen.random(size=1, dtype=np.float64)
120
+ def_gen.random(size=1, dtype="float64")
121
+ def_gen.random(size=1, dtype="f8")
122
+ def_gen.random(out=D_out)
123
+ def_gen.random(size=1, dtype="float64")
124
+ def_gen.random(size=1, dtype="float64", out=D_out)
125
+
126
+ def_gen.standard_cauchy()
127
+ def_gen.standard_cauchy(size=None)
128
+ def_gen.standard_cauchy(size=1)
129
+
130
+ def_gen.standard_exponential()
131
+ def_gen.standard_exponential(method="inv")
132
+ def_gen.standard_exponential(dtype=np.float32)
133
+ def_gen.standard_exponential(dtype="float32")
134
+ def_gen.standard_exponential(dtype="double")
135
+ def_gen.standard_exponential(dtype=np.float64)
136
+ def_gen.standard_exponential(size=None)
137
+ def_gen.standard_exponential(size=None, method="inv")
138
+ def_gen.standard_exponential(size=1, method="inv")
139
+ def_gen.standard_exponential(size=1, dtype=np.float32)
140
+ def_gen.standard_exponential(size=1, dtype="f4", method="inv")
141
+ def_gen.standard_exponential(size=1, dtype="float32", out=S_out)
142
+ def_gen.standard_exponential(dtype=np.float32, out=S_out)
143
+ def_gen.standard_exponential(size=1, dtype=np.float64, method="inv")
144
+ def_gen.standard_exponential(size=1, dtype="float64")
145
+ def_gen.standard_exponential(size=1, dtype="f8")
146
+ def_gen.standard_exponential(out=D_out)
147
+ def_gen.standard_exponential(size=1, dtype="float64")
148
+ def_gen.standard_exponential(size=1, dtype="float64", out=D_out)
149
+
150
+ def_gen.zipf(1.5)
151
+ def_gen.zipf(1.5, size=None)
152
+ def_gen.zipf(1.5, size=1)
153
+ def_gen.zipf(D_arr_1p5)
154
+ def_gen.zipf(D_arr_1p5, size=1)
155
+ def_gen.zipf(D_arr_like_1p5)
156
+ def_gen.zipf(D_arr_like_1p5, size=1)
157
+
158
+ def_gen.weibull(0.5)
159
+ def_gen.weibull(0.5, size=None)
160
+ def_gen.weibull(0.5, size=1)
161
+ def_gen.weibull(D_arr_0p5)
162
+ def_gen.weibull(D_arr_0p5, size=1)
163
+ def_gen.weibull(D_arr_like_0p5)
164
+ def_gen.weibull(D_arr_like_0p5, size=1)
165
+
166
+ def_gen.standard_t(0.5)
167
+ def_gen.standard_t(0.5, size=None)
168
+ def_gen.standard_t(0.5, size=1)
169
+ def_gen.standard_t(D_arr_0p5)
170
+ def_gen.standard_t(D_arr_0p5, size=1)
171
+ def_gen.standard_t(D_arr_like_0p5)
172
+ def_gen.standard_t(D_arr_like_0p5, size=1)
173
+
174
+ def_gen.poisson(0.5)
175
+ def_gen.poisson(0.5, size=None)
176
+ def_gen.poisson(0.5, size=1)
177
+ def_gen.poisson(D_arr_0p5)
178
+ def_gen.poisson(D_arr_0p5, size=1)
179
+ def_gen.poisson(D_arr_like_0p5)
180
+ def_gen.poisson(D_arr_like_0p5, size=1)
181
+
182
+ def_gen.power(0.5)
183
+ def_gen.power(0.5, size=None)
184
+ def_gen.power(0.5, size=1)
185
+ def_gen.power(D_arr_0p5)
186
+ def_gen.power(D_arr_0p5, size=1)
187
+ def_gen.power(D_arr_like_0p5)
188
+ def_gen.power(D_arr_like_0p5, size=1)
189
+
190
+ def_gen.pareto(0.5)
191
+ def_gen.pareto(0.5, size=None)
192
+ def_gen.pareto(0.5, size=1)
193
+ def_gen.pareto(D_arr_0p5)
194
+ def_gen.pareto(D_arr_0p5, size=1)
195
+ def_gen.pareto(D_arr_like_0p5)
196
+ def_gen.pareto(D_arr_like_0p5, size=1)
197
+
198
+ def_gen.chisquare(0.5)
199
+ def_gen.chisquare(0.5, size=None)
200
+ def_gen.chisquare(0.5, size=1)
201
+ def_gen.chisquare(D_arr_0p5)
202
+ def_gen.chisquare(D_arr_0p5, size=1)
203
+ def_gen.chisquare(D_arr_like_0p5)
204
+ def_gen.chisquare(D_arr_like_0p5, size=1)
205
+
206
+ def_gen.exponential(0.5)
207
+ def_gen.exponential(0.5, size=None)
208
+ def_gen.exponential(0.5, size=1)
209
+ def_gen.exponential(D_arr_0p5)
210
+ def_gen.exponential(D_arr_0p5, size=1)
211
+ def_gen.exponential(D_arr_like_0p5)
212
+ def_gen.exponential(D_arr_like_0p5, size=1)
213
+
214
+ def_gen.geometric(0.5)
215
+ def_gen.geometric(0.5, size=None)
216
+ def_gen.geometric(0.5, size=1)
217
+ def_gen.geometric(D_arr_0p5)
218
+ def_gen.geometric(D_arr_0p5, size=1)
219
+ def_gen.geometric(D_arr_like_0p5)
220
+ def_gen.geometric(D_arr_like_0p5, size=1)
221
+
222
+ def_gen.logseries(0.5)
223
+ def_gen.logseries(0.5, size=None)
224
+ def_gen.logseries(0.5, size=1)
225
+ def_gen.logseries(D_arr_0p5)
226
+ def_gen.logseries(D_arr_0p5, size=1)
227
+ def_gen.logseries(D_arr_like_0p5)
228
+ def_gen.logseries(D_arr_like_0p5, size=1)
229
+
230
+ def_gen.rayleigh(0.5)
231
+ def_gen.rayleigh(0.5, size=None)
232
+ def_gen.rayleigh(0.5, size=1)
233
+ def_gen.rayleigh(D_arr_0p5)
234
+ def_gen.rayleigh(D_arr_0p5, size=1)
235
+ def_gen.rayleigh(D_arr_like_0p5)
236
+ def_gen.rayleigh(D_arr_like_0p5, size=1)
237
+
238
+ def_gen.standard_gamma(0.5)
239
+ def_gen.standard_gamma(0.5, size=None)
240
+ def_gen.standard_gamma(0.5, dtype="float32")
241
+ def_gen.standard_gamma(0.5, size=None, dtype="float32")
242
+ def_gen.standard_gamma(0.5, size=1)
243
+ def_gen.standard_gamma(D_arr_0p5)
244
+ def_gen.standard_gamma(D_arr_0p5, dtype="f4")
245
+ def_gen.standard_gamma(0.5, size=1, dtype="float32", out=S_out)
246
+ def_gen.standard_gamma(D_arr_0p5, dtype=np.float32, out=S_out)
247
+ def_gen.standard_gamma(D_arr_0p5, size=1)
248
+ def_gen.standard_gamma(D_arr_like_0p5)
249
+ def_gen.standard_gamma(D_arr_like_0p5, size=1)
250
+ def_gen.standard_gamma(0.5, out=D_out)
251
+ def_gen.standard_gamma(D_arr_like_0p5, out=D_out)
252
+ def_gen.standard_gamma(D_arr_like_0p5, size=1)
253
+ def_gen.standard_gamma(D_arr_like_0p5, size=1, out=D_out, dtype=np.float64)
254
+
255
+ def_gen.vonmises(0.5, 0.5)
256
+ def_gen.vonmises(0.5, 0.5, size=None)
257
+ def_gen.vonmises(0.5, 0.5, size=1)
258
+ def_gen.vonmises(D_arr_0p5, 0.5)
259
+ def_gen.vonmises(0.5, D_arr_0p5)
260
+ def_gen.vonmises(D_arr_0p5, 0.5, size=1)
261
+ def_gen.vonmises(0.5, D_arr_0p5, size=1)
262
+ def_gen.vonmises(D_arr_like_0p5, 0.5)
263
+ def_gen.vonmises(0.5, D_arr_like_0p5)
264
+ def_gen.vonmises(D_arr_0p5, D_arr_0p5)
265
+ def_gen.vonmises(D_arr_like_0p5, D_arr_like_0p5)
266
+ def_gen.vonmises(D_arr_0p5, D_arr_0p5, size=1)
267
+ def_gen.vonmises(D_arr_like_0p5, D_arr_like_0p5, size=1)
268
+
269
+ def_gen.wald(0.5, 0.5)
270
+ def_gen.wald(0.5, 0.5, size=None)
271
+ def_gen.wald(0.5, 0.5, size=1)
272
+ def_gen.wald(D_arr_0p5, 0.5)
273
+ def_gen.wald(0.5, D_arr_0p5)
274
+ def_gen.wald(D_arr_0p5, 0.5, size=1)
275
+ def_gen.wald(0.5, D_arr_0p5, size=1)
276
+ def_gen.wald(D_arr_like_0p5, 0.5)
277
+ def_gen.wald(0.5, D_arr_like_0p5)
278
+ def_gen.wald(D_arr_0p5, D_arr_0p5)
279
+ def_gen.wald(D_arr_like_0p5, D_arr_like_0p5)
280
+ def_gen.wald(D_arr_0p5, D_arr_0p5, size=1)
281
+ def_gen.wald(D_arr_like_0p5, D_arr_like_0p5, size=1)
282
+
283
+ def_gen.uniform(0.5, 0.5)
284
+ def_gen.uniform(0.5, 0.5, size=None)
285
+ def_gen.uniform(0.5, 0.5, size=1)
286
+ def_gen.uniform(D_arr_0p5, 0.5)
287
+ def_gen.uniform(0.5, D_arr_0p5)
288
+ def_gen.uniform(D_arr_0p5, 0.5, size=1)
289
+ def_gen.uniform(0.5, D_arr_0p5, size=1)
290
+ def_gen.uniform(D_arr_like_0p5, 0.5)
291
+ def_gen.uniform(0.5, D_arr_like_0p5)
292
+ def_gen.uniform(D_arr_0p5, D_arr_0p5)
293
+ def_gen.uniform(D_arr_like_0p5, D_arr_like_0p5)
294
+ def_gen.uniform(D_arr_0p5, D_arr_0p5, size=1)
295
+ def_gen.uniform(D_arr_like_0p5, D_arr_like_0p5, size=1)
296
+
297
+ def_gen.beta(0.5, 0.5)
298
+ def_gen.beta(0.5, 0.5, size=None)
299
+ def_gen.beta(0.5, 0.5, size=1)
300
+ def_gen.beta(D_arr_0p5, 0.5)
301
+ def_gen.beta(0.5, D_arr_0p5)
302
+ def_gen.beta(D_arr_0p5, 0.5, size=1)
303
+ def_gen.beta(0.5, D_arr_0p5, size=1)
304
+ def_gen.beta(D_arr_like_0p5, 0.5)
305
+ def_gen.beta(0.5, D_arr_like_0p5)
306
+ def_gen.beta(D_arr_0p5, D_arr_0p5)
307
+ def_gen.beta(D_arr_like_0p5, D_arr_like_0p5)
308
+ def_gen.beta(D_arr_0p5, D_arr_0p5, size=1)
309
+ def_gen.beta(D_arr_like_0p5, D_arr_like_0p5, size=1)
310
+
311
+ def_gen.f(0.5, 0.5)
312
+ def_gen.f(0.5, 0.5, size=None)
313
+ def_gen.f(0.5, 0.5, size=1)
314
+ def_gen.f(D_arr_0p5, 0.5)
315
+ def_gen.f(0.5, D_arr_0p5)
316
+ def_gen.f(D_arr_0p5, 0.5, size=1)
317
+ def_gen.f(0.5, D_arr_0p5, size=1)
318
+ def_gen.f(D_arr_like_0p5, 0.5)
319
+ def_gen.f(0.5, D_arr_like_0p5)
320
+ def_gen.f(D_arr_0p5, D_arr_0p5)
321
+ def_gen.f(D_arr_like_0p5, D_arr_like_0p5)
322
+ def_gen.f(D_arr_0p5, D_arr_0p5, size=1)
323
+ def_gen.f(D_arr_like_0p5, D_arr_like_0p5, size=1)
324
+
325
+ def_gen.gamma(0.5, 0.5)
326
+ def_gen.gamma(0.5, 0.5, size=None)
327
+ def_gen.gamma(0.5, 0.5, size=1)
328
+ def_gen.gamma(D_arr_0p5, 0.5)
329
+ def_gen.gamma(0.5, D_arr_0p5)
330
+ def_gen.gamma(D_arr_0p5, 0.5, size=1)
331
+ def_gen.gamma(0.5, D_arr_0p5, size=1)
332
+ def_gen.gamma(D_arr_like_0p5, 0.5)
333
+ def_gen.gamma(0.5, D_arr_like_0p5)
334
+ def_gen.gamma(D_arr_0p5, D_arr_0p5)
335
+ def_gen.gamma(D_arr_like_0p5, D_arr_like_0p5)
336
+ def_gen.gamma(D_arr_0p5, D_arr_0p5, size=1)
337
+ def_gen.gamma(D_arr_like_0p5, D_arr_like_0p5, size=1)
338
+
339
+ def_gen.gumbel(0.5, 0.5)
340
+ def_gen.gumbel(0.5, 0.5, size=None)
341
+ def_gen.gumbel(0.5, 0.5, size=1)
342
+ def_gen.gumbel(D_arr_0p5, 0.5)
343
+ def_gen.gumbel(0.5, D_arr_0p5)
344
+ def_gen.gumbel(D_arr_0p5, 0.5, size=1)
345
+ def_gen.gumbel(0.5, D_arr_0p5, size=1)
346
+ def_gen.gumbel(D_arr_like_0p5, 0.5)
347
+ def_gen.gumbel(0.5, D_arr_like_0p5)
348
+ def_gen.gumbel(D_arr_0p5, D_arr_0p5)
349
+ def_gen.gumbel(D_arr_like_0p5, D_arr_like_0p5)
350
+ def_gen.gumbel(D_arr_0p5, D_arr_0p5, size=1)
351
+ def_gen.gumbel(D_arr_like_0p5, D_arr_like_0p5, size=1)
352
+
353
+ def_gen.laplace(0.5, 0.5)
354
+ def_gen.laplace(0.5, 0.5, size=None)
355
+ def_gen.laplace(0.5, 0.5, size=1)
356
+ def_gen.laplace(D_arr_0p5, 0.5)
357
+ def_gen.laplace(0.5, D_arr_0p5)
358
+ def_gen.laplace(D_arr_0p5, 0.5, size=1)
359
+ def_gen.laplace(0.5, D_arr_0p5, size=1)
360
+ def_gen.laplace(D_arr_like_0p5, 0.5)
361
+ def_gen.laplace(0.5, D_arr_like_0p5)
362
+ def_gen.laplace(D_arr_0p5, D_arr_0p5)
363
+ def_gen.laplace(D_arr_like_0p5, D_arr_like_0p5)
364
+ def_gen.laplace(D_arr_0p5, D_arr_0p5, size=1)
365
+ def_gen.laplace(D_arr_like_0p5, D_arr_like_0p5, size=1)
366
+
367
+ def_gen.logistic(0.5, 0.5)
368
+ def_gen.logistic(0.5, 0.5, size=None)
369
+ def_gen.logistic(0.5, 0.5, size=1)
370
+ def_gen.logistic(D_arr_0p5, 0.5)
371
+ def_gen.logistic(0.5, D_arr_0p5)
372
+ def_gen.logistic(D_arr_0p5, 0.5, size=1)
373
+ def_gen.logistic(0.5, D_arr_0p5, size=1)
374
+ def_gen.logistic(D_arr_like_0p5, 0.5)
375
+ def_gen.logistic(0.5, D_arr_like_0p5)
376
+ def_gen.logistic(D_arr_0p5, D_arr_0p5)
377
+ def_gen.logistic(D_arr_like_0p5, D_arr_like_0p5)
378
+ def_gen.logistic(D_arr_0p5, D_arr_0p5, size=1)
379
+ def_gen.logistic(D_arr_like_0p5, D_arr_like_0p5, size=1)
380
+
381
+ def_gen.lognormal(0.5, 0.5)
382
+ def_gen.lognormal(0.5, 0.5, size=None)
383
+ def_gen.lognormal(0.5, 0.5, size=1)
384
+ def_gen.lognormal(D_arr_0p5, 0.5)
385
+ def_gen.lognormal(0.5, D_arr_0p5)
386
+ def_gen.lognormal(D_arr_0p5, 0.5, size=1)
387
+ def_gen.lognormal(0.5, D_arr_0p5, size=1)
388
+ def_gen.lognormal(D_arr_like_0p5, 0.5)
389
+ def_gen.lognormal(0.5, D_arr_like_0p5)
390
+ def_gen.lognormal(D_arr_0p5, D_arr_0p5)
391
+ def_gen.lognormal(D_arr_like_0p5, D_arr_like_0p5)
392
+ def_gen.lognormal(D_arr_0p5, D_arr_0p5, size=1)
393
+ def_gen.lognormal(D_arr_like_0p5, D_arr_like_0p5, size=1)
394
+
395
+ def_gen.noncentral_chisquare(0.5, 0.5)
396
+ def_gen.noncentral_chisquare(0.5, 0.5, size=None)
397
+ def_gen.noncentral_chisquare(0.5, 0.5, size=1)
398
+ def_gen.noncentral_chisquare(D_arr_0p5, 0.5)
399
+ def_gen.noncentral_chisquare(0.5, D_arr_0p5)
400
+ def_gen.noncentral_chisquare(D_arr_0p5, 0.5, size=1)
401
+ def_gen.noncentral_chisquare(0.5, D_arr_0p5, size=1)
402
+ def_gen.noncentral_chisquare(D_arr_like_0p5, 0.5)
403
+ def_gen.noncentral_chisquare(0.5, D_arr_like_0p5)
404
+ def_gen.noncentral_chisquare(D_arr_0p5, D_arr_0p5)
405
+ def_gen.noncentral_chisquare(D_arr_like_0p5, D_arr_like_0p5)
406
+ def_gen.noncentral_chisquare(D_arr_0p5, D_arr_0p5, size=1)
407
+ def_gen.noncentral_chisquare(D_arr_like_0p5, D_arr_like_0p5, size=1)
408
+
409
+ def_gen.normal(0.5, 0.5)
410
+ def_gen.normal(0.5, 0.5, size=None)
411
+ def_gen.normal(0.5, 0.5, size=1)
412
+ def_gen.normal(D_arr_0p5, 0.5)
413
+ def_gen.normal(0.5, D_arr_0p5)
414
+ def_gen.normal(D_arr_0p5, 0.5, size=1)
415
+ def_gen.normal(0.5, D_arr_0p5, size=1)
416
+ def_gen.normal(D_arr_like_0p5, 0.5)
417
+ def_gen.normal(0.5, D_arr_like_0p5)
418
+ def_gen.normal(D_arr_0p5, D_arr_0p5)
419
+ def_gen.normal(D_arr_like_0p5, D_arr_like_0p5)
420
+ def_gen.normal(D_arr_0p5, D_arr_0p5, size=1)
421
+ def_gen.normal(D_arr_like_0p5, D_arr_like_0p5, size=1)
422
+
423
+ def_gen.triangular(0.1, 0.5, 0.9)
424
+ def_gen.triangular(0.1, 0.5, 0.9, size=None)
425
+ def_gen.triangular(0.1, 0.5, 0.9, size=1)
426
+ def_gen.triangular(D_arr_0p1, 0.5, 0.9)
427
+ def_gen.triangular(0.1, D_arr_0p5, 0.9)
428
+ def_gen.triangular(D_arr_0p1, 0.5, D_arr_like_0p9, size=1)
429
+ def_gen.triangular(0.1, D_arr_0p5, 0.9, size=1)
430
+ def_gen.triangular(D_arr_like_0p1, 0.5, D_arr_0p9)
431
+ def_gen.triangular(0.5, D_arr_like_0p5, 0.9)
432
+ def_gen.triangular(D_arr_0p1, D_arr_0p5, 0.9)
433
+ def_gen.triangular(D_arr_like_0p1, D_arr_like_0p5, 0.9)
434
+ def_gen.triangular(D_arr_0p1, D_arr_0p5, D_arr_0p9, size=1)
435
+ def_gen.triangular(D_arr_like_0p1, D_arr_like_0p5, D_arr_like_0p9, size=1)
436
+
437
+ def_gen.noncentral_f(0.1, 0.5, 0.9)
438
+ def_gen.noncentral_f(0.1, 0.5, 0.9, size=None)
439
+ def_gen.noncentral_f(0.1, 0.5, 0.9, size=1)
440
+ def_gen.noncentral_f(D_arr_0p1, 0.5, 0.9)
441
+ def_gen.noncentral_f(0.1, D_arr_0p5, 0.9)
442
+ def_gen.noncentral_f(D_arr_0p1, 0.5, D_arr_like_0p9, size=1)
443
+ def_gen.noncentral_f(0.1, D_arr_0p5, 0.9, size=1)
444
+ def_gen.noncentral_f(D_arr_like_0p1, 0.5, D_arr_0p9)
445
+ def_gen.noncentral_f(0.5, D_arr_like_0p5, 0.9)
446
+ def_gen.noncentral_f(D_arr_0p1, D_arr_0p5, 0.9)
447
+ def_gen.noncentral_f(D_arr_like_0p1, D_arr_like_0p5, 0.9)
448
+ def_gen.noncentral_f(D_arr_0p1, D_arr_0p5, D_arr_0p9, size=1)
449
+ def_gen.noncentral_f(D_arr_like_0p1, D_arr_like_0p5, D_arr_like_0p9, size=1)
450
+
451
+ def_gen.binomial(10, 0.5)
452
+ def_gen.binomial(10, 0.5, size=None)
453
+ def_gen.binomial(10, 0.5, size=1)
454
+ def_gen.binomial(I_arr_10, 0.5)
455
+ def_gen.binomial(10, D_arr_0p5)
456
+ def_gen.binomial(I_arr_10, 0.5, size=1)
457
+ def_gen.binomial(10, D_arr_0p5, size=1)
458
+ def_gen.binomial(I_arr_like_10, 0.5)
459
+ def_gen.binomial(10, D_arr_like_0p5)
460
+ def_gen.binomial(I_arr_10, D_arr_0p5)
461
+ def_gen.binomial(I_arr_like_10, D_arr_like_0p5)
462
+ def_gen.binomial(I_arr_10, D_arr_0p5, size=1)
463
+ def_gen.binomial(I_arr_like_10, D_arr_like_0p5, size=1)
464
+
465
+ def_gen.negative_binomial(10, 0.5)
466
+ def_gen.negative_binomial(10, 0.5, size=None)
467
+ def_gen.negative_binomial(10, 0.5, size=1)
468
+ def_gen.negative_binomial(I_arr_10, 0.5)
469
+ def_gen.negative_binomial(10, D_arr_0p5)
470
+ def_gen.negative_binomial(I_arr_10, 0.5, size=1)
471
+ def_gen.negative_binomial(10, D_arr_0p5, size=1)
472
+ def_gen.negative_binomial(I_arr_like_10, 0.5)
473
+ def_gen.negative_binomial(10, D_arr_like_0p5)
474
+ def_gen.negative_binomial(I_arr_10, D_arr_0p5)
475
+ def_gen.negative_binomial(I_arr_like_10, D_arr_like_0p5)
476
+ def_gen.negative_binomial(I_arr_10, D_arr_0p5, size=1)
477
+ def_gen.negative_binomial(I_arr_like_10, D_arr_like_0p5, size=1)
478
+
479
+ def_gen.hypergeometric(20, 20, 10)
480
+ def_gen.hypergeometric(20, 20, 10, size=None)
481
+ def_gen.hypergeometric(20, 20, 10, size=1)
482
+ def_gen.hypergeometric(I_arr_20, 20, 10)
483
+ def_gen.hypergeometric(20, I_arr_20, 10)
484
+ def_gen.hypergeometric(I_arr_20, 20, I_arr_like_10, size=1)
485
+ def_gen.hypergeometric(20, I_arr_20, 10, size=1)
486
+ def_gen.hypergeometric(I_arr_like_20, 20, I_arr_10)
487
+ def_gen.hypergeometric(20, I_arr_like_20, 10)
488
+ def_gen.hypergeometric(I_arr_20, I_arr_20, 10)
489
+ def_gen.hypergeometric(I_arr_like_20, I_arr_like_20, 10)
490
+ def_gen.hypergeometric(I_arr_20, I_arr_20, I_arr_10, size=1)
491
+ def_gen.hypergeometric(I_arr_like_20, I_arr_like_20, I_arr_like_10, size=1)
492
+
493
+ I_int64_100: np.ndarray[Any, np.dtype[np.int64]] = np.array([100], dtype=np.int64)
494
+
495
+ def_gen.integers(0, 100)
496
+ def_gen.integers(100)
497
+ def_gen.integers([100])
498
+ def_gen.integers(0, [100])
499
+
500
+ I_bool_low: np.ndarray[Any, np.dtype[np.bool_]] = np.array([0], dtype=np.bool_)
501
+ I_bool_low_like: list[int] = [0]
502
+ I_bool_high_open: np.ndarray[Any, np.dtype[np.bool_]] = np.array([1], dtype=np.bool_)
503
+ I_bool_high_closed: np.ndarray[Any, np.dtype[np.bool_]] = np.array([1], dtype=np.bool_)
504
+
505
+ def_gen.integers(2, dtype=bool)
506
+ def_gen.integers(0, 2, dtype=bool)
507
+ def_gen.integers(1, dtype=bool, endpoint=True)
508
+ def_gen.integers(0, 1, dtype=bool, endpoint=True)
509
+ def_gen.integers(I_bool_low_like, 1, dtype=bool, endpoint=True)
510
+ def_gen.integers(I_bool_high_open, dtype=bool)
511
+ def_gen.integers(I_bool_low, I_bool_high_open, dtype=bool)
512
+ def_gen.integers(0, I_bool_high_open, dtype=bool)
513
+ def_gen.integers(I_bool_high_closed, dtype=bool, endpoint=True)
514
+ def_gen.integers(I_bool_low, I_bool_high_closed, dtype=bool, endpoint=True)
515
+ def_gen.integers(0, I_bool_high_closed, dtype=bool, endpoint=True)
516
+
517
+ def_gen.integers(2, dtype=np.bool_)
518
+ def_gen.integers(0, 2, dtype=np.bool_)
519
+ def_gen.integers(1, dtype=np.bool_, endpoint=True)
520
+ def_gen.integers(0, 1, dtype=np.bool_, endpoint=True)
521
+ def_gen.integers(I_bool_low_like, 1, dtype=np.bool_, endpoint=True)
522
+ def_gen.integers(I_bool_high_open, dtype=np.bool_)
523
+ def_gen.integers(I_bool_low, I_bool_high_open, dtype=np.bool_)
524
+ def_gen.integers(0, I_bool_high_open, dtype=np.bool_)
525
+ def_gen.integers(I_bool_high_closed, dtype=np.bool_, endpoint=True)
526
+ def_gen.integers(I_bool_low, I_bool_high_closed, dtype=np.bool_, endpoint=True)
527
+ def_gen.integers(0, I_bool_high_closed, dtype=np.bool_, endpoint=True)
528
+
529
+ I_u1_low: np.ndarray[Any, np.dtype[np.uint8]] = np.array([0], dtype=np.uint8)
530
+ I_u1_low_like: list[int] = [0]
531
+ I_u1_high_open: np.ndarray[Any, np.dtype[np.uint8]] = np.array([255], dtype=np.uint8)
532
+ I_u1_high_closed: np.ndarray[Any, np.dtype[np.uint8]] = np.array([255], dtype=np.uint8)
533
+
534
+ def_gen.integers(256, dtype="u1")
535
+ def_gen.integers(0, 256, dtype="u1")
536
+ def_gen.integers(255, dtype="u1", endpoint=True)
537
+ def_gen.integers(0, 255, dtype="u1", endpoint=True)
538
+ def_gen.integers(I_u1_low_like, 255, dtype="u1", endpoint=True)
539
+ def_gen.integers(I_u1_high_open, dtype="u1")
540
+ def_gen.integers(I_u1_low, I_u1_high_open, dtype="u1")
541
+ def_gen.integers(0, I_u1_high_open, dtype="u1")
542
+ def_gen.integers(I_u1_high_closed, dtype="u1", endpoint=True)
543
+ def_gen.integers(I_u1_low, I_u1_high_closed, dtype="u1", endpoint=True)
544
+ def_gen.integers(0, I_u1_high_closed, dtype="u1", endpoint=True)
545
+
546
+ def_gen.integers(256, dtype="uint8")
547
+ def_gen.integers(0, 256, dtype="uint8")
548
+ def_gen.integers(255, dtype="uint8", endpoint=True)
549
+ def_gen.integers(0, 255, dtype="uint8", endpoint=True)
550
+ def_gen.integers(I_u1_low_like, 255, dtype="uint8", endpoint=True)
551
+ def_gen.integers(I_u1_high_open, dtype="uint8")
552
+ def_gen.integers(I_u1_low, I_u1_high_open, dtype="uint8")
553
+ def_gen.integers(0, I_u1_high_open, dtype="uint8")
554
+ def_gen.integers(I_u1_high_closed, dtype="uint8", endpoint=True)
555
+ def_gen.integers(I_u1_low, I_u1_high_closed, dtype="uint8", endpoint=True)
556
+ def_gen.integers(0, I_u1_high_closed, dtype="uint8", endpoint=True)
557
+
558
+ def_gen.integers(256, dtype=np.uint8)
559
+ def_gen.integers(0, 256, dtype=np.uint8)
560
+ def_gen.integers(255, dtype=np.uint8, endpoint=True)
561
+ def_gen.integers(0, 255, dtype=np.uint8, endpoint=True)
562
+ def_gen.integers(I_u1_low_like, 255, dtype=np.uint8, endpoint=True)
563
+ def_gen.integers(I_u1_high_open, dtype=np.uint8)
564
+ def_gen.integers(I_u1_low, I_u1_high_open, dtype=np.uint8)
565
+ def_gen.integers(0, I_u1_high_open, dtype=np.uint8)
566
+ def_gen.integers(I_u1_high_closed, dtype=np.uint8, endpoint=True)
567
+ def_gen.integers(I_u1_low, I_u1_high_closed, dtype=np.uint8, endpoint=True)
568
+ def_gen.integers(0, I_u1_high_closed, dtype=np.uint8, endpoint=True)
569
+
570
+ I_u2_low: np.ndarray[Any, np.dtype[np.uint16]] = np.array([0], dtype=np.uint16)
571
+ I_u2_low_like: list[int] = [0]
572
+ I_u2_high_open: np.ndarray[Any, np.dtype[np.uint16]] = np.array([65535], dtype=np.uint16)
573
+ I_u2_high_closed: np.ndarray[Any, np.dtype[np.uint16]] = np.array([65535], dtype=np.uint16)
574
+
575
+ def_gen.integers(65536, dtype="u2")
576
+ def_gen.integers(0, 65536, dtype="u2")
577
+ def_gen.integers(65535, dtype="u2", endpoint=True)
578
+ def_gen.integers(0, 65535, dtype="u2", endpoint=True)
579
+ def_gen.integers(I_u2_low_like, 65535, dtype="u2", endpoint=True)
580
+ def_gen.integers(I_u2_high_open, dtype="u2")
581
+ def_gen.integers(I_u2_low, I_u2_high_open, dtype="u2")
582
+ def_gen.integers(0, I_u2_high_open, dtype="u2")
583
+ def_gen.integers(I_u2_high_closed, dtype="u2", endpoint=True)
584
+ def_gen.integers(I_u2_low, I_u2_high_closed, dtype="u2", endpoint=True)
585
+ def_gen.integers(0, I_u2_high_closed, dtype="u2", endpoint=True)
586
+
587
+ def_gen.integers(65536, dtype="uint16")
588
+ def_gen.integers(0, 65536, dtype="uint16")
589
+ def_gen.integers(65535, dtype="uint16", endpoint=True)
590
+ def_gen.integers(0, 65535, dtype="uint16", endpoint=True)
591
+ def_gen.integers(I_u2_low_like, 65535, dtype="uint16", endpoint=True)
592
+ def_gen.integers(I_u2_high_open, dtype="uint16")
593
+ def_gen.integers(I_u2_low, I_u2_high_open, dtype="uint16")
594
+ def_gen.integers(0, I_u2_high_open, dtype="uint16")
595
+ def_gen.integers(I_u2_high_closed, dtype="uint16", endpoint=True)
596
+ def_gen.integers(I_u2_low, I_u2_high_closed, dtype="uint16", endpoint=True)
597
+ def_gen.integers(0, I_u2_high_closed, dtype="uint16", endpoint=True)
598
+
599
+ def_gen.integers(65536, dtype=np.uint16)
600
+ def_gen.integers(0, 65536, dtype=np.uint16)
601
+ def_gen.integers(65535, dtype=np.uint16, endpoint=True)
602
+ def_gen.integers(0, 65535, dtype=np.uint16, endpoint=True)
603
+ def_gen.integers(I_u2_low_like, 65535, dtype=np.uint16, endpoint=True)
604
+ def_gen.integers(I_u2_high_open, dtype=np.uint16)
605
+ def_gen.integers(I_u2_low, I_u2_high_open, dtype=np.uint16)
606
+ def_gen.integers(0, I_u2_high_open, dtype=np.uint16)
607
+ def_gen.integers(I_u2_high_closed, dtype=np.uint16, endpoint=True)
608
+ def_gen.integers(I_u2_low, I_u2_high_closed, dtype=np.uint16, endpoint=True)
609
+ def_gen.integers(0, I_u2_high_closed, dtype=np.uint16, endpoint=True)
610
+
611
+ I_u4_low: np.ndarray[Any, np.dtype[np.uint32]] = np.array([0], dtype=np.uint32)
612
+ I_u4_low_like: list[int] = [0]
613
+ I_u4_high_open: np.ndarray[Any, np.dtype[np.uint32]] = np.array([4294967295], dtype=np.uint32)
614
+ I_u4_high_closed: np.ndarray[Any, np.dtype[np.uint32]] = np.array([4294967295], dtype=np.uint32)
615
+
616
+ def_gen.integers(4294967296, dtype="u4")
617
+ def_gen.integers(0, 4294967296, dtype="u4")
618
+ def_gen.integers(4294967295, dtype="u4", endpoint=True)
619
+ def_gen.integers(0, 4294967295, dtype="u4", endpoint=True)
620
+ def_gen.integers(I_u4_low_like, 4294967295, dtype="u4", endpoint=True)
621
+ def_gen.integers(I_u4_high_open, dtype="u4")
622
+ def_gen.integers(I_u4_low, I_u4_high_open, dtype="u4")
623
+ def_gen.integers(0, I_u4_high_open, dtype="u4")
624
+ def_gen.integers(I_u4_high_closed, dtype="u4", endpoint=True)
625
+ def_gen.integers(I_u4_low, I_u4_high_closed, dtype="u4", endpoint=True)
626
+ def_gen.integers(0, I_u4_high_closed, dtype="u4", endpoint=True)
627
+
628
+ def_gen.integers(4294967296, dtype="uint32")
629
+ def_gen.integers(0, 4294967296, dtype="uint32")
630
+ def_gen.integers(4294967295, dtype="uint32", endpoint=True)
631
+ def_gen.integers(0, 4294967295, dtype="uint32", endpoint=True)
632
+ def_gen.integers(I_u4_low_like, 4294967295, dtype="uint32", endpoint=True)
633
+ def_gen.integers(I_u4_high_open, dtype="uint32")
634
+ def_gen.integers(I_u4_low, I_u4_high_open, dtype="uint32")
635
+ def_gen.integers(0, I_u4_high_open, dtype="uint32")
636
+ def_gen.integers(I_u4_high_closed, dtype="uint32", endpoint=True)
637
+ def_gen.integers(I_u4_low, I_u4_high_closed, dtype="uint32", endpoint=True)
638
+ def_gen.integers(0, I_u4_high_closed, dtype="uint32", endpoint=True)
639
+
640
+ def_gen.integers(4294967296, dtype=np.uint32)
641
+ def_gen.integers(0, 4294967296, dtype=np.uint32)
642
+ def_gen.integers(4294967295, dtype=np.uint32, endpoint=True)
643
+ def_gen.integers(0, 4294967295, dtype=np.uint32, endpoint=True)
644
+ def_gen.integers(I_u4_low_like, 4294967295, dtype=np.uint32, endpoint=True)
645
+ def_gen.integers(I_u4_high_open, dtype=np.uint32)
646
+ def_gen.integers(I_u4_low, I_u4_high_open, dtype=np.uint32)
647
+ def_gen.integers(0, I_u4_high_open, dtype=np.uint32)
648
+ def_gen.integers(I_u4_high_closed, dtype=np.uint32, endpoint=True)
649
+ def_gen.integers(I_u4_low, I_u4_high_closed, dtype=np.uint32, endpoint=True)
650
+ def_gen.integers(0, I_u4_high_closed, dtype=np.uint32, endpoint=True)
651
+
652
+ I_u8_low: np.ndarray[Any, np.dtype[np.uint64]] = np.array([0], dtype=np.uint64)
653
+ I_u8_low_like: list[int] = [0]
654
+ I_u8_high_open: np.ndarray[Any, np.dtype[np.uint64]] = np.array([18446744073709551615], dtype=np.uint64)
655
+ I_u8_high_closed: np.ndarray[Any, np.dtype[np.uint64]] = np.array([18446744073709551615], dtype=np.uint64)
656
+
657
+ def_gen.integers(18446744073709551616, dtype="u8")
658
+ def_gen.integers(0, 18446744073709551616, dtype="u8")
659
+ def_gen.integers(18446744073709551615, dtype="u8", endpoint=True)
660
+ def_gen.integers(0, 18446744073709551615, dtype="u8", endpoint=True)
661
+ def_gen.integers(I_u8_low_like, 18446744073709551615, dtype="u8", endpoint=True)
662
+ def_gen.integers(I_u8_high_open, dtype="u8")
663
+ def_gen.integers(I_u8_low, I_u8_high_open, dtype="u8")
664
+ def_gen.integers(0, I_u8_high_open, dtype="u8")
665
+ def_gen.integers(I_u8_high_closed, dtype="u8", endpoint=True)
666
+ def_gen.integers(I_u8_low, I_u8_high_closed, dtype="u8", endpoint=True)
667
+ def_gen.integers(0, I_u8_high_closed, dtype="u8", endpoint=True)
668
+
669
+ def_gen.integers(18446744073709551616, dtype="uint64")
670
+ def_gen.integers(0, 18446744073709551616, dtype="uint64")
671
+ def_gen.integers(18446744073709551615, dtype="uint64", endpoint=True)
672
+ def_gen.integers(0, 18446744073709551615, dtype="uint64", endpoint=True)
673
+ def_gen.integers(I_u8_low_like, 18446744073709551615, dtype="uint64", endpoint=True)
674
+ def_gen.integers(I_u8_high_open, dtype="uint64")
675
+ def_gen.integers(I_u8_low, I_u8_high_open, dtype="uint64")
676
+ def_gen.integers(0, I_u8_high_open, dtype="uint64")
677
+ def_gen.integers(I_u8_high_closed, dtype="uint64", endpoint=True)
678
+ def_gen.integers(I_u8_low, I_u8_high_closed, dtype="uint64", endpoint=True)
679
+ def_gen.integers(0, I_u8_high_closed, dtype="uint64", endpoint=True)
680
+
681
+ def_gen.integers(18446744073709551616, dtype=np.uint64)
682
+ def_gen.integers(0, 18446744073709551616, dtype=np.uint64)
683
+ def_gen.integers(18446744073709551615, dtype=np.uint64, endpoint=True)
684
+ def_gen.integers(0, 18446744073709551615, dtype=np.uint64, endpoint=True)
685
+ def_gen.integers(I_u8_low_like, 18446744073709551615, dtype=np.uint64, endpoint=True)
686
+ def_gen.integers(I_u8_high_open, dtype=np.uint64)
687
+ def_gen.integers(I_u8_low, I_u8_high_open, dtype=np.uint64)
688
+ def_gen.integers(0, I_u8_high_open, dtype=np.uint64)
689
+ def_gen.integers(I_u8_high_closed, dtype=np.uint64, endpoint=True)
690
+ def_gen.integers(I_u8_low, I_u8_high_closed, dtype=np.uint64, endpoint=True)
691
+ def_gen.integers(0, I_u8_high_closed, dtype=np.uint64, endpoint=True)
692
+
693
+ I_i1_low: np.ndarray[Any, np.dtype[np.int8]] = np.array([-128], dtype=np.int8)
694
+ I_i1_low_like: list[int] = [-128]
695
+ I_i1_high_open: np.ndarray[Any, np.dtype[np.int8]] = np.array([127], dtype=np.int8)
696
+ I_i1_high_closed: np.ndarray[Any, np.dtype[np.int8]] = np.array([127], dtype=np.int8)
697
+
698
+ def_gen.integers(128, dtype="i1")
699
+ def_gen.integers(-128, 128, dtype="i1")
700
+ def_gen.integers(127, dtype="i1", endpoint=True)
701
+ def_gen.integers(-128, 127, dtype="i1", endpoint=True)
702
+ def_gen.integers(I_i1_low_like, 127, dtype="i1", endpoint=True)
703
+ def_gen.integers(I_i1_high_open, dtype="i1")
704
+ def_gen.integers(I_i1_low, I_i1_high_open, dtype="i1")
705
+ def_gen.integers(-128, I_i1_high_open, dtype="i1")
706
+ def_gen.integers(I_i1_high_closed, dtype="i1", endpoint=True)
707
+ def_gen.integers(I_i1_low, I_i1_high_closed, dtype="i1", endpoint=True)
708
+ def_gen.integers(-128, I_i1_high_closed, dtype="i1", endpoint=True)
709
+
710
+ def_gen.integers(128, dtype="int8")
711
+ def_gen.integers(-128, 128, dtype="int8")
712
+ def_gen.integers(127, dtype="int8", endpoint=True)
713
+ def_gen.integers(-128, 127, dtype="int8", endpoint=True)
714
+ def_gen.integers(I_i1_low_like, 127, dtype="int8", endpoint=True)
715
+ def_gen.integers(I_i1_high_open, dtype="int8")
716
+ def_gen.integers(I_i1_low, I_i1_high_open, dtype="int8")
717
+ def_gen.integers(-128, I_i1_high_open, dtype="int8")
718
+ def_gen.integers(I_i1_high_closed, dtype="int8", endpoint=True)
719
+ def_gen.integers(I_i1_low, I_i1_high_closed, dtype="int8", endpoint=True)
720
+ def_gen.integers(-128, I_i1_high_closed, dtype="int8", endpoint=True)
721
+
722
+ def_gen.integers(128, dtype=np.int8)
723
+ def_gen.integers(-128, 128, dtype=np.int8)
724
+ def_gen.integers(127, dtype=np.int8, endpoint=True)
725
+ def_gen.integers(-128, 127, dtype=np.int8, endpoint=True)
726
+ def_gen.integers(I_i1_low_like, 127, dtype=np.int8, endpoint=True)
727
+ def_gen.integers(I_i1_high_open, dtype=np.int8)
728
+ def_gen.integers(I_i1_low, I_i1_high_open, dtype=np.int8)
729
+ def_gen.integers(-128, I_i1_high_open, dtype=np.int8)
730
+ def_gen.integers(I_i1_high_closed, dtype=np.int8, endpoint=True)
731
+ def_gen.integers(I_i1_low, I_i1_high_closed, dtype=np.int8, endpoint=True)
732
+ def_gen.integers(-128, I_i1_high_closed, dtype=np.int8, endpoint=True)
733
+
734
+ I_i2_low: np.ndarray[Any, np.dtype[np.int16]] = np.array([-32768], dtype=np.int16)
735
+ I_i2_low_like: list[int] = [-32768]
736
+ I_i2_high_open: np.ndarray[Any, np.dtype[np.int16]] = np.array([32767], dtype=np.int16)
737
+ I_i2_high_closed: np.ndarray[Any, np.dtype[np.int16]] = np.array([32767], dtype=np.int16)
738
+
739
+ def_gen.integers(32768, dtype="i2")
740
+ def_gen.integers(-32768, 32768, dtype="i2")
741
+ def_gen.integers(32767, dtype="i2", endpoint=True)
742
+ def_gen.integers(-32768, 32767, dtype="i2", endpoint=True)
743
+ def_gen.integers(I_i2_low_like, 32767, dtype="i2", endpoint=True)
744
+ def_gen.integers(I_i2_high_open, dtype="i2")
745
+ def_gen.integers(I_i2_low, I_i2_high_open, dtype="i2")
746
+ def_gen.integers(-32768, I_i2_high_open, dtype="i2")
747
+ def_gen.integers(I_i2_high_closed, dtype="i2", endpoint=True)
748
+ def_gen.integers(I_i2_low, I_i2_high_closed, dtype="i2", endpoint=True)
749
+ def_gen.integers(-32768, I_i2_high_closed, dtype="i2", endpoint=True)
750
+
751
+ def_gen.integers(32768, dtype="int16")
752
+ def_gen.integers(-32768, 32768, dtype="int16")
753
+ def_gen.integers(32767, dtype="int16", endpoint=True)
754
+ def_gen.integers(-32768, 32767, dtype="int16", endpoint=True)
755
+ def_gen.integers(I_i2_low_like, 32767, dtype="int16", endpoint=True)
756
+ def_gen.integers(I_i2_high_open, dtype="int16")
757
+ def_gen.integers(I_i2_low, I_i2_high_open, dtype="int16")
758
+ def_gen.integers(-32768, I_i2_high_open, dtype="int16")
759
+ def_gen.integers(I_i2_high_closed, dtype="int16", endpoint=True)
760
+ def_gen.integers(I_i2_low, I_i2_high_closed, dtype="int16", endpoint=True)
761
+ def_gen.integers(-32768, I_i2_high_closed, dtype="int16", endpoint=True)
762
+
763
+ def_gen.integers(32768, dtype=np.int16)
764
+ def_gen.integers(-32768, 32768, dtype=np.int16)
765
+ def_gen.integers(32767, dtype=np.int16, endpoint=True)
766
+ def_gen.integers(-32768, 32767, dtype=np.int16, endpoint=True)
767
+ def_gen.integers(I_i2_low_like, 32767, dtype=np.int16, endpoint=True)
768
+ def_gen.integers(I_i2_high_open, dtype=np.int16)
769
+ def_gen.integers(I_i2_low, I_i2_high_open, dtype=np.int16)
770
+ def_gen.integers(-32768, I_i2_high_open, dtype=np.int16)
771
+ def_gen.integers(I_i2_high_closed, dtype=np.int16, endpoint=True)
772
+ def_gen.integers(I_i2_low, I_i2_high_closed, dtype=np.int16, endpoint=True)
773
+ def_gen.integers(-32768, I_i2_high_closed, dtype=np.int16, endpoint=True)
774
+
775
+ I_i4_low: np.ndarray[Any, np.dtype[np.int32]] = np.array([-2147483648], dtype=np.int32)
776
+ I_i4_low_like: list[int] = [-2147483648]
777
+ I_i4_high_open: np.ndarray[Any, np.dtype[np.int32]] = np.array([2147483647], dtype=np.int32)
778
+ I_i4_high_closed: np.ndarray[Any, np.dtype[np.int32]] = np.array([2147483647], dtype=np.int32)
779
+
780
+ def_gen.integers(2147483648, dtype="i4")
781
+ def_gen.integers(-2147483648, 2147483648, dtype="i4")
782
+ def_gen.integers(2147483647, dtype="i4", endpoint=True)
783
+ def_gen.integers(-2147483648, 2147483647, dtype="i4", endpoint=True)
784
+ def_gen.integers(I_i4_low_like, 2147483647, dtype="i4", endpoint=True)
785
+ def_gen.integers(I_i4_high_open, dtype="i4")
786
+ def_gen.integers(I_i4_low, I_i4_high_open, dtype="i4")
787
+ def_gen.integers(-2147483648, I_i4_high_open, dtype="i4")
788
+ def_gen.integers(I_i4_high_closed, dtype="i4", endpoint=True)
789
+ def_gen.integers(I_i4_low, I_i4_high_closed, dtype="i4", endpoint=True)
790
+ def_gen.integers(-2147483648, I_i4_high_closed, dtype="i4", endpoint=True)
791
+
792
+ def_gen.integers(2147483648, dtype="int32")
793
+ def_gen.integers(-2147483648, 2147483648, dtype="int32")
794
+ def_gen.integers(2147483647, dtype="int32", endpoint=True)
795
+ def_gen.integers(-2147483648, 2147483647, dtype="int32", endpoint=True)
796
+ def_gen.integers(I_i4_low_like, 2147483647, dtype="int32", endpoint=True)
797
+ def_gen.integers(I_i4_high_open, dtype="int32")
798
+ def_gen.integers(I_i4_low, I_i4_high_open, dtype="int32")
799
+ def_gen.integers(-2147483648, I_i4_high_open, dtype="int32")
800
+ def_gen.integers(I_i4_high_closed, dtype="int32", endpoint=True)
801
+ def_gen.integers(I_i4_low, I_i4_high_closed, dtype="int32", endpoint=True)
802
+ def_gen.integers(-2147483648, I_i4_high_closed, dtype="int32", endpoint=True)
803
+
804
+ def_gen.integers(2147483648, dtype=np.int32)
805
+ def_gen.integers(-2147483648, 2147483648, dtype=np.int32)
806
+ def_gen.integers(2147483647, dtype=np.int32, endpoint=True)
807
+ def_gen.integers(-2147483648, 2147483647, dtype=np.int32, endpoint=True)
808
+ def_gen.integers(I_i4_low_like, 2147483647, dtype=np.int32, endpoint=True)
809
+ def_gen.integers(I_i4_high_open, dtype=np.int32)
810
+ def_gen.integers(I_i4_low, I_i4_high_open, dtype=np.int32)
811
+ def_gen.integers(-2147483648, I_i4_high_open, dtype=np.int32)
812
+ def_gen.integers(I_i4_high_closed, dtype=np.int32, endpoint=True)
813
+ def_gen.integers(I_i4_low, I_i4_high_closed, dtype=np.int32, endpoint=True)
814
+ def_gen.integers(-2147483648, I_i4_high_closed, dtype=np.int32, endpoint=True)
815
+
816
+ I_i8_low: np.ndarray[Any, np.dtype[np.int64]] = np.array([-9223372036854775808], dtype=np.int64)
817
+ I_i8_low_like: list[int] = [-9223372036854775808]
818
+ I_i8_high_open: np.ndarray[Any, np.dtype[np.int64]] = np.array([9223372036854775807], dtype=np.int64)
819
+ I_i8_high_closed: np.ndarray[Any, np.dtype[np.int64]] = np.array([9223372036854775807], dtype=np.int64)
820
+
821
+ def_gen.integers(9223372036854775808, dtype="i8")
822
+ def_gen.integers(-9223372036854775808, 9223372036854775808, dtype="i8")
823
+ def_gen.integers(9223372036854775807, dtype="i8", endpoint=True)
824
+ def_gen.integers(-9223372036854775808, 9223372036854775807, dtype="i8", endpoint=True)
825
+ def_gen.integers(I_i8_low_like, 9223372036854775807, dtype="i8", endpoint=True)
826
+ def_gen.integers(I_i8_high_open, dtype="i8")
827
+ def_gen.integers(I_i8_low, I_i8_high_open, dtype="i8")
828
+ def_gen.integers(-9223372036854775808, I_i8_high_open, dtype="i8")
829
+ def_gen.integers(I_i8_high_closed, dtype="i8", endpoint=True)
830
+ def_gen.integers(I_i8_low, I_i8_high_closed, dtype="i8", endpoint=True)
831
+ def_gen.integers(-9223372036854775808, I_i8_high_closed, dtype="i8", endpoint=True)
832
+
833
+ def_gen.integers(9223372036854775808, dtype="int64")
834
+ def_gen.integers(-9223372036854775808, 9223372036854775808, dtype="int64")
835
+ def_gen.integers(9223372036854775807, dtype="int64", endpoint=True)
836
+ def_gen.integers(-9223372036854775808, 9223372036854775807, dtype="int64", endpoint=True)
837
+ def_gen.integers(I_i8_low_like, 9223372036854775807, dtype="int64", endpoint=True)
838
+ def_gen.integers(I_i8_high_open, dtype="int64")
839
+ def_gen.integers(I_i8_low, I_i8_high_open, dtype="int64")
840
+ def_gen.integers(-9223372036854775808, I_i8_high_open, dtype="int64")
841
+ def_gen.integers(I_i8_high_closed, dtype="int64", endpoint=True)
842
+ def_gen.integers(I_i8_low, I_i8_high_closed, dtype="int64", endpoint=True)
843
+ def_gen.integers(-9223372036854775808, I_i8_high_closed, dtype="int64", endpoint=True)
844
+
845
+ def_gen.integers(9223372036854775808, dtype=np.int64)
846
+ def_gen.integers(-9223372036854775808, 9223372036854775808, dtype=np.int64)
847
+ def_gen.integers(9223372036854775807, dtype=np.int64, endpoint=True)
848
+ def_gen.integers(-9223372036854775808, 9223372036854775807, dtype=np.int64, endpoint=True)
849
+ def_gen.integers(I_i8_low_like, 9223372036854775807, dtype=np.int64, endpoint=True)
850
+ def_gen.integers(I_i8_high_open, dtype=np.int64)
851
+ def_gen.integers(I_i8_low, I_i8_high_open, dtype=np.int64)
852
+ def_gen.integers(-9223372036854775808, I_i8_high_open, dtype=np.int64)
853
+ def_gen.integers(I_i8_high_closed, dtype=np.int64, endpoint=True)
854
+ def_gen.integers(I_i8_low, I_i8_high_closed, dtype=np.int64, endpoint=True)
855
+ def_gen.integers(-9223372036854775808, I_i8_high_closed, dtype=np.int64, endpoint=True)
856
+
857
+
858
+ def_gen.bit_generator
859
+
860
+ def_gen.bytes(2)
861
+
862
+ def_gen.choice(5)
863
+ def_gen.choice(5, 3)
864
+ def_gen.choice(5, 3, replace=True)
865
+ def_gen.choice(5, 3, p=[1 / 5] * 5)
866
+ def_gen.choice(5, 3, p=[1 / 5] * 5, replace=False)
867
+
868
+ def_gen.choice(["pooh", "rabbit", "piglet", "Christopher"])
869
+ def_gen.choice(["pooh", "rabbit", "piglet", "Christopher"], 3)
870
+ def_gen.choice(["pooh", "rabbit", "piglet", "Christopher"], 3, p=[1 / 4] * 4)
871
+ def_gen.choice(["pooh", "rabbit", "piglet", "Christopher"], 3, replace=True)
872
+ def_gen.choice(["pooh", "rabbit", "piglet", "Christopher"], 3, replace=False, p=np.array([1 / 8, 1 / 8, 1 / 2, 1 / 4]))
873
+
874
+ def_gen.dirichlet([0.5, 0.5])
875
+ def_gen.dirichlet(np.array([0.5, 0.5]))
876
+ def_gen.dirichlet(np.array([0.5, 0.5]), size=3)
877
+
878
+ def_gen.multinomial(20, [1 / 6.0] * 6)
879
+ def_gen.multinomial(20, np.array([0.5, 0.5]))
880
+ def_gen.multinomial(20, [1 / 6.0] * 6, size=2)
881
+ def_gen.multinomial([[10], [20]], [1 / 6.0] * 6, size=(2, 2))
882
+ def_gen.multinomial(np.array([[10], [20]]), np.array([0.5, 0.5]), size=(2, 2))
883
+
884
+ def_gen.multivariate_hypergeometric([3, 5, 7], 2)
885
+ def_gen.multivariate_hypergeometric(np.array([3, 5, 7]), 2)
886
+ def_gen.multivariate_hypergeometric(np.array([3, 5, 7]), 2, size=4)
887
+ def_gen.multivariate_hypergeometric(np.array([3, 5, 7]), 2, size=(4, 7))
888
+ def_gen.multivariate_hypergeometric([3, 5, 7], 2, method="count")
889
+ def_gen.multivariate_hypergeometric(np.array([3, 5, 7]), 2, method="marginals")
890
+
891
+ def_gen.multivariate_normal([0.0], [[1.0]])
892
+ def_gen.multivariate_normal([0.0], np.array([[1.0]]))
893
+ def_gen.multivariate_normal(np.array([0.0]), [[1.0]])
894
+ def_gen.multivariate_normal([0.0], np.array([[1.0]]))
895
+
896
+ def_gen.permutation(10)
897
+ def_gen.permutation([1, 2, 3, 4])
898
+ def_gen.permutation(np.array([1, 2, 3, 4]))
899
+ def_gen.permutation(D_2D, axis=1)
900
+ def_gen.permuted(D_2D)
901
+ def_gen.permuted(D_2D_like)
902
+ def_gen.permuted(D_2D, axis=1)
903
+ def_gen.permuted(D_2D, out=D_2D)
904
+ def_gen.permuted(D_2D_like, out=D_2D)
905
+ def_gen.permuted(D_2D_like, out=D_2D)
906
+ def_gen.permuted(D_2D, axis=1, out=D_2D)
907
+
908
+ def_gen.shuffle(np.arange(10))
909
+ def_gen.shuffle([1, 2, 3, 4, 5])
910
+ def_gen.shuffle(D_2D, axis=1)
911
+
912
+ def_gen.__str__()
913
+ def_gen.__repr__()
914
+ def_gen_state: dict[str, Any]
915
+ def_gen_state = def_gen.__getstate__()
916
+ def_gen.__setstate__(def_gen_state)
917
+
918
+ # RandomState
919
+ random_st: np.random.RandomState = np.random.RandomState()
920
+
921
+ random_st.standard_normal()
922
+ random_st.standard_normal(size=None)
923
+ random_st.standard_normal(size=1)
924
+
925
+ random_st.random()
926
+ random_st.random(size=None)
927
+ random_st.random(size=1)
928
+
929
+ random_st.standard_cauchy()
930
+ random_st.standard_cauchy(size=None)
931
+ random_st.standard_cauchy(size=1)
932
+
933
+ random_st.standard_exponential()
934
+ random_st.standard_exponential(size=None)
935
+ random_st.standard_exponential(size=1)
936
+
937
+ random_st.zipf(1.5)
938
+ random_st.zipf(1.5, size=None)
939
+ random_st.zipf(1.5, size=1)
940
+ random_st.zipf(D_arr_1p5)
941
+ random_st.zipf(D_arr_1p5, size=1)
942
+ random_st.zipf(D_arr_like_1p5)
943
+ random_st.zipf(D_arr_like_1p5, size=1)
944
+
945
+ random_st.weibull(0.5)
946
+ random_st.weibull(0.5, size=None)
947
+ random_st.weibull(0.5, size=1)
948
+ random_st.weibull(D_arr_0p5)
949
+ random_st.weibull(D_arr_0p5, size=1)
950
+ random_st.weibull(D_arr_like_0p5)
951
+ random_st.weibull(D_arr_like_0p5, size=1)
952
+
953
+ random_st.standard_t(0.5)
954
+ random_st.standard_t(0.5, size=None)
955
+ random_st.standard_t(0.5, size=1)
956
+ random_st.standard_t(D_arr_0p5)
957
+ random_st.standard_t(D_arr_0p5, size=1)
958
+ random_st.standard_t(D_arr_like_0p5)
959
+ random_st.standard_t(D_arr_like_0p5, size=1)
960
+
961
+ random_st.poisson(0.5)
962
+ random_st.poisson(0.5, size=None)
963
+ random_st.poisson(0.5, size=1)
964
+ random_st.poisson(D_arr_0p5)
965
+ random_st.poisson(D_arr_0p5, size=1)
966
+ random_st.poisson(D_arr_like_0p5)
967
+ random_st.poisson(D_arr_like_0p5, size=1)
968
+
969
+ random_st.power(0.5)
970
+ random_st.power(0.5, size=None)
971
+ random_st.power(0.5, size=1)
972
+ random_st.power(D_arr_0p5)
973
+ random_st.power(D_arr_0p5, size=1)
974
+ random_st.power(D_arr_like_0p5)
975
+ random_st.power(D_arr_like_0p5, size=1)
976
+
977
+ random_st.pareto(0.5)
978
+ random_st.pareto(0.5, size=None)
979
+ random_st.pareto(0.5, size=1)
980
+ random_st.pareto(D_arr_0p5)
981
+ random_st.pareto(D_arr_0p5, size=1)
982
+ random_st.pareto(D_arr_like_0p5)
983
+ random_st.pareto(D_arr_like_0p5, size=1)
984
+
985
+ random_st.chisquare(0.5)
986
+ random_st.chisquare(0.5, size=None)
987
+ random_st.chisquare(0.5, size=1)
988
+ random_st.chisquare(D_arr_0p5)
989
+ random_st.chisquare(D_arr_0p5, size=1)
990
+ random_st.chisquare(D_arr_like_0p5)
991
+ random_st.chisquare(D_arr_like_0p5, size=1)
992
+
993
+ random_st.exponential(0.5)
994
+ random_st.exponential(0.5, size=None)
995
+ random_st.exponential(0.5, size=1)
996
+ random_st.exponential(D_arr_0p5)
997
+ random_st.exponential(D_arr_0p5, size=1)
998
+ random_st.exponential(D_arr_like_0p5)
999
+ random_st.exponential(D_arr_like_0p5, size=1)
1000
+
1001
+ random_st.geometric(0.5)
1002
+ random_st.geometric(0.5, size=None)
1003
+ random_st.geometric(0.5, size=1)
1004
+ random_st.geometric(D_arr_0p5)
1005
+ random_st.geometric(D_arr_0p5, size=1)
1006
+ random_st.geometric(D_arr_like_0p5)
1007
+ random_st.geometric(D_arr_like_0p5, size=1)
1008
+
1009
+ random_st.logseries(0.5)
1010
+ random_st.logseries(0.5, size=None)
1011
+ random_st.logseries(0.5, size=1)
1012
+ random_st.logseries(D_arr_0p5)
1013
+ random_st.logseries(D_arr_0p5, size=1)
1014
+ random_st.logseries(D_arr_like_0p5)
1015
+ random_st.logseries(D_arr_like_0p5, size=1)
1016
+
1017
+ random_st.rayleigh(0.5)
1018
+ random_st.rayleigh(0.5, size=None)
1019
+ random_st.rayleigh(0.5, size=1)
1020
+ random_st.rayleigh(D_arr_0p5)
1021
+ random_st.rayleigh(D_arr_0p5, size=1)
1022
+ random_st.rayleigh(D_arr_like_0p5)
1023
+ random_st.rayleigh(D_arr_like_0p5, size=1)
1024
+
1025
+ random_st.standard_gamma(0.5)
1026
+ random_st.standard_gamma(0.5, size=None)
1027
+ random_st.standard_gamma(0.5, size=1)
1028
+ random_st.standard_gamma(D_arr_0p5)
1029
+ random_st.standard_gamma(D_arr_0p5, size=1)
1030
+ random_st.standard_gamma(D_arr_like_0p5)
1031
+ random_st.standard_gamma(D_arr_like_0p5, size=1)
1032
+ random_st.standard_gamma(D_arr_like_0p5, size=1)
1033
+
1034
+ random_st.vonmises(0.5, 0.5)
1035
+ random_st.vonmises(0.5, 0.5, size=None)
1036
+ random_st.vonmises(0.5, 0.5, size=1)
1037
+ random_st.vonmises(D_arr_0p5, 0.5)
1038
+ random_st.vonmises(0.5, D_arr_0p5)
1039
+ random_st.vonmises(D_arr_0p5, 0.5, size=1)
1040
+ random_st.vonmises(0.5, D_arr_0p5, size=1)
1041
+ random_st.vonmises(D_arr_like_0p5, 0.5)
1042
+ random_st.vonmises(0.5, D_arr_like_0p5)
1043
+ random_st.vonmises(D_arr_0p5, D_arr_0p5)
1044
+ random_st.vonmises(D_arr_like_0p5, D_arr_like_0p5)
1045
+ random_st.vonmises(D_arr_0p5, D_arr_0p5, size=1)
1046
+ random_st.vonmises(D_arr_like_0p5, D_arr_like_0p5, size=1)
1047
+
1048
+ random_st.wald(0.5, 0.5)
1049
+ random_st.wald(0.5, 0.5, size=None)
1050
+ random_st.wald(0.5, 0.5, size=1)
1051
+ random_st.wald(D_arr_0p5, 0.5)
1052
+ random_st.wald(0.5, D_arr_0p5)
1053
+ random_st.wald(D_arr_0p5, 0.5, size=1)
1054
+ random_st.wald(0.5, D_arr_0p5, size=1)
1055
+ random_st.wald(D_arr_like_0p5, 0.5)
1056
+ random_st.wald(0.5, D_arr_like_0p5)
1057
+ random_st.wald(D_arr_0p5, D_arr_0p5)
1058
+ random_st.wald(D_arr_like_0p5, D_arr_like_0p5)
1059
+ random_st.wald(D_arr_0p5, D_arr_0p5, size=1)
1060
+ random_st.wald(D_arr_like_0p5, D_arr_like_0p5, size=1)
1061
+
1062
+ random_st.uniform(0.5, 0.5)
1063
+ random_st.uniform(0.5, 0.5, size=None)
1064
+ random_st.uniform(0.5, 0.5, size=1)
1065
+ random_st.uniform(D_arr_0p5, 0.5)
1066
+ random_st.uniform(0.5, D_arr_0p5)
1067
+ random_st.uniform(D_arr_0p5, 0.5, size=1)
1068
+ random_st.uniform(0.5, D_arr_0p5, size=1)
1069
+ random_st.uniform(D_arr_like_0p5, 0.5)
1070
+ random_st.uniform(0.5, D_arr_like_0p5)
1071
+ random_st.uniform(D_arr_0p5, D_arr_0p5)
1072
+ random_st.uniform(D_arr_like_0p5, D_arr_like_0p5)
1073
+ random_st.uniform(D_arr_0p5, D_arr_0p5, size=1)
1074
+ random_st.uniform(D_arr_like_0p5, D_arr_like_0p5, size=1)
1075
+
1076
+ random_st.beta(0.5, 0.5)
1077
+ random_st.beta(0.5, 0.5, size=None)
1078
+ random_st.beta(0.5, 0.5, size=1)
1079
+ random_st.beta(D_arr_0p5, 0.5)
1080
+ random_st.beta(0.5, D_arr_0p5)
1081
+ random_st.beta(D_arr_0p5, 0.5, size=1)
1082
+ random_st.beta(0.5, D_arr_0p5, size=1)
1083
+ random_st.beta(D_arr_like_0p5, 0.5)
1084
+ random_st.beta(0.5, D_arr_like_0p5)
1085
+ random_st.beta(D_arr_0p5, D_arr_0p5)
1086
+ random_st.beta(D_arr_like_0p5, D_arr_like_0p5)
1087
+ random_st.beta(D_arr_0p5, D_arr_0p5, size=1)
1088
+ random_st.beta(D_arr_like_0p5, D_arr_like_0p5, size=1)
1089
+
1090
+ random_st.f(0.5, 0.5)
1091
+ random_st.f(0.5, 0.5, size=None)
1092
+ random_st.f(0.5, 0.5, size=1)
1093
+ random_st.f(D_arr_0p5, 0.5)
1094
+ random_st.f(0.5, D_arr_0p5)
1095
+ random_st.f(D_arr_0p5, 0.5, size=1)
1096
+ random_st.f(0.5, D_arr_0p5, size=1)
1097
+ random_st.f(D_arr_like_0p5, 0.5)
1098
+ random_st.f(0.5, D_arr_like_0p5)
1099
+ random_st.f(D_arr_0p5, D_arr_0p5)
1100
+ random_st.f(D_arr_like_0p5, D_arr_like_0p5)
1101
+ random_st.f(D_arr_0p5, D_arr_0p5, size=1)
1102
+ random_st.f(D_arr_like_0p5, D_arr_like_0p5, size=1)
1103
+
1104
+ random_st.gamma(0.5, 0.5)
1105
+ random_st.gamma(0.5, 0.5, size=None)
1106
+ random_st.gamma(0.5, 0.5, size=1)
1107
+ random_st.gamma(D_arr_0p5, 0.5)
1108
+ random_st.gamma(0.5, D_arr_0p5)
1109
+ random_st.gamma(D_arr_0p5, 0.5, size=1)
1110
+ random_st.gamma(0.5, D_arr_0p5, size=1)
1111
+ random_st.gamma(D_arr_like_0p5, 0.5)
1112
+ random_st.gamma(0.5, D_arr_like_0p5)
1113
+ random_st.gamma(D_arr_0p5, D_arr_0p5)
1114
+ random_st.gamma(D_arr_like_0p5, D_arr_like_0p5)
1115
+ random_st.gamma(D_arr_0p5, D_arr_0p5, size=1)
1116
+ random_st.gamma(D_arr_like_0p5, D_arr_like_0p5, size=1)
1117
+
1118
+ random_st.gumbel(0.5, 0.5)
1119
+ random_st.gumbel(0.5, 0.5, size=None)
1120
+ random_st.gumbel(0.5, 0.5, size=1)
1121
+ random_st.gumbel(D_arr_0p5, 0.5)
1122
+ random_st.gumbel(0.5, D_arr_0p5)
1123
+ random_st.gumbel(D_arr_0p5, 0.5, size=1)
1124
+ random_st.gumbel(0.5, D_arr_0p5, size=1)
1125
+ random_st.gumbel(D_arr_like_0p5, 0.5)
1126
+ random_st.gumbel(0.5, D_arr_like_0p5)
1127
+ random_st.gumbel(D_arr_0p5, D_arr_0p5)
1128
+ random_st.gumbel(D_arr_like_0p5, D_arr_like_0p5)
1129
+ random_st.gumbel(D_arr_0p5, D_arr_0p5, size=1)
1130
+ random_st.gumbel(D_arr_like_0p5, D_arr_like_0p5, size=1)
1131
+
1132
+ random_st.laplace(0.5, 0.5)
1133
+ random_st.laplace(0.5, 0.5, size=None)
1134
+ random_st.laplace(0.5, 0.5, size=1)
1135
+ random_st.laplace(D_arr_0p5, 0.5)
1136
+ random_st.laplace(0.5, D_arr_0p5)
1137
+ random_st.laplace(D_arr_0p5, 0.5, size=1)
1138
+ random_st.laplace(0.5, D_arr_0p5, size=1)
1139
+ random_st.laplace(D_arr_like_0p5, 0.5)
1140
+ random_st.laplace(0.5, D_arr_like_0p5)
1141
+ random_st.laplace(D_arr_0p5, D_arr_0p5)
1142
+ random_st.laplace(D_arr_like_0p5, D_arr_like_0p5)
1143
+ random_st.laplace(D_arr_0p5, D_arr_0p5, size=1)
1144
+ random_st.laplace(D_arr_like_0p5, D_arr_like_0p5, size=1)
1145
+
1146
+ random_st.logistic(0.5, 0.5)
1147
+ random_st.logistic(0.5, 0.5, size=None)
1148
+ random_st.logistic(0.5, 0.5, size=1)
1149
+ random_st.logistic(D_arr_0p5, 0.5)
1150
+ random_st.logistic(0.5, D_arr_0p5)
1151
+ random_st.logistic(D_arr_0p5, 0.5, size=1)
1152
+ random_st.logistic(0.5, D_arr_0p5, size=1)
1153
+ random_st.logistic(D_arr_like_0p5, 0.5)
1154
+ random_st.logistic(0.5, D_arr_like_0p5)
1155
+ random_st.logistic(D_arr_0p5, D_arr_0p5)
1156
+ random_st.logistic(D_arr_like_0p5, D_arr_like_0p5)
1157
+ random_st.logistic(D_arr_0p5, D_arr_0p5, size=1)
1158
+ random_st.logistic(D_arr_like_0p5, D_arr_like_0p5, size=1)
1159
+
1160
+ random_st.lognormal(0.5, 0.5)
1161
+ random_st.lognormal(0.5, 0.5, size=None)
1162
+ random_st.lognormal(0.5, 0.5, size=1)
1163
+ random_st.lognormal(D_arr_0p5, 0.5)
1164
+ random_st.lognormal(0.5, D_arr_0p5)
1165
+ random_st.lognormal(D_arr_0p5, 0.5, size=1)
1166
+ random_st.lognormal(0.5, D_arr_0p5, size=1)
1167
+ random_st.lognormal(D_arr_like_0p5, 0.5)
1168
+ random_st.lognormal(0.5, D_arr_like_0p5)
1169
+ random_st.lognormal(D_arr_0p5, D_arr_0p5)
1170
+ random_st.lognormal(D_arr_like_0p5, D_arr_like_0p5)
1171
+ random_st.lognormal(D_arr_0p5, D_arr_0p5, size=1)
1172
+ random_st.lognormal(D_arr_like_0p5, D_arr_like_0p5, size=1)
1173
+
1174
+ random_st.noncentral_chisquare(0.5, 0.5)
1175
+ random_st.noncentral_chisquare(0.5, 0.5, size=None)
1176
+ random_st.noncentral_chisquare(0.5, 0.5, size=1)
1177
+ random_st.noncentral_chisquare(D_arr_0p5, 0.5)
1178
+ random_st.noncentral_chisquare(0.5, D_arr_0p5)
1179
+ random_st.noncentral_chisquare(D_arr_0p5, 0.5, size=1)
1180
+ random_st.noncentral_chisquare(0.5, D_arr_0p5, size=1)
1181
+ random_st.noncentral_chisquare(D_arr_like_0p5, 0.5)
1182
+ random_st.noncentral_chisquare(0.5, D_arr_like_0p5)
1183
+ random_st.noncentral_chisquare(D_arr_0p5, D_arr_0p5)
1184
+ random_st.noncentral_chisquare(D_arr_like_0p5, D_arr_like_0p5)
1185
+ random_st.noncentral_chisquare(D_arr_0p5, D_arr_0p5, size=1)
1186
+ random_st.noncentral_chisquare(D_arr_like_0p5, D_arr_like_0p5, size=1)
1187
+
1188
+ random_st.normal(0.5, 0.5)
1189
+ random_st.normal(0.5, 0.5, size=None)
1190
+ random_st.normal(0.5, 0.5, size=1)
1191
+ random_st.normal(D_arr_0p5, 0.5)
1192
+ random_st.normal(0.5, D_arr_0p5)
1193
+ random_st.normal(D_arr_0p5, 0.5, size=1)
1194
+ random_st.normal(0.5, D_arr_0p5, size=1)
1195
+ random_st.normal(D_arr_like_0p5, 0.5)
1196
+ random_st.normal(0.5, D_arr_like_0p5)
1197
+ random_st.normal(D_arr_0p5, D_arr_0p5)
1198
+ random_st.normal(D_arr_like_0p5, D_arr_like_0p5)
1199
+ random_st.normal(D_arr_0p5, D_arr_0p5, size=1)
1200
+ random_st.normal(D_arr_like_0p5, D_arr_like_0p5, size=1)
1201
+
1202
+ random_st.triangular(0.1, 0.5, 0.9)
1203
+ random_st.triangular(0.1, 0.5, 0.9, size=None)
1204
+ random_st.triangular(0.1, 0.5, 0.9, size=1)
1205
+ random_st.triangular(D_arr_0p1, 0.5, 0.9)
1206
+ random_st.triangular(0.1, D_arr_0p5, 0.9)
1207
+ random_st.triangular(D_arr_0p1, 0.5, D_arr_like_0p9, size=1)
1208
+ random_st.triangular(0.1, D_arr_0p5, 0.9, size=1)
1209
+ random_st.triangular(D_arr_like_0p1, 0.5, D_arr_0p9)
1210
+ random_st.triangular(0.5, D_arr_like_0p5, 0.9)
1211
+ random_st.triangular(D_arr_0p1, D_arr_0p5, 0.9)
1212
+ random_st.triangular(D_arr_like_0p1, D_arr_like_0p5, 0.9)
1213
+ random_st.triangular(D_arr_0p1, D_arr_0p5, D_arr_0p9, size=1)
1214
+ random_st.triangular(D_arr_like_0p1, D_arr_like_0p5, D_arr_like_0p9, size=1)
1215
+
1216
+ random_st.noncentral_f(0.1, 0.5, 0.9)
1217
+ random_st.noncentral_f(0.1, 0.5, 0.9, size=None)
1218
+ random_st.noncentral_f(0.1, 0.5, 0.9, size=1)
1219
+ random_st.noncentral_f(D_arr_0p1, 0.5, 0.9)
1220
+ random_st.noncentral_f(0.1, D_arr_0p5, 0.9)
1221
+ random_st.noncentral_f(D_arr_0p1, 0.5, D_arr_like_0p9, size=1)
1222
+ random_st.noncentral_f(0.1, D_arr_0p5, 0.9, size=1)
1223
+ random_st.noncentral_f(D_arr_like_0p1, 0.5, D_arr_0p9)
1224
+ random_st.noncentral_f(0.5, D_arr_like_0p5, 0.9)
1225
+ random_st.noncentral_f(D_arr_0p1, D_arr_0p5, 0.9)
1226
+ random_st.noncentral_f(D_arr_like_0p1, D_arr_like_0p5, 0.9)
1227
+ random_st.noncentral_f(D_arr_0p1, D_arr_0p5, D_arr_0p9, size=1)
1228
+ random_st.noncentral_f(D_arr_like_0p1, D_arr_like_0p5, D_arr_like_0p9, size=1)
1229
+
1230
+ random_st.binomial(10, 0.5)
1231
+ random_st.binomial(10, 0.5, size=None)
1232
+ random_st.binomial(10, 0.5, size=1)
1233
+ random_st.binomial(I_arr_10, 0.5)
1234
+ random_st.binomial(10, D_arr_0p5)
1235
+ random_st.binomial(I_arr_10, 0.5, size=1)
1236
+ random_st.binomial(10, D_arr_0p5, size=1)
1237
+ random_st.binomial(I_arr_like_10, 0.5)
1238
+ random_st.binomial(10, D_arr_like_0p5)
1239
+ random_st.binomial(I_arr_10, D_arr_0p5)
1240
+ random_st.binomial(I_arr_like_10, D_arr_like_0p5)
1241
+ random_st.binomial(I_arr_10, D_arr_0p5, size=1)
1242
+ random_st.binomial(I_arr_like_10, D_arr_like_0p5, size=1)
1243
+
1244
+ random_st.negative_binomial(10, 0.5)
1245
+ random_st.negative_binomial(10, 0.5, size=None)
1246
+ random_st.negative_binomial(10, 0.5, size=1)
1247
+ random_st.negative_binomial(I_arr_10, 0.5)
1248
+ random_st.negative_binomial(10, D_arr_0p5)
1249
+ random_st.negative_binomial(I_arr_10, 0.5, size=1)
1250
+ random_st.negative_binomial(10, D_arr_0p5, size=1)
1251
+ random_st.negative_binomial(I_arr_like_10, 0.5)
1252
+ random_st.negative_binomial(10, D_arr_like_0p5)
1253
+ random_st.negative_binomial(I_arr_10, D_arr_0p5)
1254
+ random_st.negative_binomial(I_arr_like_10, D_arr_like_0p5)
1255
+ random_st.negative_binomial(I_arr_10, D_arr_0p5, size=1)
1256
+ random_st.negative_binomial(I_arr_like_10, D_arr_like_0p5, size=1)
1257
+
1258
+ random_st.hypergeometric(20, 20, 10)
1259
+ random_st.hypergeometric(20, 20, 10, size=None)
1260
+ random_st.hypergeometric(20, 20, 10, size=1)
1261
+ random_st.hypergeometric(I_arr_20, 20, 10)
1262
+ random_st.hypergeometric(20, I_arr_20, 10)
1263
+ random_st.hypergeometric(I_arr_20, 20, I_arr_like_10, size=1)
1264
+ random_st.hypergeometric(20, I_arr_20, 10, size=1)
1265
+ random_st.hypergeometric(I_arr_like_20, 20, I_arr_10)
1266
+ random_st.hypergeometric(20, I_arr_like_20, 10)
1267
+ random_st.hypergeometric(I_arr_20, I_arr_20, 10)
1268
+ random_st.hypergeometric(I_arr_like_20, I_arr_like_20, 10)
1269
+ random_st.hypergeometric(I_arr_20, I_arr_20, I_arr_10, size=1)
1270
+ random_st.hypergeometric(I_arr_like_20, I_arr_like_20, I_arr_like_10, size=1)
1271
+
1272
+ random_st.randint(0, 100)
1273
+ random_st.randint(100)
1274
+ random_st.randint([100])
1275
+ random_st.randint(0, [100])
1276
+
1277
+ random_st.randint(2, dtype=bool)
1278
+ random_st.randint(0, 2, dtype=bool)
1279
+ random_st.randint(I_bool_high_open, dtype=bool)
1280
+ random_st.randint(I_bool_low, I_bool_high_open, dtype=bool)
1281
+ random_st.randint(0, I_bool_high_open, dtype=bool)
1282
+
1283
+ random_st.randint(2, dtype=np.bool_)
1284
+ random_st.randint(0, 2, dtype=np.bool_)
1285
+ random_st.randint(I_bool_high_open, dtype=np.bool_)
1286
+ random_st.randint(I_bool_low, I_bool_high_open, dtype=np.bool_)
1287
+ random_st.randint(0, I_bool_high_open, dtype=np.bool_)
1288
+
1289
+ random_st.randint(256, dtype="u1")
1290
+ random_st.randint(0, 256, dtype="u1")
1291
+ random_st.randint(I_u1_high_open, dtype="u1")
1292
+ random_st.randint(I_u1_low, I_u1_high_open, dtype="u1")
1293
+ random_st.randint(0, I_u1_high_open, dtype="u1")
1294
+
1295
+ random_st.randint(256, dtype="uint8")
1296
+ random_st.randint(0, 256, dtype="uint8")
1297
+ random_st.randint(I_u1_high_open, dtype="uint8")
1298
+ random_st.randint(I_u1_low, I_u1_high_open, dtype="uint8")
1299
+ random_st.randint(0, I_u1_high_open, dtype="uint8")
1300
+
1301
+ random_st.randint(256, dtype=np.uint8)
1302
+ random_st.randint(0, 256, dtype=np.uint8)
1303
+ random_st.randint(I_u1_high_open, dtype=np.uint8)
1304
+ random_st.randint(I_u1_low, I_u1_high_open, dtype=np.uint8)
1305
+ random_st.randint(0, I_u1_high_open, dtype=np.uint8)
1306
+
1307
+ random_st.randint(65536, dtype="u2")
1308
+ random_st.randint(0, 65536, dtype="u2")
1309
+ random_st.randint(I_u2_high_open, dtype="u2")
1310
+ random_st.randint(I_u2_low, I_u2_high_open, dtype="u2")
1311
+ random_st.randint(0, I_u2_high_open, dtype="u2")
1312
+
1313
+ random_st.randint(65536, dtype="uint16")
1314
+ random_st.randint(0, 65536, dtype="uint16")
1315
+ random_st.randint(I_u2_high_open, dtype="uint16")
1316
+ random_st.randint(I_u2_low, I_u2_high_open, dtype="uint16")
1317
+ random_st.randint(0, I_u2_high_open, dtype="uint16")
1318
+
1319
+ random_st.randint(65536, dtype=np.uint16)
1320
+ random_st.randint(0, 65536, dtype=np.uint16)
1321
+ random_st.randint(I_u2_high_open, dtype=np.uint16)
1322
+ random_st.randint(I_u2_low, I_u2_high_open, dtype=np.uint16)
1323
+ random_st.randint(0, I_u2_high_open, dtype=np.uint16)
1324
+
1325
+ random_st.randint(4294967296, dtype="u4")
1326
+ random_st.randint(0, 4294967296, dtype="u4")
1327
+ random_st.randint(I_u4_high_open, dtype="u4")
1328
+ random_st.randint(I_u4_low, I_u4_high_open, dtype="u4")
1329
+ random_st.randint(0, I_u4_high_open, dtype="u4")
1330
+
1331
+ random_st.randint(4294967296, dtype="uint32")
1332
+ random_st.randint(0, 4294967296, dtype="uint32")
1333
+ random_st.randint(I_u4_high_open, dtype="uint32")
1334
+ random_st.randint(I_u4_low, I_u4_high_open, dtype="uint32")
1335
+ random_st.randint(0, I_u4_high_open, dtype="uint32")
1336
+
1337
+ random_st.randint(4294967296, dtype=np.uint32)
1338
+ random_st.randint(0, 4294967296, dtype=np.uint32)
1339
+ random_st.randint(I_u4_high_open, dtype=np.uint32)
1340
+ random_st.randint(I_u4_low, I_u4_high_open, dtype=np.uint32)
1341
+ random_st.randint(0, I_u4_high_open, dtype=np.uint32)
1342
+
1343
+
1344
+ random_st.randint(18446744073709551616, dtype="u8")
1345
+ random_st.randint(0, 18446744073709551616, dtype="u8")
1346
+ random_st.randint(I_u8_high_open, dtype="u8")
1347
+ random_st.randint(I_u8_low, I_u8_high_open, dtype="u8")
1348
+ random_st.randint(0, I_u8_high_open, dtype="u8")
1349
+
1350
+ random_st.randint(18446744073709551616, dtype="uint64")
1351
+ random_st.randint(0, 18446744073709551616, dtype="uint64")
1352
+ random_st.randint(I_u8_high_open, dtype="uint64")
1353
+ random_st.randint(I_u8_low, I_u8_high_open, dtype="uint64")
1354
+ random_st.randint(0, I_u8_high_open, dtype="uint64")
1355
+
1356
+ random_st.randint(18446744073709551616, dtype=np.uint64)
1357
+ random_st.randint(0, 18446744073709551616, dtype=np.uint64)
1358
+ random_st.randint(I_u8_high_open, dtype=np.uint64)
1359
+ random_st.randint(I_u8_low, I_u8_high_open, dtype=np.uint64)
1360
+ random_st.randint(0, I_u8_high_open, dtype=np.uint64)
1361
+
1362
+ random_st.randint(128, dtype="i1")
1363
+ random_st.randint(-128, 128, dtype="i1")
1364
+ random_st.randint(I_i1_high_open, dtype="i1")
1365
+ random_st.randint(I_i1_low, I_i1_high_open, dtype="i1")
1366
+ random_st.randint(-128, I_i1_high_open, dtype="i1")
1367
+
1368
+ random_st.randint(128, dtype="int8")
1369
+ random_st.randint(-128, 128, dtype="int8")
1370
+ random_st.randint(I_i1_high_open, dtype="int8")
1371
+ random_st.randint(I_i1_low, I_i1_high_open, dtype="int8")
1372
+ random_st.randint(-128, I_i1_high_open, dtype="int8")
1373
+
1374
+ random_st.randint(128, dtype=np.int8)
1375
+ random_st.randint(-128, 128, dtype=np.int8)
1376
+ random_st.randint(I_i1_high_open, dtype=np.int8)
1377
+ random_st.randint(I_i1_low, I_i1_high_open, dtype=np.int8)
1378
+ random_st.randint(-128, I_i1_high_open, dtype=np.int8)
1379
+
1380
+ random_st.randint(32768, dtype="i2")
1381
+ random_st.randint(-32768, 32768, dtype="i2")
1382
+ random_st.randint(I_i2_high_open, dtype="i2")
1383
+ random_st.randint(I_i2_low, I_i2_high_open, dtype="i2")
1384
+ random_st.randint(-32768, I_i2_high_open, dtype="i2")
1385
+ random_st.randint(32768, dtype="int16")
1386
+ random_st.randint(-32768, 32768, dtype="int16")
1387
+ random_st.randint(I_i2_high_open, dtype="int16")
1388
+ random_st.randint(I_i2_low, I_i2_high_open, dtype="int16")
1389
+ random_st.randint(-32768, I_i2_high_open, dtype="int16")
1390
+ random_st.randint(32768, dtype=np.int16)
1391
+ random_st.randint(-32768, 32768, dtype=np.int16)
1392
+ random_st.randint(I_i2_high_open, dtype=np.int16)
1393
+ random_st.randint(I_i2_low, I_i2_high_open, dtype=np.int16)
1394
+ random_st.randint(-32768, I_i2_high_open, dtype=np.int16)
1395
+
1396
+ random_st.randint(2147483648, dtype="i4")
1397
+ random_st.randint(-2147483648, 2147483648, dtype="i4")
1398
+ random_st.randint(I_i4_high_open, dtype="i4")
1399
+ random_st.randint(I_i4_low, I_i4_high_open, dtype="i4")
1400
+ random_st.randint(-2147483648, I_i4_high_open, dtype="i4")
1401
+
1402
+ random_st.randint(2147483648, dtype="int32")
1403
+ random_st.randint(-2147483648, 2147483648, dtype="int32")
1404
+ random_st.randint(I_i4_high_open, dtype="int32")
1405
+ random_st.randint(I_i4_low, I_i4_high_open, dtype="int32")
1406
+ random_st.randint(-2147483648, I_i4_high_open, dtype="int32")
1407
+
1408
+ random_st.randint(2147483648, dtype=np.int32)
1409
+ random_st.randint(-2147483648, 2147483648, dtype=np.int32)
1410
+ random_st.randint(I_i4_high_open, dtype=np.int32)
1411
+ random_st.randint(I_i4_low, I_i4_high_open, dtype=np.int32)
1412
+ random_st.randint(-2147483648, I_i4_high_open, dtype=np.int32)
1413
+
1414
+ random_st.randint(9223372036854775808, dtype="i8")
1415
+ random_st.randint(-9223372036854775808, 9223372036854775808, dtype="i8")
1416
+ random_st.randint(I_i8_high_open, dtype="i8")
1417
+ random_st.randint(I_i8_low, I_i8_high_open, dtype="i8")
1418
+ random_st.randint(-9223372036854775808, I_i8_high_open, dtype="i8")
1419
+
1420
+ random_st.randint(9223372036854775808, dtype="int64")
1421
+ random_st.randint(-9223372036854775808, 9223372036854775808, dtype="int64")
1422
+ random_st.randint(I_i8_high_open, dtype="int64")
1423
+ random_st.randint(I_i8_low, I_i8_high_open, dtype="int64")
1424
+ random_st.randint(-9223372036854775808, I_i8_high_open, dtype="int64")
1425
+
1426
+ random_st.randint(9223372036854775808, dtype=np.int64)
1427
+ random_st.randint(-9223372036854775808, 9223372036854775808, dtype=np.int64)
1428
+ random_st.randint(I_i8_high_open, dtype=np.int64)
1429
+ random_st.randint(I_i8_low, I_i8_high_open, dtype=np.int64)
1430
+ random_st.randint(-9223372036854775808, I_i8_high_open, dtype=np.int64)
1431
+
1432
+ bg: np.random.BitGenerator = random_st._bit_generator
1433
+
1434
+ random_st.bytes(2)
1435
+
1436
+ random_st.choice(5)
1437
+ random_st.choice(5, 3)
1438
+ random_st.choice(5, 3, replace=True)
1439
+ random_st.choice(5, 3, p=[1 / 5] * 5)
1440
+ random_st.choice(5, 3, p=[1 / 5] * 5, replace=False)
1441
+
1442
+ random_st.choice(["pooh", "rabbit", "piglet", "Christopher"])
1443
+ random_st.choice(["pooh", "rabbit", "piglet", "Christopher"], 3)
1444
+ random_st.choice(["pooh", "rabbit", "piglet", "Christopher"], 3, p=[1 / 4] * 4)
1445
+ random_st.choice(["pooh", "rabbit", "piglet", "Christopher"], 3, replace=True)
1446
+ random_st.choice(["pooh", "rabbit", "piglet", "Christopher"], 3, replace=False, p=np.array([1 / 8, 1 / 8, 1 / 2, 1 / 4]))
1447
+
1448
+ random_st.dirichlet([0.5, 0.5])
1449
+ random_st.dirichlet(np.array([0.5, 0.5]))
1450
+ random_st.dirichlet(np.array([0.5, 0.5]), size=3)
1451
+
1452
+ random_st.multinomial(20, [1 / 6.0] * 6)
1453
+ random_st.multinomial(20, np.array([0.5, 0.5]))
1454
+ random_st.multinomial(20, [1 / 6.0] * 6, size=2)
1455
+
1456
+ random_st.multivariate_normal([0.0], [[1.0]])
1457
+ random_st.multivariate_normal([0.0], np.array([[1.0]]))
1458
+ random_st.multivariate_normal(np.array([0.0]), [[1.0]])
1459
+ random_st.multivariate_normal([0.0], np.array([[1.0]]))
1460
+
1461
+ random_st.permutation(10)
1462
+ random_st.permutation([1, 2, 3, 4])
1463
+ random_st.permutation(np.array([1, 2, 3, 4]))
1464
+ random_st.permutation(D_2D)
1465
+
1466
+ random_st.shuffle(np.arange(10))
1467
+ random_st.shuffle([1, 2, 3, 4, 5])
1468
+ random_st.shuffle(D_2D)
1469
+
1470
+ np.random.RandomState(SEED_PCG64)
1471
+ np.random.RandomState(0)
1472
+ np.random.RandomState([0, 1, 2])
1473
+ random_st.__str__()
1474
+ random_st.__repr__()
1475
+ random_st_state = random_st.__getstate__()
1476
+ random_st.__setstate__(random_st_state)
1477
+ random_st.seed()
1478
+ random_st.seed(1)
1479
+ random_st.seed([0, 1])
1480
+ random_st_get_state = random_st.get_state()
1481
+ random_st_get_state_legacy = random_st.get_state(legacy=True)
1482
+ random_st.set_state(random_st_get_state)
1483
+
1484
+ random_st.rand()
1485
+ random_st.rand(1)
1486
+ random_st.rand(1, 2)
1487
+ random_st.randn()
1488
+ random_st.randn(1)
1489
+ random_st.randn(1, 2)
1490
+ random_st.random_sample()
1491
+ random_st.random_sample(1)
1492
+ random_st.random_sample(size=(1, 2))
1493
+
1494
+ random_st.tomaxint()
1495
+ random_st.tomaxint(1)
1496
+ random_st.tomaxint((1,))
1497
+
1498
+ np.random.set_bit_generator(SEED_PCG64)
1499
+ np.random.get_bit_generator()
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/pass/simple.py ADDED
@@ -0,0 +1,165 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """Simple expression that should pass with mypy."""
2
+ import operator
3
+
4
+ import numpy as np
5
+ from collections.abc import Iterable
6
+
7
+ # Basic checks
8
+ array = np.array([1, 2])
9
+
10
+
11
+ def ndarray_func(x):
12
+ # type: (np.ndarray) -> np.ndarray
13
+ return x
14
+
15
+
16
+ ndarray_func(np.array([1, 2]))
17
+ array == 1
18
+ array.dtype == float
19
+
20
+ # Dtype construction
21
+ np.dtype(float)
22
+ np.dtype(np.float64)
23
+ np.dtype(None)
24
+ np.dtype("float64")
25
+ np.dtype(np.dtype(float))
26
+ np.dtype(("U", 10))
27
+ np.dtype((np.int32, (2, 2)))
28
+ # Define the arguments on the previous line to prevent bidirectional
29
+ # type inference in mypy from broadening the types.
30
+ two_tuples_dtype = [("R", "u1"), ("G", "u1"), ("B", "u1")]
31
+ np.dtype(two_tuples_dtype)
32
+
33
+ three_tuples_dtype = [("R", "u1", 2)]
34
+ np.dtype(three_tuples_dtype)
35
+
36
+ mixed_tuples_dtype = [("R", "u1"), ("G", np.str_, 1)]
37
+ np.dtype(mixed_tuples_dtype)
38
+
39
+ shape_tuple_dtype = [("R", "u1", (2, 2))]
40
+ np.dtype(shape_tuple_dtype)
41
+
42
+ shape_like_dtype = [("R", "u1", (2, 2)), ("G", np.str_, 1)]
43
+ np.dtype(shape_like_dtype)
44
+
45
+ object_dtype = [("field1", object)]
46
+ np.dtype(object_dtype)
47
+
48
+ np.dtype((np.int32, (np.int8, 4)))
49
+
50
+ # Dtype comparison
51
+ np.dtype(float) == float
52
+ np.dtype(float) != np.float64
53
+ np.dtype(float) < None
54
+ np.dtype(float) <= "float64"
55
+ np.dtype(float) > np.dtype(float)
56
+ np.dtype(float) >= np.dtype(("U", 10))
57
+
58
+ # Iteration and indexing
59
+ def iterable_func(x):
60
+ # type: (Iterable) -> Iterable
61
+ return x
62
+
63
+
64
+ iterable_func(array)
65
+ [element for element in array]
66
+ iter(array)
67
+ zip(array, array)
68
+ array[1]
69
+ array[:]
70
+ array[...]
71
+ array[:] = 0
72
+
73
+ array_2d = np.ones((3, 3))
74
+ array_2d[:2, :2]
75
+ array_2d[..., 0]
76
+ array_2d[:2, :2] = 0
77
+
78
+ # Other special methods
79
+ len(array)
80
+ str(array)
81
+ array_scalar = np.array(1)
82
+ int(array_scalar)
83
+ float(array_scalar)
84
+ # currently does not work due to https://github.com/python/typeshed/issues/1904
85
+ # complex(array_scalar)
86
+ bytes(array_scalar)
87
+ operator.index(array_scalar)
88
+ bool(array_scalar)
89
+
90
+ # comparisons
91
+ array < 1
92
+ array <= 1
93
+ array == 1
94
+ array != 1
95
+ array > 1
96
+ array >= 1
97
+ 1 < array
98
+ 1 <= array
99
+ 1 == array
100
+ 1 != array
101
+ 1 > array
102
+ 1 >= array
103
+
104
+ # binary arithmetic
105
+ array + 1
106
+ 1 + array
107
+ array += 1
108
+
109
+ array - 1
110
+ 1 - array
111
+ array -= 1
112
+
113
+ array * 1
114
+ 1 * array
115
+ array *= 1
116
+
117
+ nonzero_array = np.array([1, 2])
118
+ array / 1
119
+ 1 / nonzero_array
120
+ float_array = np.array([1.0, 2.0])
121
+ float_array /= 1
122
+
123
+ array // 1
124
+ 1 // nonzero_array
125
+ array //= 1
126
+
127
+ array % 1
128
+ 1 % nonzero_array
129
+ array %= 1
130
+
131
+ divmod(array, 1)
132
+ divmod(1, nonzero_array)
133
+
134
+ array ** 1
135
+ 1 ** array
136
+ array **= 1
137
+
138
+ array << 1
139
+ 1 << array
140
+ array <<= 1
141
+
142
+ array >> 1
143
+ 1 >> array
144
+ array >>= 1
145
+
146
+ array & 1
147
+ 1 & array
148
+ array &= 1
149
+
150
+ array ^ 1
151
+ 1 ^ array
152
+ array ^= 1
153
+
154
+ array | 1
155
+ 1 | array
156
+ array |= 1
157
+
158
+ # unary arithmetic
159
+ -array
160
+ +array
161
+ abs(array)
162
+ ~array
163
+
164
+ # Other methods
165
+ np.array([1, 2]).transpose()
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/reveal/arithmetic.pyi ADDED
@@ -0,0 +1,516 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import sys
2
+ from typing import Any
3
+
4
+ import numpy as np
5
+ import numpy.typing as npt
6
+ from numpy._typing import _32Bit,_64Bit, _128Bit
7
+
8
+ if sys.version_info >= (3, 11):
9
+ from typing import assert_type
10
+ else:
11
+ from typing_extensions import assert_type
12
+
13
+ # Can't directly import `np.float128` as it is not available on all platforms
14
+ f16: np.floating[_128Bit]
15
+
16
+ c16 = np.complex128()
17
+ f8 = np.float64()
18
+ i8 = np.int64()
19
+ u8 = np.uint64()
20
+
21
+ c8 = np.complex64()
22
+ f4 = np.float32()
23
+ i4 = np.int32()
24
+ u4 = np.uint32()
25
+
26
+ dt = np.datetime64(0, "D")
27
+ td = np.timedelta64(0, "D")
28
+
29
+ b_ = np.bool_()
30
+
31
+ b = bool()
32
+ c = complex()
33
+ f = float()
34
+ i = int()
35
+
36
+ AR_b: npt.NDArray[np.bool_]
37
+ AR_u: npt.NDArray[np.uint32]
38
+ AR_i: npt.NDArray[np.int64]
39
+ AR_f: npt.NDArray[np.float64]
40
+ AR_c: npt.NDArray[np.complex128]
41
+ AR_m: npt.NDArray[np.timedelta64]
42
+ AR_M: npt.NDArray[np.datetime64]
43
+ AR_O: npt.NDArray[np.object_]
44
+ AR_number: npt.NDArray[np.number[Any]]
45
+
46
+ AR_LIKE_b: list[bool]
47
+ AR_LIKE_u: list[np.uint32]
48
+ AR_LIKE_i: list[int]
49
+ AR_LIKE_f: list[float]
50
+ AR_LIKE_c: list[complex]
51
+ AR_LIKE_m: list[np.timedelta64]
52
+ AR_LIKE_M: list[np.datetime64]
53
+ AR_LIKE_O: list[np.object_]
54
+
55
+ # Array subtraction
56
+
57
+ assert_type(AR_number - AR_number, npt.NDArray[np.number[Any]])
58
+
59
+ assert_type(AR_b - AR_LIKE_u, npt.NDArray[np.unsignedinteger[Any]])
60
+ assert_type(AR_b - AR_LIKE_i, npt.NDArray[np.signedinteger[Any]])
61
+ assert_type(AR_b - AR_LIKE_f, npt.NDArray[np.floating[Any]])
62
+ assert_type(AR_b - AR_LIKE_c, npt.NDArray[np.complexfloating[Any, Any]])
63
+ assert_type(AR_b - AR_LIKE_m, npt.NDArray[np.timedelta64])
64
+ assert_type(AR_b - AR_LIKE_O, Any)
65
+
66
+ assert_type(AR_LIKE_u - AR_b, npt.NDArray[np.unsignedinteger[Any]])
67
+ assert_type(AR_LIKE_i - AR_b, npt.NDArray[np.signedinteger[Any]])
68
+ assert_type(AR_LIKE_f - AR_b, npt.NDArray[np.floating[Any]])
69
+ assert_type(AR_LIKE_c - AR_b, npt.NDArray[np.complexfloating[Any, Any]])
70
+ assert_type(AR_LIKE_m - AR_b, npt.NDArray[np.timedelta64])
71
+ assert_type(AR_LIKE_M - AR_b, npt.NDArray[np.datetime64])
72
+ assert_type(AR_LIKE_O - AR_b, Any)
73
+
74
+ assert_type(AR_u - AR_LIKE_b, npt.NDArray[np.unsignedinteger[Any]])
75
+ assert_type(AR_u - AR_LIKE_u, npt.NDArray[np.unsignedinteger[Any]])
76
+ assert_type(AR_u - AR_LIKE_i, npt.NDArray[np.signedinteger[Any]])
77
+ assert_type(AR_u - AR_LIKE_f, npt.NDArray[np.floating[Any]])
78
+ assert_type(AR_u - AR_LIKE_c, npt.NDArray[np.complexfloating[Any, Any]])
79
+ assert_type(AR_u - AR_LIKE_m, npt.NDArray[np.timedelta64])
80
+ assert_type(AR_u - AR_LIKE_O, Any)
81
+
82
+ assert_type(AR_LIKE_b - AR_u, npt.NDArray[np.unsignedinteger[Any]])
83
+ assert_type(AR_LIKE_u - AR_u, npt.NDArray[np.unsignedinteger[Any]])
84
+ assert_type(AR_LIKE_i - AR_u, npt.NDArray[np.signedinteger[Any]])
85
+ assert_type(AR_LIKE_f - AR_u, npt.NDArray[np.floating[Any]])
86
+ assert_type(AR_LIKE_c - AR_u, npt.NDArray[np.complexfloating[Any, Any]])
87
+ assert_type(AR_LIKE_m - AR_u, npt.NDArray[np.timedelta64])
88
+ assert_type(AR_LIKE_M - AR_u, npt.NDArray[np.datetime64])
89
+ assert_type(AR_LIKE_O - AR_u, Any)
90
+
91
+ assert_type(AR_i - AR_LIKE_b, npt.NDArray[np.signedinteger[Any]])
92
+ assert_type(AR_i - AR_LIKE_u, npt.NDArray[np.signedinteger[Any]])
93
+ assert_type(AR_i - AR_LIKE_i, npt.NDArray[np.signedinteger[Any]])
94
+ assert_type(AR_i - AR_LIKE_f, npt.NDArray[np.floating[Any]])
95
+ assert_type(AR_i - AR_LIKE_c, npt.NDArray[np.complexfloating[Any, Any]])
96
+ assert_type(AR_i - AR_LIKE_m, npt.NDArray[np.timedelta64])
97
+ assert_type(AR_i - AR_LIKE_O, Any)
98
+
99
+ assert_type(AR_LIKE_b - AR_i, npt.NDArray[np.signedinteger[Any]])
100
+ assert_type(AR_LIKE_u - AR_i, npt.NDArray[np.signedinteger[Any]])
101
+ assert_type(AR_LIKE_i - AR_i, npt.NDArray[np.signedinteger[Any]])
102
+ assert_type(AR_LIKE_f - AR_i, npt.NDArray[np.floating[Any]])
103
+ assert_type(AR_LIKE_c - AR_i, npt.NDArray[np.complexfloating[Any, Any]])
104
+ assert_type(AR_LIKE_m - AR_i, npt.NDArray[np.timedelta64])
105
+ assert_type(AR_LIKE_M - AR_i, npt.NDArray[np.datetime64])
106
+ assert_type(AR_LIKE_O - AR_i, Any)
107
+
108
+ assert_type(AR_f - AR_LIKE_b, npt.NDArray[np.floating[Any]])
109
+ assert_type(AR_f - AR_LIKE_u, npt.NDArray[np.floating[Any]])
110
+ assert_type(AR_f - AR_LIKE_i, npt.NDArray[np.floating[Any]])
111
+ assert_type(AR_f - AR_LIKE_f, npt.NDArray[np.floating[Any]])
112
+ assert_type(AR_f - AR_LIKE_c, npt.NDArray[np.complexfloating[Any, Any]])
113
+ assert_type(AR_f - AR_LIKE_O, Any)
114
+
115
+ assert_type(AR_LIKE_b - AR_f, npt.NDArray[np.floating[Any]])
116
+ assert_type(AR_LIKE_u - AR_f, npt.NDArray[np.floating[Any]])
117
+ assert_type(AR_LIKE_i - AR_f, npt.NDArray[np.floating[Any]])
118
+ assert_type(AR_LIKE_f - AR_f, npt.NDArray[np.floating[Any]])
119
+ assert_type(AR_LIKE_c - AR_f, npt.NDArray[np.complexfloating[Any, Any]])
120
+ assert_type(AR_LIKE_O - AR_f, Any)
121
+
122
+ assert_type(AR_c - AR_LIKE_b, npt.NDArray[np.complexfloating[Any, Any]])
123
+ assert_type(AR_c - AR_LIKE_u, npt.NDArray[np.complexfloating[Any, Any]])
124
+ assert_type(AR_c - AR_LIKE_i, npt.NDArray[np.complexfloating[Any, Any]])
125
+ assert_type(AR_c - AR_LIKE_f, npt.NDArray[np.complexfloating[Any, Any]])
126
+ assert_type(AR_c - AR_LIKE_c, npt.NDArray[np.complexfloating[Any, Any]])
127
+ assert_type(AR_c - AR_LIKE_O, Any)
128
+
129
+ assert_type(AR_LIKE_b - AR_c, npt.NDArray[np.complexfloating[Any, Any]])
130
+ assert_type(AR_LIKE_u - AR_c, npt.NDArray[np.complexfloating[Any, Any]])
131
+ assert_type(AR_LIKE_i - AR_c, npt.NDArray[np.complexfloating[Any, Any]])
132
+ assert_type(AR_LIKE_f - AR_c, npt.NDArray[np.complexfloating[Any, Any]])
133
+ assert_type(AR_LIKE_c - AR_c, npt.NDArray[np.complexfloating[Any, Any]])
134
+ assert_type(AR_LIKE_O - AR_c, Any)
135
+
136
+ assert_type(AR_m - AR_LIKE_b, npt.NDArray[np.timedelta64])
137
+ assert_type(AR_m - AR_LIKE_u, npt.NDArray[np.timedelta64])
138
+ assert_type(AR_m - AR_LIKE_i, npt.NDArray[np.timedelta64])
139
+ assert_type(AR_m - AR_LIKE_m, npt.NDArray[np.timedelta64])
140
+ assert_type(AR_m - AR_LIKE_O, Any)
141
+
142
+ assert_type(AR_LIKE_b - AR_m, npt.NDArray[np.timedelta64])
143
+ assert_type(AR_LIKE_u - AR_m, npt.NDArray[np.timedelta64])
144
+ assert_type(AR_LIKE_i - AR_m, npt.NDArray[np.timedelta64])
145
+ assert_type(AR_LIKE_m - AR_m, npt.NDArray[np.timedelta64])
146
+ assert_type(AR_LIKE_M - AR_m, npt.NDArray[np.datetime64])
147
+ assert_type(AR_LIKE_O - AR_m, Any)
148
+
149
+ assert_type(AR_M - AR_LIKE_b, npt.NDArray[np.datetime64])
150
+ assert_type(AR_M - AR_LIKE_u, npt.NDArray[np.datetime64])
151
+ assert_type(AR_M - AR_LIKE_i, npt.NDArray[np.datetime64])
152
+ assert_type(AR_M - AR_LIKE_m, npt.NDArray[np.datetime64])
153
+ assert_type(AR_M - AR_LIKE_M, npt.NDArray[np.timedelta64])
154
+ assert_type(AR_M - AR_LIKE_O, Any)
155
+
156
+ assert_type(AR_LIKE_M - AR_M, npt.NDArray[np.timedelta64])
157
+ assert_type(AR_LIKE_O - AR_M, Any)
158
+
159
+ assert_type(AR_O - AR_LIKE_b, Any)
160
+ assert_type(AR_O - AR_LIKE_u, Any)
161
+ assert_type(AR_O - AR_LIKE_i, Any)
162
+ assert_type(AR_O - AR_LIKE_f, Any)
163
+ assert_type(AR_O - AR_LIKE_c, Any)
164
+ assert_type(AR_O - AR_LIKE_m, Any)
165
+ assert_type(AR_O - AR_LIKE_M, Any)
166
+ assert_type(AR_O - AR_LIKE_O, Any)
167
+
168
+ assert_type(AR_LIKE_b - AR_O, Any)
169
+ assert_type(AR_LIKE_u - AR_O, Any)
170
+ assert_type(AR_LIKE_i - AR_O, Any)
171
+ assert_type(AR_LIKE_f - AR_O, Any)
172
+ assert_type(AR_LIKE_c - AR_O, Any)
173
+ assert_type(AR_LIKE_m - AR_O, Any)
174
+ assert_type(AR_LIKE_M - AR_O, Any)
175
+ assert_type(AR_LIKE_O - AR_O, Any)
176
+
177
+ # Array floor division
178
+
179
+ assert_type(AR_b // AR_LIKE_b, npt.NDArray[np.int8])
180
+ assert_type(AR_b // AR_LIKE_u, npt.NDArray[np.unsignedinteger[Any]])
181
+ assert_type(AR_b // AR_LIKE_i, npt.NDArray[np.signedinteger[Any]])
182
+ assert_type(AR_b // AR_LIKE_f, npt.NDArray[np.floating[Any]])
183
+ assert_type(AR_b // AR_LIKE_O, Any)
184
+
185
+ assert_type(AR_LIKE_b // AR_b, npt.NDArray[np.int8])
186
+ assert_type(AR_LIKE_u // AR_b, npt.NDArray[np.unsignedinteger[Any]])
187
+ assert_type(AR_LIKE_i // AR_b, npt.NDArray[np.signedinteger[Any]])
188
+ assert_type(AR_LIKE_f // AR_b, npt.NDArray[np.floating[Any]])
189
+ assert_type(AR_LIKE_O // AR_b, Any)
190
+
191
+ assert_type(AR_u // AR_LIKE_b, npt.NDArray[np.unsignedinteger[Any]])
192
+ assert_type(AR_u // AR_LIKE_u, npt.NDArray[np.unsignedinteger[Any]])
193
+ assert_type(AR_u // AR_LIKE_i, npt.NDArray[np.signedinteger[Any]])
194
+ assert_type(AR_u // AR_LIKE_f, npt.NDArray[np.floating[Any]])
195
+ assert_type(AR_u // AR_LIKE_O, Any)
196
+
197
+ assert_type(AR_LIKE_b // AR_u, npt.NDArray[np.unsignedinteger[Any]])
198
+ assert_type(AR_LIKE_u // AR_u, npt.NDArray[np.unsignedinteger[Any]])
199
+ assert_type(AR_LIKE_i // AR_u, npt.NDArray[np.signedinteger[Any]])
200
+ assert_type(AR_LIKE_f // AR_u, npt.NDArray[np.floating[Any]])
201
+ assert_type(AR_LIKE_m // AR_u, npt.NDArray[np.timedelta64])
202
+ assert_type(AR_LIKE_O // AR_u, Any)
203
+
204
+ assert_type(AR_i // AR_LIKE_b, npt.NDArray[np.signedinteger[Any]])
205
+ assert_type(AR_i // AR_LIKE_u, npt.NDArray[np.signedinteger[Any]])
206
+ assert_type(AR_i // AR_LIKE_i, npt.NDArray[np.signedinteger[Any]])
207
+ assert_type(AR_i // AR_LIKE_f, npt.NDArray[np.floating[Any]])
208
+ assert_type(AR_i // AR_LIKE_O, Any)
209
+
210
+ assert_type(AR_LIKE_b // AR_i, npt.NDArray[np.signedinteger[Any]])
211
+ assert_type(AR_LIKE_u // AR_i, npt.NDArray[np.signedinteger[Any]])
212
+ assert_type(AR_LIKE_i // AR_i, npt.NDArray[np.signedinteger[Any]])
213
+ assert_type(AR_LIKE_f // AR_i, npt.NDArray[np.floating[Any]])
214
+ assert_type(AR_LIKE_m // AR_i, npt.NDArray[np.timedelta64])
215
+ assert_type(AR_LIKE_O // AR_i, Any)
216
+
217
+ assert_type(AR_f // AR_LIKE_b, npt.NDArray[np.floating[Any]])
218
+ assert_type(AR_f // AR_LIKE_u, npt.NDArray[np.floating[Any]])
219
+ assert_type(AR_f // AR_LIKE_i, npt.NDArray[np.floating[Any]])
220
+ assert_type(AR_f // AR_LIKE_f, npt.NDArray[np.floating[Any]])
221
+ assert_type(AR_f // AR_LIKE_O, Any)
222
+
223
+ assert_type(AR_LIKE_b // AR_f, npt.NDArray[np.floating[Any]])
224
+ assert_type(AR_LIKE_u // AR_f, npt.NDArray[np.floating[Any]])
225
+ assert_type(AR_LIKE_i // AR_f, npt.NDArray[np.floating[Any]])
226
+ assert_type(AR_LIKE_f // AR_f, npt.NDArray[np.floating[Any]])
227
+ assert_type(AR_LIKE_m // AR_f, npt.NDArray[np.timedelta64])
228
+ assert_type(AR_LIKE_O // AR_f, Any)
229
+
230
+ assert_type(AR_m // AR_LIKE_u, npt.NDArray[np.timedelta64])
231
+ assert_type(AR_m // AR_LIKE_i, npt.NDArray[np.timedelta64])
232
+ assert_type(AR_m // AR_LIKE_f, npt.NDArray[np.timedelta64])
233
+ assert_type(AR_m // AR_LIKE_m, npt.NDArray[np.int64])
234
+ assert_type(AR_m // AR_LIKE_O, Any)
235
+
236
+ assert_type(AR_LIKE_m // AR_m, npt.NDArray[np.int64])
237
+ assert_type(AR_LIKE_O // AR_m, Any)
238
+
239
+ assert_type(AR_O // AR_LIKE_b, Any)
240
+ assert_type(AR_O // AR_LIKE_u, Any)
241
+ assert_type(AR_O // AR_LIKE_i, Any)
242
+ assert_type(AR_O // AR_LIKE_f, Any)
243
+ assert_type(AR_O // AR_LIKE_m, Any)
244
+ assert_type(AR_O // AR_LIKE_M, Any)
245
+ assert_type(AR_O // AR_LIKE_O, Any)
246
+
247
+ assert_type(AR_LIKE_b // AR_O, Any)
248
+ assert_type(AR_LIKE_u // AR_O, Any)
249
+ assert_type(AR_LIKE_i // AR_O, Any)
250
+ assert_type(AR_LIKE_f // AR_O, Any)
251
+ assert_type(AR_LIKE_m // AR_O, Any)
252
+ assert_type(AR_LIKE_M // AR_O, Any)
253
+ assert_type(AR_LIKE_O // AR_O, Any)
254
+
255
+ # unary ops
256
+
257
+ assert_type(-f16, np.floating[_128Bit])
258
+ assert_type(-c16, np.complex128)
259
+ assert_type(-c8, np.complex64)
260
+ assert_type(-f8, np.float64)
261
+ assert_type(-f4, np.float32)
262
+ assert_type(-i8, np.int64)
263
+ assert_type(-i4, np.int32)
264
+ assert_type(-u8, np.uint64)
265
+ assert_type(-u4, np.uint32)
266
+ assert_type(-td, np.timedelta64)
267
+ assert_type(-AR_f, npt.NDArray[np.float64])
268
+
269
+ assert_type(+f16, np.floating[_128Bit])
270
+ assert_type(+c16, np.complex128)
271
+ assert_type(+c8, np.complex64)
272
+ assert_type(+f8, np.float64)
273
+ assert_type(+f4, np.float32)
274
+ assert_type(+i8, np.int64)
275
+ assert_type(+i4, np.int32)
276
+ assert_type(+u8, np.uint64)
277
+ assert_type(+u4, np.uint32)
278
+ assert_type(+td, np.timedelta64)
279
+ assert_type(+AR_f, npt.NDArray[np.float64])
280
+
281
+ assert_type(abs(f16), np.floating[_128Bit])
282
+ assert_type(abs(c16), np.float64)
283
+ assert_type(abs(c8), np.float32)
284
+ assert_type(abs(f8), np.float64)
285
+ assert_type(abs(f4), np.float32)
286
+ assert_type(abs(i8), np.int64)
287
+ assert_type(abs(i4), np.int32)
288
+ assert_type(abs(u8), np.uint64)
289
+ assert_type(abs(u4), np.uint32)
290
+ assert_type(abs(td), np.timedelta64)
291
+ assert_type(abs(b_), np.bool_)
292
+
293
+ # Time structures
294
+
295
+ assert_type(dt + td, np.datetime64)
296
+ assert_type(dt + i, np.datetime64)
297
+ assert_type(dt + i4, np.datetime64)
298
+ assert_type(dt + i8, np.datetime64)
299
+ assert_type(dt - dt, np.timedelta64)
300
+ assert_type(dt - i, np.datetime64)
301
+ assert_type(dt - i4, np.datetime64)
302
+ assert_type(dt - i8, np.datetime64)
303
+
304
+ assert_type(td + td, np.timedelta64)
305
+ assert_type(td + i, np.timedelta64)
306
+ assert_type(td + i4, np.timedelta64)
307
+ assert_type(td + i8, np.timedelta64)
308
+ assert_type(td - td, np.timedelta64)
309
+ assert_type(td - i, np.timedelta64)
310
+ assert_type(td - i4, np.timedelta64)
311
+ assert_type(td - i8, np.timedelta64)
312
+ assert_type(td / f, np.timedelta64)
313
+ assert_type(td / f4, np.timedelta64)
314
+ assert_type(td / f8, np.timedelta64)
315
+ assert_type(td / td, np.float64)
316
+ assert_type(td // td, np.int64)
317
+
318
+ # boolean
319
+
320
+ assert_type(b_ / b, np.float64)
321
+ assert_type(b_ / b_, np.float64)
322
+ assert_type(b_ / i, np.float64)
323
+ assert_type(b_ / i8, np.float64)
324
+ assert_type(b_ / i4, np.float64)
325
+ assert_type(b_ / u8, np.float64)
326
+ assert_type(b_ / u4, np.float64)
327
+ assert_type(b_ / f, np.float64)
328
+ assert_type(b_ / f16, np.floating[_128Bit])
329
+ assert_type(b_ / f8, np.float64)
330
+ assert_type(b_ / f4, np.float32)
331
+ assert_type(b_ / c, np.complex128)
332
+ assert_type(b_ / c16, np.complex128)
333
+ assert_type(b_ / c8, np.complex64)
334
+
335
+ assert_type(b / b_, np.float64)
336
+ assert_type(b_ / b_, np.float64)
337
+ assert_type(i / b_, np.float64)
338
+ assert_type(i8 / b_, np.float64)
339
+ assert_type(i4 / b_, np.float64)
340
+ assert_type(u8 / b_, np.float64)
341
+ assert_type(u4 / b_, np.float64)
342
+ assert_type(f / b_, np.float64)
343
+ assert_type(f16 / b_, np.floating[_128Bit])
344
+ assert_type(f8 / b_, np.float64)
345
+ assert_type(f4 / b_, np.float32)
346
+ assert_type(c / b_, np.complex128)
347
+ assert_type(c16 / b_, np.complex128)
348
+ assert_type(c8 / b_, np.complex64)
349
+
350
+ # Complex
351
+
352
+ assert_type(c16 + f16, np.complexfloating[_64Bit | _128Bit, _64Bit | _128Bit])
353
+ assert_type(c16 + c16, np.complex128)
354
+ assert_type(c16 + f8, np.complex128)
355
+ assert_type(c16 + i8, np.complex128)
356
+ assert_type(c16 + c8, np.complexfloating[_32Bit | _64Bit, _32Bit | _64Bit])
357
+ assert_type(c16 + f4, np.complexfloating[_32Bit | _64Bit, _32Bit | _64Bit])
358
+ assert_type(c16 + i4, np.complexfloating[_32Bit | _64Bit, _32Bit | _64Bit])
359
+ assert_type(c16 + b_, np.complex128)
360
+ assert_type(c16 + b, np.complex128)
361
+ assert_type(c16 + c, np.complex128)
362
+ assert_type(c16 + f, np.complex128)
363
+ assert_type(c16 + AR_f, npt.NDArray[np.complexfloating[Any, Any]])
364
+
365
+ assert_type(f16 + c16, np.complexfloating[_64Bit | _128Bit, _64Bit | _128Bit])
366
+ assert_type(c16 + c16, np.complex128)
367
+ assert_type(f8 + c16, np.complex128)
368
+ assert_type(i8 + c16, np.complex128)
369
+ assert_type(c8 + c16, np.complexfloating[_32Bit | _64Bit, _32Bit | _64Bit])
370
+ assert_type(f4 + c16, np.complexfloating[_32Bit | _64Bit, _32Bit | _64Bit])
371
+ assert_type(i4 + c16, np.complexfloating[_32Bit | _64Bit, _32Bit | _64Bit])
372
+ assert_type(b_ + c16, np.complex128)
373
+ assert_type(b + c16, np.complex128)
374
+ assert_type(c + c16, np.complex128)
375
+ assert_type(f + c16, np.complex128)
376
+ assert_type(AR_f + c16, npt.NDArray[np.complexfloating[Any, Any]])
377
+
378
+ assert_type(c8 + f16, np.complexfloating[_32Bit | _128Bit, _32Bit | _128Bit])
379
+ assert_type(c8 + c16, np.complexfloating[_32Bit | _64Bit, _32Bit | _64Bit])
380
+ assert_type(c8 + f8, np.complexfloating[_32Bit | _64Bit, _32Bit | _64Bit])
381
+ assert_type(c8 + i8, np.complexfloating[_32Bit | _64Bit, _32Bit | _64Bit])
382
+ assert_type(c8 + c8, np.complex64)
383
+ assert_type(c8 + f4, np.complex64)
384
+ assert_type(c8 + i4, np.complex64)
385
+ assert_type(c8 + b_, np.complex64)
386
+ assert_type(c8 + b, np.complex64)
387
+ assert_type(c8 + c, np.complexfloating[_32Bit | _64Bit, _32Bit | _64Bit])
388
+ assert_type(c8 + f, np.complexfloating[_32Bit | _64Bit, _32Bit | _64Bit])
389
+ assert_type(c8 + AR_f, npt.NDArray[np.complexfloating[Any, Any]])
390
+
391
+ assert_type(f16 + c8, np.complexfloating[_32Bit | _128Bit, _32Bit | _128Bit])
392
+ assert_type(c16 + c8, np.complexfloating[_32Bit | _64Bit, _32Bit | _64Bit])
393
+ assert_type(f8 + c8, np.complexfloating[_32Bit | _64Bit, _32Bit | _64Bit])
394
+ assert_type(i8 + c8, np.complexfloating[_32Bit | _64Bit, _32Bit | _64Bit])
395
+ assert_type(c8 + c8, np.complex64)
396
+ assert_type(f4 + c8, np.complex64)
397
+ assert_type(i4 + c8, np.complex64)
398
+ assert_type(b_ + c8, np.complex64)
399
+ assert_type(b + c8, np.complex64)
400
+ assert_type(c + c8, np.complexfloating[_32Bit | _64Bit, _32Bit | _64Bit])
401
+ assert_type(f + c8, np.complexfloating[_32Bit | _64Bit, _32Bit | _64Bit])
402
+ assert_type(AR_f + c8, npt.NDArray[np.complexfloating[Any, Any]])
403
+
404
+ # Float
405
+
406
+ assert_type(f8 + f16, np.floating[_64Bit | _128Bit])
407
+ assert_type(f8 + f8, np.float64)
408
+ assert_type(f8 + i8, np.float64)
409
+ assert_type(f8 + f4, np.floating[_32Bit | _64Bit])
410
+ assert_type(f8 + i4, np.floating[_32Bit | _64Bit])
411
+ assert_type(f8 + b_, np.float64)
412
+ assert_type(f8 + b, np.float64)
413
+ assert_type(f8 + c, np.complex128)
414
+ assert_type(f8 + f, np.float64)
415
+ assert_type(f8 + AR_f, npt.NDArray[np.floating[Any]])
416
+
417
+ assert_type(f16 + f8, np.floating[_64Bit | _128Bit])
418
+ assert_type(f8 + f8, np.float64)
419
+ assert_type(i8 + f8, np.float64)
420
+ assert_type(f4 + f8, np.floating[_32Bit | _64Bit])
421
+ assert_type(i4 + f8, np.floating[_32Bit | _64Bit])
422
+ assert_type(b_ + f8, np.float64)
423
+ assert_type(b + f8, np.float64)
424
+ assert_type(c + f8, np.complex128)
425
+ assert_type(f + f8, np.float64)
426
+ assert_type(AR_f + f8, npt.NDArray[np.floating[Any]])
427
+
428
+ assert_type(f4 + f16, np.floating[_32Bit | _128Bit])
429
+ assert_type(f4 + f8, np.floating[_32Bit | _64Bit])
430
+ assert_type(f4 + i8, np.floating[_32Bit | _64Bit])
431
+ assert_type(f4 + f4, np.float32)
432
+ assert_type(f4 + i4, np.float32)
433
+ assert_type(f4 + b_, np.float32)
434
+ assert_type(f4 + b, np.float32)
435
+ assert_type(f4 + c, np.complexfloating[_32Bit | _64Bit, _32Bit | _64Bit])
436
+ assert_type(f4 + f, np.floating[_32Bit | _64Bit])
437
+ assert_type(f4 + AR_f, npt.NDArray[np.floating[Any]])
438
+
439
+ assert_type(f16 + f4, np.floating[_32Bit | _128Bit])
440
+ assert_type(f8 + f4, np.floating[_32Bit | _64Bit])
441
+ assert_type(i8 + f4, np.floating[_32Bit | _64Bit])
442
+ assert_type(f4 + f4, np.float32)
443
+ assert_type(i4 + f4, np.float32)
444
+ assert_type(b_ + f4, np.float32)
445
+ assert_type(b + f4, np.float32)
446
+ assert_type(c + f4, np.complexfloating[_32Bit | _64Bit, _32Bit | _64Bit])
447
+ assert_type(f + f4, np.floating[_32Bit | _64Bit])
448
+ assert_type(AR_f + f4, npt.NDArray[np.floating[Any]])
449
+
450
+ # Int
451
+
452
+ assert_type(i8 + i8, np.int64)
453
+ assert_type(i8 + u8, Any)
454
+ assert_type(i8 + i4, np.signedinteger[_32Bit | _64Bit])
455
+ assert_type(i8 + u4, Any)
456
+ assert_type(i8 + b_, np.int64)
457
+ assert_type(i8 + b, np.int64)
458
+ assert_type(i8 + c, np.complex128)
459
+ assert_type(i8 + f, np.float64)
460
+ assert_type(i8 + AR_f, npt.NDArray[np.floating[Any]])
461
+
462
+ assert_type(u8 + u8, np.uint64)
463
+ assert_type(u8 + i4, Any)
464
+ assert_type(u8 + u4, np.unsignedinteger[_32Bit | _64Bit])
465
+ assert_type(u8 + b_, np.uint64)
466
+ assert_type(u8 + b, np.uint64)
467
+ assert_type(u8 + c, np.complex128)
468
+ assert_type(u8 + f, np.float64)
469
+ assert_type(u8 + AR_f, npt.NDArray[np.floating[Any]])
470
+
471
+ assert_type(i8 + i8, np.int64)
472
+ assert_type(u8 + i8, Any)
473
+ assert_type(i4 + i8, np.signedinteger[_32Bit | _64Bit])
474
+ assert_type(u4 + i8, Any)
475
+ assert_type(b_ + i8, np.int64)
476
+ assert_type(b + i8, np.int64)
477
+ assert_type(c + i8, np.complex128)
478
+ assert_type(f + i8, np.float64)
479
+ assert_type(AR_f + i8, npt.NDArray[np.floating[Any]])
480
+
481
+ assert_type(u8 + u8, np.uint64)
482
+ assert_type(i4 + u8, Any)
483
+ assert_type(u4 + u8, np.unsignedinteger[_32Bit | _64Bit])
484
+ assert_type(b_ + u8, np.uint64)
485
+ assert_type(b + u8, np.uint64)
486
+ assert_type(c + u8, np.complex128)
487
+ assert_type(f + u8, np.float64)
488
+ assert_type(AR_f + u8, npt.NDArray[np.floating[Any]])
489
+
490
+ assert_type(i4 + i8, np.signedinteger[_32Bit | _64Bit])
491
+ assert_type(i4 + i4, np.int32)
492
+ assert_type(i4 + b_, np.int32)
493
+ assert_type(i4 + b, np.int32)
494
+ assert_type(i4 + AR_f, npt.NDArray[np.floating[Any]])
495
+
496
+ assert_type(u4 + i8, Any)
497
+ assert_type(u4 + i4, Any)
498
+ assert_type(u4 + u8, np.unsignedinteger[_32Bit | _64Bit])
499
+ assert_type(u4 + u4, np.uint32)
500
+ assert_type(u4 + b_, np.uint32)
501
+ assert_type(u4 + b, np.uint32)
502
+ assert_type(u4 + AR_f, npt.NDArray[np.floating[Any]])
503
+
504
+ assert_type(i8 + i4, np.signedinteger[_32Bit | _64Bit])
505
+ assert_type(i4 + i4, np.int32)
506
+ assert_type(b_ + i4, np.int32)
507
+ assert_type(b + i4, np.int32)
508
+ assert_type(AR_f + i4, npt.NDArray[np.floating[Any]])
509
+
510
+ assert_type(i8 + u4, Any)
511
+ assert_type(i4 + u4, Any)
512
+ assert_type(u8 + u4, np.unsignedinteger[_32Bit | _64Bit])
513
+ assert_type(u4 + u4, np.uint32)
514
+ assert_type(b_ + u4, np.uint32)
515
+ assert_type(b + u4, np.uint32)
516
+ assert_type(AR_f + u4, npt.NDArray[np.floating[Any]])
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/reveal/array_constructors.pyi ADDED
@@ -0,0 +1,221 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import sys
2
+ from typing import Any, TypeVar
3
+ from pathlib import Path
4
+ from collections import deque
5
+
6
+ import numpy as np
7
+ import numpy.typing as npt
8
+
9
+ if sys.version_info >= (3, 11):
10
+ from typing import assert_type
11
+ else:
12
+ from typing_extensions import assert_type
13
+
14
+ _SCT = TypeVar("_SCT", bound=np.generic, covariant=True)
15
+
16
+ class SubClass(np.ndarray[Any, np.dtype[_SCT]]): ...
17
+
18
+ i8: np.int64
19
+
20
+ A: npt.NDArray[np.float64]
21
+ B: SubClass[np.float64]
22
+ C: list[int]
23
+
24
+ def func(i: int, j: int, **kwargs: Any) -> SubClass[np.float64]: ...
25
+
26
+ assert_type(np.empty_like(A), npt.NDArray[np.float64])
27
+ assert_type(np.empty_like(B), SubClass[np.float64])
28
+ assert_type(np.empty_like([1, 1.0]), npt.NDArray[Any])
29
+ assert_type(np.empty_like(A, dtype=np.int64), npt.NDArray[np.int64])
30
+ assert_type(np.empty_like(A, dtype='c16'), npt.NDArray[Any])
31
+
32
+ assert_type(np.array(A), npt.NDArray[np.float64])
33
+ assert_type(np.array(B), npt.NDArray[np.float64])
34
+ assert_type(np.array(B, subok=True), SubClass[np.float64])
35
+ assert_type(np.array([1, 1.0]), npt.NDArray[Any])
36
+ assert_type(np.array(deque([1, 2, 3])), npt.NDArray[Any])
37
+ assert_type(np.array(A, dtype=np.int64), npt.NDArray[np.int64])
38
+ assert_type(np.array(A, dtype='c16'), npt.NDArray[Any])
39
+ assert_type(np.array(A, like=A), npt.NDArray[np.float64])
40
+
41
+ assert_type(np.zeros([1, 5, 6]), npt.NDArray[np.float64])
42
+ assert_type(np.zeros([1, 5, 6], dtype=np.int64), npt.NDArray[np.int64])
43
+ assert_type(np.zeros([1, 5, 6], dtype='c16'), npt.NDArray[Any])
44
+
45
+ assert_type(np.empty([1, 5, 6]), npt.NDArray[np.float64])
46
+ assert_type(np.empty([1, 5, 6], dtype=np.int64), npt.NDArray[np.int64])
47
+ assert_type(np.empty([1, 5, 6], dtype='c16'), npt.NDArray[Any])
48
+
49
+ assert_type(np.concatenate(A), npt.NDArray[np.float64])
50
+ assert_type(np.concatenate([A, A]), Any)
51
+ assert_type(np.concatenate([[1], A]), npt.NDArray[Any])
52
+ assert_type(np.concatenate([[1], [1]]), npt.NDArray[Any])
53
+ assert_type(np.concatenate((A, A)), npt.NDArray[np.float64])
54
+ assert_type(np.concatenate(([1], [1])), npt.NDArray[Any])
55
+ assert_type(np.concatenate([1, 1.0]), npt.NDArray[Any])
56
+ assert_type(np.concatenate(A, dtype=np.int64), npt.NDArray[np.int64])
57
+ assert_type(np.concatenate(A, dtype='c16'), npt.NDArray[Any])
58
+ assert_type(np.concatenate([1, 1.0], out=A), npt.NDArray[np.float64])
59
+
60
+ assert_type(np.asarray(A), npt.NDArray[np.float64])
61
+ assert_type(np.asarray(B), npt.NDArray[np.float64])
62
+ assert_type(np.asarray([1, 1.0]), npt.NDArray[Any])
63
+ assert_type(np.asarray(A, dtype=np.int64), npt.NDArray[np.int64])
64
+ assert_type(np.asarray(A, dtype='c16'), npt.NDArray[Any])
65
+
66
+ assert_type(np.asanyarray(A), npt.NDArray[np.float64])
67
+ assert_type(np.asanyarray(B), SubClass[np.float64])
68
+ assert_type(np.asanyarray([1, 1.0]), npt.NDArray[Any])
69
+ assert_type(np.asanyarray(A, dtype=np.int64), npt.NDArray[np.int64])
70
+ assert_type(np.asanyarray(A, dtype='c16'), npt.NDArray[Any])
71
+
72
+ assert_type(np.ascontiguousarray(A), npt.NDArray[np.float64])
73
+ assert_type(np.ascontiguousarray(B), npt.NDArray[np.float64])
74
+ assert_type(np.ascontiguousarray([1, 1.0]), npt.NDArray[Any])
75
+ assert_type(np.ascontiguousarray(A, dtype=np.int64), npt.NDArray[np.int64])
76
+ assert_type(np.ascontiguousarray(A, dtype='c16'), npt.NDArray[Any])
77
+
78
+ assert_type(np.asfortranarray(A), npt.NDArray[np.float64])
79
+ assert_type(np.asfortranarray(B), npt.NDArray[np.float64])
80
+ assert_type(np.asfortranarray([1, 1.0]), npt.NDArray[Any])
81
+ assert_type(np.asfortranarray(A, dtype=np.int64), npt.NDArray[np.int64])
82
+ assert_type(np.asfortranarray(A, dtype='c16'), npt.NDArray[Any])
83
+
84
+ assert_type(np.fromstring("1 1 1", sep=" "), npt.NDArray[np.float64])
85
+ assert_type(np.fromstring(b"1 1 1", sep=" "), npt.NDArray[np.float64])
86
+ assert_type(np.fromstring("1 1 1", dtype=np.int64, sep=" "), npt.NDArray[np.int64])
87
+ assert_type(np.fromstring(b"1 1 1", dtype=np.int64, sep=" "), npt.NDArray[np.int64])
88
+ assert_type(np.fromstring("1 1 1", dtype="c16", sep=" "), npt.NDArray[Any])
89
+ assert_type(np.fromstring(b"1 1 1", dtype="c16", sep=" "), npt.NDArray[Any])
90
+
91
+ assert_type(np.fromfile("test.txt", sep=" "), npt.NDArray[np.float64])
92
+ assert_type(np.fromfile("test.txt", dtype=np.int64, sep=" "), npt.NDArray[np.int64])
93
+ assert_type(np.fromfile("test.txt", dtype="c16", sep=" "), npt.NDArray[Any])
94
+ with open("test.txt") as f:
95
+ assert_type(np.fromfile(f, sep=" "), npt.NDArray[np.float64])
96
+ assert_type(np.fromfile(b"test.txt", sep=" "), npt.NDArray[np.float64])
97
+ assert_type(np.fromfile(Path("test.txt"), sep=" "), npt.NDArray[np.float64])
98
+
99
+ assert_type(np.fromiter("12345", np.float64), npt.NDArray[np.float64])
100
+ assert_type(np.fromiter("12345", float), npt.NDArray[Any])
101
+
102
+ assert_type(np.frombuffer(A), npt.NDArray[np.float64])
103
+ assert_type(np.frombuffer(A, dtype=np.int64), npt.NDArray[np.int64])
104
+ assert_type(np.frombuffer(A, dtype="c16"), npt.NDArray[Any])
105
+
106
+ assert_type(np.arange(False, True), npt.NDArray[np.signedinteger[Any]])
107
+ assert_type(np.arange(10), npt.NDArray[np.signedinteger[Any]])
108
+ assert_type(np.arange(0, 10, step=2), npt.NDArray[np.signedinteger[Any]])
109
+ assert_type(np.arange(10.0), npt.NDArray[np.floating[Any]])
110
+ assert_type(np.arange(start=0, stop=10.0), npt.NDArray[np.floating[Any]])
111
+ assert_type(np.arange(np.timedelta64(0)), npt.NDArray[np.timedelta64])
112
+ assert_type(np.arange(0, np.timedelta64(10)), npt.NDArray[np.timedelta64])
113
+ assert_type(np.arange(np.datetime64("0"), np.datetime64("10")), npt.NDArray[np.datetime64])
114
+ assert_type(np.arange(10, dtype=np.float64), npt.NDArray[np.float64])
115
+ assert_type(np.arange(0, 10, step=2, dtype=np.int16), npt.NDArray[np.int16])
116
+ assert_type(np.arange(10, dtype=int), npt.NDArray[Any])
117
+ assert_type(np.arange(0, 10, dtype="f8"), npt.NDArray[Any])
118
+
119
+ assert_type(np.require(A), npt.NDArray[np.float64])
120
+ assert_type(np.require(B), SubClass[np.float64])
121
+ assert_type(np.require(B, requirements=None), SubClass[np.float64])
122
+ assert_type(np.require(B, dtype=int), np.ndarray[Any, Any])
123
+ assert_type(np.require(B, requirements="E"), np.ndarray[Any, Any])
124
+ assert_type(np.require(B, requirements=["ENSUREARRAY"]), np.ndarray[Any, Any])
125
+ assert_type(np.require(B, requirements={"F", "E"}), np.ndarray[Any, Any])
126
+ assert_type(np.require(B, requirements=["C", "OWNDATA"]), SubClass[np.float64])
127
+ assert_type(np.require(B, requirements="W"), SubClass[np.float64])
128
+ assert_type(np.require(B, requirements="A"), SubClass[np.float64])
129
+ assert_type(np.require(C), np.ndarray[Any, Any])
130
+
131
+ assert_type(np.linspace(0, 10), npt.NDArray[np.floating[Any]])
132
+ assert_type(np.linspace(0, 10j), npt.NDArray[np.complexfloating[Any, Any]])
133
+ assert_type(np.linspace(0, 10, dtype=np.int64), npt.NDArray[np.int64])
134
+ assert_type(np.linspace(0, 10, dtype=int), npt.NDArray[Any])
135
+ assert_type(np.linspace(0, 10, retstep=True), tuple[npt.NDArray[np.floating[Any]], np.floating[Any]])
136
+ assert_type(np.linspace(0j, 10, retstep=True), tuple[npt.NDArray[np.complexfloating[Any, Any]], np.complexfloating[Any, Any]])
137
+ assert_type(np.linspace(0, 10, retstep=True, dtype=np.int64), tuple[npt.NDArray[np.int64], np.int64])
138
+ assert_type(np.linspace(0j, 10, retstep=True, dtype=int), tuple[npt.NDArray[Any], Any])
139
+
140
+ assert_type(np.logspace(0, 10), npt.NDArray[np.floating[Any]])
141
+ assert_type(np.logspace(0, 10j), npt.NDArray[np.complexfloating[Any, Any]])
142
+ assert_type(np.logspace(0, 10, dtype=np.int64), npt.NDArray[np.int64])
143
+ assert_type(np.logspace(0, 10, dtype=int), npt.NDArray[Any])
144
+
145
+ assert_type(np.geomspace(0, 10), npt.NDArray[np.floating[Any]])
146
+ assert_type(np.geomspace(0, 10j), npt.NDArray[np.complexfloating[Any, Any]])
147
+ assert_type(np.geomspace(0, 10, dtype=np.int64), npt.NDArray[np.int64])
148
+ assert_type(np.geomspace(0, 10, dtype=int), npt.NDArray[Any])
149
+
150
+ assert_type(np.zeros_like(A), npt.NDArray[np.float64])
151
+ assert_type(np.zeros_like(C), npt.NDArray[Any])
152
+ assert_type(np.zeros_like(A, dtype=float), npt.NDArray[Any])
153
+ assert_type(np.zeros_like(B), SubClass[np.float64])
154
+ assert_type(np.zeros_like(B, dtype=np.int64), npt.NDArray[np.int64])
155
+
156
+ assert_type(np.ones_like(A), npt.NDArray[np.float64])
157
+ assert_type(np.ones_like(C), npt.NDArray[Any])
158
+ assert_type(np.ones_like(A, dtype=float), npt.NDArray[Any])
159
+ assert_type(np.ones_like(B), SubClass[np.float64])
160
+ assert_type(np.ones_like(B, dtype=np.int64), npt.NDArray[np.int64])
161
+
162
+ assert_type(np.full_like(A, i8), npt.NDArray[np.float64])
163
+ assert_type(np.full_like(C, i8), npt.NDArray[Any])
164
+ assert_type(np.full_like(A, i8, dtype=int), npt.NDArray[Any])
165
+ assert_type(np.full_like(B, i8), SubClass[np.float64])
166
+ assert_type(np.full_like(B, i8, dtype=np.int64), npt.NDArray[np.int64])
167
+
168
+ assert_type(np.ones(1), npt.NDArray[np.float64])
169
+ assert_type(np.ones([1, 1, 1]), npt.NDArray[np.float64])
170
+ assert_type(np.ones(5, dtype=np.int64), npt.NDArray[np.int64])
171
+ assert_type(np.ones(5, dtype=int), npt.NDArray[Any])
172
+
173
+ assert_type(np.full(1, i8), npt.NDArray[Any])
174
+ assert_type(np.full([1, 1, 1], i8), npt.NDArray[Any])
175
+ assert_type(np.full(1, i8, dtype=np.float64), npt.NDArray[np.float64])
176
+ assert_type(np.full(1, i8, dtype=float), npt.NDArray[Any])
177
+
178
+ assert_type(np.indices([1, 2, 3]), npt.NDArray[np.int_])
179
+ assert_type(np.indices([1, 2, 3], sparse=True), tuple[npt.NDArray[np.int_], ...])
180
+
181
+ assert_type(np.fromfunction(func, (3, 5)), SubClass[np.float64])
182
+
183
+ assert_type(np.identity(10), npt.NDArray[np.float64])
184
+ assert_type(np.identity(10, dtype=np.int64), npt.NDArray[np.int64])
185
+ assert_type(np.identity(10, dtype=int), npt.NDArray[Any])
186
+
187
+ assert_type(np.atleast_1d(A), npt.NDArray[np.float64])
188
+ assert_type(np.atleast_1d(C), npt.NDArray[Any])
189
+ assert_type(np.atleast_1d(A, A), list[npt.NDArray[Any]])
190
+ assert_type(np.atleast_1d(A, C), list[npt.NDArray[Any]])
191
+ assert_type(np.atleast_1d(C, C), list[npt.NDArray[Any]])
192
+
193
+ assert_type(np.atleast_2d(A), npt.NDArray[np.float64])
194
+
195
+ assert_type(np.atleast_3d(A), npt.NDArray[np.float64])
196
+
197
+ assert_type(np.vstack([A, A]), np.ndarray[Any, Any])
198
+ assert_type(np.vstack([A, A], dtype=np.float64), npt.NDArray[np.float64])
199
+ assert_type(np.vstack([A, C]), npt.NDArray[Any])
200
+ assert_type(np.vstack([C, C]), npt.NDArray[Any])
201
+
202
+ assert_type(np.hstack([A, A]), np.ndarray[Any, Any])
203
+ assert_type(np.hstack([A, A], dtype=np.float64), npt.NDArray[np.float64])
204
+
205
+ assert_type(np.stack([A, A]), Any)
206
+ assert_type(np.stack([A, A], dtype=np.float64), npt.NDArray[np.float64])
207
+ assert_type(np.stack([A, C]), npt.NDArray[Any])
208
+ assert_type(np.stack([C, C]), npt.NDArray[Any])
209
+ assert_type(np.stack([A, A], axis=0), Any)
210
+ assert_type(np.stack([A, A], out=B), SubClass[np.float64])
211
+
212
+ assert_type(np.block([[A, A], [A, A]]), npt.NDArray[Any])
213
+ assert_type(np.block(C), npt.NDArray[Any])
214
+
215
+ if sys.version_info >= (3, 12):
216
+ from collections.abc import Buffer
217
+
218
+ def create_array(obj: npt.ArrayLike) -> npt.NDArray[Any]: ...
219
+
220
+ buffer: Buffer
221
+ assert_type(create_array(buffer), npt.NDArray[Any])
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/reveal/datasource.pyi ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import sys
2
+ from pathlib import Path
3
+ from typing import IO, Any
4
+
5
+ import numpy as np
6
+
7
+ if sys.version_info >= (3, 11):
8
+ from typing import assert_type
9
+ else:
10
+ from typing_extensions import assert_type
11
+
12
+ path1: Path
13
+ path2: str
14
+
15
+ d1 = np.DataSource(path1)
16
+ d2 = np.DataSource(path2)
17
+ d3 = np.DataSource(None)
18
+
19
+ assert_type(d1.abspath("..."), str)
20
+ assert_type(d2.abspath("..."), str)
21
+ assert_type(d3.abspath("..."), str)
22
+
23
+ assert_type(d1.exists("..."), bool)
24
+ assert_type(d2.exists("..."), bool)
25
+ assert_type(d3.exists("..."), bool)
26
+
27
+ assert_type(d1.open("...", "r"), IO[Any])
28
+ assert_type(d2.open("...", encoding="utf8"), IO[Any])
29
+ assert_type(d3.open("...", newline="/n"), IO[Any])
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/reveal/fromnumeric.pyi ADDED
@@ -0,0 +1,305 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """Tests for :mod:`core.fromnumeric`."""
2
+
3
+ import sys
4
+ from typing import Any
5
+
6
+ import numpy as np
7
+ import numpy.typing as npt
8
+
9
+ if sys.version_info >= (3, 11):
10
+ from typing import assert_type
11
+ else:
12
+ from typing_extensions import assert_type
13
+
14
+ class NDArraySubclass(npt.NDArray[np.complex128]):
15
+ ...
16
+
17
+ AR_b: npt.NDArray[np.bool_]
18
+ AR_f4: npt.NDArray[np.float32]
19
+ AR_c16: npt.NDArray[np.complex128]
20
+ AR_u8: npt.NDArray[np.uint64]
21
+ AR_i8: npt.NDArray[np.int64]
22
+ AR_O: npt.NDArray[np.object_]
23
+ AR_subclass: NDArraySubclass
24
+
25
+ b: np.bool_
26
+ f4: np.float32
27
+ i8: np.int64
28
+ f: float
29
+
30
+ assert_type(np.take(b, 0), np.bool_)
31
+ assert_type(np.take(f4, 0), np.float32)
32
+ assert_type(np.take(f, 0), Any)
33
+ assert_type(np.take(AR_b, 0), np.bool_)
34
+ assert_type(np.take(AR_f4, 0), np.float32)
35
+ assert_type(np.take(AR_b, [0]), npt.NDArray[np.bool_])
36
+ assert_type(np.take(AR_f4, [0]), npt.NDArray[np.float32])
37
+ assert_type(np.take([1], [0]), npt.NDArray[Any])
38
+ assert_type(np.take(AR_f4, [0], out=AR_subclass), NDArraySubclass)
39
+
40
+ assert_type(np.reshape(b, 1), npt.NDArray[np.bool_])
41
+ assert_type(np.reshape(f4, 1), npt.NDArray[np.float32])
42
+ assert_type(np.reshape(f, 1), npt.NDArray[Any])
43
+ assert_type(np.reshape(AR_b, 1), npt.NDArray[np.bool_])
44
+ assert_type(np.reshape(AR_f4, 1), npt.NDArray[np.float32])
45
+
46
+ assert_type(np.choose(1, [True, True]), Any)
47
+ assert_type(np.choose([1], [True, True]), npt.NDArray[Any])
48
+ assert_type(np.choose([1], AR_b), npt.NDArray[np.bool_])
49
+ assert_type(np.choose([1], AR_b, out=AR_f4), npt.NDArray[np.float32])
50
+
51
+ assert_type(np.repeat(b, 1), npt.NDArray[np.bool_])
52
+ assert_type(np.repeat(f4, 1), npt.NDArray[np.float32])
53
+ assert_type(np.repeat(f, 1), npt.NDArray[Any])
54
+ assert_type(np.repeat(AR_b, 1), npt.NDArray[np.bool_])
55
+ assert_type(np.repeat(AR_f4, 1), npt.NDArray[np.float32])
56
+
57
+ # TODO: array_bdd tests for np.put()
58
+
59
+ assert_type(np.swapaxes([[0, 1]], 0, 0), npt.NDArray[Any])
60
+ assert_type(np.swapaxes(AR_b, 0, 0), npt.NDArray[np.bool_])
61
+ assert_type(np.swapaxes(AR_f4, 0, 0), npt.NDArray[np.float32])
62
+
63
+ assert_type(np.transpose(b), npt.NDArray[np.bool_])
64
+ assert_type(np.transpose(f4), npt.NDArray[np.float32])
65
+ assert_type(np.transpose(f), npt.NDArray[Any])
66
+ assert_type(np.transpose(AR_b), npt.NDArray[np.bool_])
67
+ assert_type(np.transpose(AR_f4), npt.NDArray[np.float32])
68
+
69
+ assert_type(np.partition(b, 0, axis=None), npt.NDArray[np.bool_])
70
+ assert_type(np.partition(f4, 0, axis=None), npt.NDArray[np.float32])
71
+ assert_type(np.partition(f, 0, axis=None), npt.NDArray[Any])
72
+ assert_type(np.partition(AR_b, 0), npt.NDArray[np.bool_])
73
+ assert_type(np.partition(AR_f4, 0), npt.NDArray[np.float32])
74
+
75
+ assert_type(np.argpartition(b, 0), npt.NDArray[np.intp])
76
+ assert_type(np.argpartition(f4, 0), npt.NDArray[np.intp])
77
+ assert_type(np.argpartition(f, 0), npt.NDArray[np.intp])
78
+ assert_type(np.argpartition(AR_b, 0), npt.NDArray[np.intp])
79
+ assert_type(np.argpartition(AR_f4, 0), npt.NDArray[np.intp])
80
+
81
+ assert_type(np.sort([2, 1], 0), npt.NDArray[Any])
82
+ assert_type(np.sort(AR_b, 0), npt.NDArray[np.bool_])
83
+ assert_type(np.sort(AR_f4, 0), npt.NDArray[np.float32])
84
+
85
+ assert_type(np.argsort(AR_b, 0), npt.NDArray[np.intp])
86
+ assert_type(np.argsort(AR_f4, 0), npt.NDArray[np.intp])
87
+
88
+ assert_type(np.argmax(AR_b), np.intp)
89
+ assert_type(np.argmax(AR_f4), np.intp)
90
+ assert_type(np.argmax(AR_b, axis=0), Any)
91
+ assert_type(np.argmax(AR_f4, axis=0), Any)
92
+ assert_type(np.argmax(AR_f4, out=AR_subclass), NDArraySubclass)
93
+
94
+ assert_type(np.argmin(AR_b), np.intp)
95
+ assert_type(np.argmin(AR_f4), np.intp)
96
+ assert_type(np.argmin(AR_b, axis=0), Any)
97
+ assert_type(np.argmin(AR_f4, axis=0), Any)
98
+ assert_type(np.argmin(AR_f4, out=AR_subclass), NDArraySubclass)
99
+
100
+ assert_type(np.searchsorted(AR_b[0], 0), np.intp)
101
+ assert_type(np.searchsorted(AR_f4[0], 0), np.intp)
102
+ assert_type(np.searchsorted(AR_b[0], [0]), npt.NDArray[np.intp])
103
+ assert_type(np.searchsorted(AR_f4[0], [0]), npt.NDArray[np.intp])
104
+
105
+ assert_type(np.resize(b, (5, 5)), npt.NDArray[np.bool_])
106
+ assert_type(np.resize(f4, (5, 5)), npt.NDArray[np.float32])
107
+ assert_type(np.resize(f, (5, 5)), npt.NDArray[Any])
108
+ assert_type(np.resize(AR_b, (5, 5)), npt.NDArray[np.bool_])
109
+ assert_type(np.resize(AR_f4, (5, 5)), npt.NDArray[np.float32])
110
+
111
+ assert_type(np.squeeze(b), np.bool_)
112
+ assert_type(np.squeeze(f4), np.float32)
113
+ assert_type(np.squeeze(f), npt.NDArray[Any])
114
+ assert_type(np.squeeze(AR_b), npt.NDArray[np.bool_])
115
+ assert_type(np.squeeze(AR_f4), npt.NDArray[np.float32])
116
+
117
+ assert_type(np.diagonal(AR_b), npt.NDArray[np.bool_])
118
+ assert_type(np.diagonal(AR_f4), npt.NDArray[np.float32])
119
+
120
+ assert_type(np.trace(AR_b), Any)
121
+ assert_type(np.trace(AR_f4), Any)
122
+ assert_type(np.trace(AR_f4, out=AR_subclass), NDArraySubclass)
123
+
124
+ assert_type(np.ravel(b), npt.NDArray[np.bool_])
125
+ assert_type(np.ravel(f4), npt.NDArray[np.float32])
126
+ assert_type(np.ravel(f), npt.NDArray[Any])
127
+ assert_type(np.ravel(AR_b), npt.NDArray[np.bool_])
128
+ assert_type(np.ravel(AR_f4), npt.NDArray[np.float32])
129
+
130
+ assert_type(np.nonzero(b), tuple[npt.NDArray[np.intp], ...])
131
+ assert_type(np.nonzero(f4), tuple[npt.NDArray[np.intp], ...])
132
+ assert_type(np.nonzero(f), tuple[npt.NDArray[np.intp], ...])
133
+ assert_type(np.nonzero(AR_b), tuple[npt.NDArray[np.intp], ...])
134
+ assert_type(np.nonzero(AR_f4), tuple[npt.NDArray[np.intp], ...])
135
+
136
+ assert_type(np.shape(b), tuple[int, ...])
137
+ assert_type(np.shape(f4), tuple[int, ...])
138
+ assert_type(np.shape(f), tuple[int, ...])
139
+ assert_type(np.shape(AR_b), tuple[int, ...])
140
+ assert_type(np.shape(AR_f4), tuple[int, ...])
141
+
142
+ assert_type(np.compress([True], b), npt.NDArray[np.bool_])
143
+ assert_type(np.compress([True], f4), npt.NDArray[np.float32])
144
+ assert_type(np.compress([True], f), npt.NDArray[Any])
145
+ assert_type(np.compress([True], AR_b), npt.NDArray[np.bool_])
146
+ assert_type(np.compress([True], AR_f4), npt.NDArray[np.float32])
147
+
148
+ assert_type(np.clip(b, 0, 1.0), np.bool_)
149
+ assert_type(np.clip(f4, -1, 1), np.float32)
150
+ assert_type(np.clip(f, 0, 1), Any)
151
+ assert_type(np.clip(AR_b, 0, 1), npt.NDArray[np.bool_])
152
+ assert_type(np.clip(AR_f4, 0, 1), npt.NDArray[np.float32])
153
+ assert_type(np.clip([0], 0, 1), npt.NDArray[Any])
154
+ assert_type(np.clip(AR_b, 0, 1, out=AR_subclass), NDArraySubclass)
155
+
156
+ assert_type(np.sum(b), np.bool_)
157
+ assert_type(np.sum(f4), np.float32)
158
+ assert_type(np.sum(f), Any)
159
+ assert_type(np.sum(AR_b), np.bool_)
160
+ assert_type(np.sum(AR_f4), np.float32)
161
+ assert_type(np.sum(AR_b, axis=0), Any)
162
+ assert_type(np.sum(AR_f4, axis=0), Any)
163
+ assert_type(np.sum(AR_f4, out=AR_subclass), NDArraySubclass)
164
+
165
+ assert_type(np.all(b), np.bool_)
166
+ assert_type(np.all(f4), np.bool_)
167
+ assert_type(np.all(f), np.bool_)
168
+ assert_type(np.all(AR_b), np.bool_)
169
+ assert_type(np.all(AR_f4), np.bool_)
170
+ assert_type(np.all(AR_b, axis=0), Any)
171
+ assert_type(np.all(AR_f4, axis=0), Any)
172
+ assert_type(np.all(AR_b, keepdims=True), Any)
173
+ assert_type(np.all(AR_f4, keepdims=True), Any)
174
+ assert_type(np.all(AR_f4, out=AR_subclass), NDArraySubclass)
175
+
176
+ assert_type(np.any(b), np.bool_)
177
+ assert_type(np.any(f4), np.bool_)
178
+ assert_type(np.any(f), np.bool_)
179
+ assert_type(np.any(AR_b), np.bool_)
180
+ assert_type(np.any(AR_f4), np.bool_)
181
+ assert_type(np.any(AR_b, axis=0), Any)
182
+ assert_type(np.any(AR_f4, axis=0), Any)
183
+ assert_type(np.any(AR_b, keepdims=True), Any)
184
+ assert_type(np.any(AR_f4, keepdims=True), Any)
185
+ assert_type(np.any(AR_f4, out=AR_subclass), NDArraySubclass)
186
+
187
+ assert_type(np.cumsum(b), npt.NDArray[np.bool_])
188
+ assert_type(np.cumsum(f4), npt.NDArray[np.float32])
189
+ assert_type(np.cumsum(f), npt.NDArray[Any])
190
+ assert_type(np.cumsum(AR_b), npt.NDArray[np.bool_])
191
+ assert_type(np.cumsum(AR_f4), npt.NDArray[np.float32])
192
+ assert_type(np.cumsum(f, dtype=float), npt.NDArray[Any])
193
+ assert_type(np.cumsum(f, dtype=np.float64), npt.NDArray[np.float64])
194
+ assert_type(np.cumsum(AR_f4, out=AR_subclass), NDArraySubclass)
195
+
196
+ assert_type(np.ptp(b), np.bool_)
197
+ assert_type(np.ptp(f4), np.float32)
198
+ assert_type(np.ptp(f), Any)
199
+ assert_type(np.ptp(AR_b), np.bool_)
200
+ assert_type(np.ptp(AR_f4), np.float32)
201
+ assert_type(np.ptp(AR_b, axis=0), Any)
202
+ assert_type(np.ptp(AR_f4, axis=0), Any)
203
+ assert_type(np.ptp(AR_b, keepdims=True), Any)
204
+ assert_type(np.ptp(AR_f4, keepdims=True), Any)
205
+ assert_type(np.ptp(AR_f4, out=AR_subclass), NDArraySubclass)
206
+
207
+ assert_type(np.amax(b), np.bool_)
208
+ assert_type(np.amax(f4), np.float32)
209
+ assert_type(np.amax(f), Any)
210
+ assert_type(np.amax(AR_b), np.bool_)
211
+ assert_type(np.amax(AR_f4), np.float32)
212
+ assert_type(np.amax(AR_b, axis=0), Any)
213
+ assert_type(np.amax(AR_f4, axis=0), Any)
214
+ assert_type(np.amax(AR_b, keepdims=True), Any)
215
+ assert_type(np.amax(AR_f4, keepdims=True), Any)
216
+ assert_type(np.amax(AR_f4, out=AR_subclass), NDArraySubclass)
217
+
218
+ assert_type(np.amin(b), np.bool_)
219
+ assert_type(np.amin(f4), np.float32)
220
+ assert_type(np.amin(f), Any)
221
+ assert_type(np.amin(AR_b), np.bool_)
222
+ assert_type(np.amin(AR_f4), np.float32)
223
+ assert_type(np.amin(AR_b, axis=0), Any)
224
+ assert_type(np.amin(AR_f4, axis=0), Any)
225
+ assert_type(np.amin(AR_b, keepdims=True), Any)
226
+ assert_type(np.amin(AR_f4, keepdims=True), Any)
227
+ assert_type(np.amin(AR_f4, out=AR_subclass), NDArraySubclass)
228
+
229
+ assert_type(np.prod(AR_b), np.int_)
230
+ assert_type(np.prod(AR_u8), np.uint64)
231
+ assert_type(np.prod(AR_i8), np.int64)
232
+ assert_type(np.prod(AR_f4), np.floating[Any])
233
+ assert_type(np.prod(AR_c16), np.complexfloating[Any, Any])
234
+ assert_type(np.prod(AR_O), Any)
235
+ assert_type(np.prod(AR_f4, axis=0), Any)
236
+ assert_type(np.prod(AR_f4, keepdims=True), Any)
237
+ assert_type(np.prod(AR_f4, dtype=np.float64), np.float64)
238
+ assert_type(np.prod(AR_f4, dtype=float), Any)
239
+ assert_type(np.prod(AR_f4, out=AR_subclass), NDArraySubclass)
240
+
241
+ assert_type(np.cumprod(AR_b), npt.NDArray[np.int_])
242
+ assert_type(np.cumprod(AR_u8), npt.NDArray[np.uint64])
243
+ assert_type(np.cumprod(AR_i8), npt.NDArray[np.int64])
244
+ assert_type(np.cumprod(AR_f4), npt.NDArray[np.floating[Any]])
245
+ assert_type(np.cumprod(AR_c16), npt.NDArray[np.complexfloating[Any, Any]])
246
+ assert_type(np.cumprod(AR_O), npt.NDArray[np.object_])
247
+ assert_type(np.cumprod(AR_f4, axis=0), npt.NDArray[np.floating[Any]])
248
+ assert_type(np.cumprod(AR_f4, dtype=np.float64), npt.NDArray[np.float64])
249
+ assert_type(np.cumprod(AR_f4, dtype=float), npt.NDArray[Any])
250
+ assert_type(np.cumprod(AR_f4, out=AR_subclass), NDArraySubclass)
251
+
252
+ assert_type(np.ndim(b), int)
253
+ assert_type(np.ndim(f4), int)
254
+ assert_type(np.ndim(f), int)
255
+ assert_type(np.ndim(AR_b), int)
256
+ assert_type(np.ndim(AR_f4), int)
257
+
258
+ assert_type(np.size(b), int)
259
+ assert_type(np.size(f4), int)
260
+ assert_type(np.size(f), int)
261
+ assert_type(np.size(AR_b), int)
262
+ assert_type(np.size(AR_f4), int)
263
+
264
+ assert_type(np.around(b), np.float16)
265
+ assert_type(np.around(f), Any)
266
+ assert_type(np.around(i8), np.int64)
267
+ assert_type(np.around(f4), np.float32)
268
+ assert_type(np.around(AR_b), npt.NDArray[np.float16])
269
+ assert_type(np.around(AR_i8), npt.NDArray[np.int64])
270
+ assert_type(np.around(AR_f4), npt.NDArray[np.float32])
271
+ assert_type(np.around([1.5]), npt.NDArray[Any])
272
+ assert_type(np.around(AR_f4, out=AR_subclass), NDArraySubclass)
273
+
274
+ assert_type(np.mean(AR_b), np.floating[Any])
275
+ assert_type(np.mean(AR_i8), np.floating[Any])
276
+ assert_type(np.mean(AR_f4), np.floating[Any])
277
+ assert_type(np.mean(AR_c16), np.complexfloating[Any, Any])
278
+ assert_type(np.mean(AR_O), Any)
279
+ assert_type(np.mean(AR_f4, axis=0), Any)
280
+ assert_type(np.mean(AR_f4, keepdims=True), Any)
281
+ assert_type(np.mean(AR_f4, dtype=float), Any)
282
+ assert_type(np.mean(AR_f4, dtype=np.float64), np.float64)
283
+ assert_type(np.mean(AR_f4, out=AR_subclass), NDArraySubclass)
284
+
285
+ assert_type(np.std(AR_b), np.floating[Any])
286
+ assert_type(np.std(AR_i8), np.floating[Any])
287
+ assert_type(np.std(AR_f4), np.floating[Any])
288
+ assert_type(np.std(AR_c16), np.floating[Any])
289
+ assert_type(np.std(AR_O), Any)
290
+ assert_type(np.std(AR_f4, axis=0), Any)
291
+ assert_type(np.std(AR_f4, keepdims=True), Any)
292
+ assert_type(np.std(AR_f4, dtype=float), Any)
293
+ assert_type(np.std(AR_f4, dtype=np.float64), np.float64)
294
+ assert_type(np.std(AR_f4, out=AR_subclass), NDArraySubclass)
295
+
296
+ assert_type(np.var(AR_b), np.floating[Any])
297
+ assert_type(np.var(AR_i8), np.floating[Any])
298
+ assert_type(np.var(AR_f4), np.floating[Any])
299
+ assert_type(np.var(AR_c16), np.floating[Any])
300
+ assert_type(np.var(AR_O), Any)
301
+ assert_type(np.var(AR_f4, axis=0), Any)
302
+ assert_type(np.var(AR_f4, keepdims=True), Any)
303
+ assert_type(np.var(AR_f4, dtype=float), Any)
304
+ assert_type(np.var(AR_f4, dtype=np.float64), np.float64)
305
+ assert_type(np.var(AR_f4, out=AR_subclass), NDArraySubclass)
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/reveal/memmap.pyi ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import sys
2
+ from typing import Any
3
+
4
+ import numpy as np
5
+
6
+ if sys.version_info >= (3, 11):
7
+ from typing import assert_type
8
+ else:
9
+ from typing_extensions import assert_type
10
+
11
+ memmap_obj: np.memmap[Any, np.dtype[np.str_]]
12
+
13
+ assert_type(np.memmap.__array_priority__, float)
14
+ assert_type(memmap_obj.__array_priority__, float)
15
+ assert_type(memmap_obj.filename, str | None)
16
+ assert_type(memmap_obj.offset, int)
17
+ assert_type(memmap_obj.mode, str)
18
+ assert_type(memmap_obj.flush(), None)
19
+
20
+ assert_type(np.memmap("file.txt", offset=5), np.memmap[Any, np.dtype[np.uint8]])
21
+ assert_type(np.memmap(b"file.txt", dtype=np.float64, shape=(10, 3)), np.memmap[Any, np.dtype[np.float64]])
22
+ with open("file.txt", "rb") as f:
23
+ assert_type(np.memmap(f, dtype=float, order="K"), np.memmap[Any, np.dtype[Any]])
24
+
25
+ assert_type(memmap_obj.__array_finalize__(object()), None)
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/reveal/mod.pyi ADDED
@@ -0,0 +1,148 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import sys
2
+ from typing import Any
3
+
4
+ import numpy as np
5
+ import numpy.typing as npt
6
+ from numpy._typing import _32Bit, _64Bit
7
+
8
+ if sys.version_info >= (3, 11):
9
+ from typing import assert_type
10
+ else:
11
+ from typing_extensions import assert_type
12
+
13
+ f8 = np.float64()
14
+ i8 = np.int64()
15
+ u8 = np.uint64()
16
+
17
+ f4 = np.float32()
18
+ i4 = np.int32()
19
+ u4 = np.uint32()
20
+
21
+ td = np.timedelta64(0, "D")
22
+ b_ = np.bool_()
23
+
24
+ b = bool()
25
+ f = float()
26
+ i = int()
27
+
28
+ AR_b: npt.NDArray[np.bool_]
29
+ AR_m: npt.NDArray[np.timedelta64]
30
+
31
+ # Time structures
32
+
33
+ assert_type(td % td, np.timedelta64)
34
+ assert_type(AR_m % td, npt.NDArray[np.timedelta64])
35
+ assert_type(td % AR_m, npt.NDArray[np.timedelta64])
36
+
37
+ assert_type(divmod(td, td), tuple[np.int64, np.timedelta64])
38
+ assert_type(divmod(AR_m, td), tuple[npt.NDArray[np.int64], npt.NDArray[np.timedelta64]])
39
+ assert_type(divmod(td, AR_m), tuple[npt.NDArray[np.int64], npt.NDArray[np.timedelta64]])
40
+
41
+ # Bool
42
+
43
+ assert_type(b_ % b, np.int8)
44
+ assert_type(b_ % i, np.int_)
45
+ assert_type(b_ % f, np.float64)
46
+ assert_type(b_ % b_, np.int8)
47
+ assert_type(b_ % i8, np.int64)
48
+ assert_type(b_ % u8, np.uint64)
49
+ assert_type(b_ % f8, np.float64)
50
+ assert_type(b_ % AR_b, npt.NDArray[np.int8])
51
+
52
+ assert_type(divmod(b_, b), tuple[np.int8, np.int8])
53
+ assert_type(divmod(b_, i), tuple[np.int_, np.int_])
54
+ assert_type(divmod(b_, f), tuple[np.float64, np.float64])
55
+ assert_type(divmod(b_, b_), tuple[np.int8, np.int8])
56
+ assert_type(divmod(b_, i8), tuple[np.int64, np.int64])
57
+ assert_type(divmod(b_, u8), tuple[np.uint64, np.uint64])
58
+ assert_type(divmod(b_, f8), tuple[np.float64, np.float64])
59
+ assert_type(divmod(b_, AR_b), tuple[npt.NDArray[np.int8], npt.NDArray[np.int8]])
60
+
61
+ assert_type(b % b_, np.int8)
62
+ assert_type(i % b_, np.int_)
63
+ assert_type(f % b_, np.float64)
64
+ assert_type(b_ % b_, np.int8)
65
+ assert_type(i8 % b_, np.int64)
66
+ assert_type(u8 % b_, np.uint64)
67
+ assert_type(f8 % b_, np.float64)
68
+ assert_type(AR_b % b_, npt.NDArray[np.int8])
69
+
70
+ assert_type(divmod(b, b_), tuple[np.int8, np.int8])
71
+ assert_type(divmod(i, b_), tuple[np.int_, np.int_])
72
+ assert_type(divmod(f, b_), tuple[np.float64, np.float64])
73
+ assert_type(divmod(b_, b_), tuple[np.int8, np.int8])
74
+ assert_type(divmod(i8, b_), tuple[np.int64, np.int64])
75
+ assert_type(divmod(u8, b_), tuple[np.uint64, np.uint64])
76
+ assert_type(divmod(f8, b_), tuple[np.float64, np.float64])
77
+ assert_type(divmod(AR_b, b_), tuple[npt.NDArray[np.int8], npt.NDArray[np.int8]])
78
+
79
+ # int
80
+
81
+ assert_type(i8 % b, np.int64)
82
+ assert_type(i8 % f, np.float64)
83
+ assert_type(i8 % i8, np.int64)
84
+ assert_type(i8 % f8, np.float64)
85
+ assert_type(i4 % i8, np.signedinteger[_32Bit | _64Bit])
86
+ assert_type(i4 % f8, np.floating[_32Bit | _64Bit])
87
+ assert_type(i4 % i4, np.int32)
88
+ assert_type(i4 % f4, np.float32)
89
+ assert_type(i8 % AR_b, npt.NDArray[np.signedinteger[Any]])
90
+
91
+ assert_type(divmod(i8, b), tuple[np.int64, np.int64])
92
+ assert_type(divmod(i8, f), tuple[np.float64, np.float64])
93
+ assert_type(divmod(i8, i8), tuple[np.int64, np.int64])
94
+ assert_type(divmod(i8, f8), tuple[np.float64, np.float64])
95
+ assert_type(divmod(i8, i4), tuple[np.signedinteger[_32Bit | _64Bit], np.signedinteger[_32Bit | _64Bit]])
96
+ assert_type(divmod(i8, f4), tuple[np.floating[_32Bit | _64Bit], np.floating[_32Bit | _64Bit]])
97
+ assert_type(divmod(i4, i4), tuple[np.int32, np.int32])
98
+ assert_type(divmod(i4, f4), tuple[np.float32, np.float32])
99
+ assert_type(divmod(i8, AR_b), tuple[npt.NDArray[np.signedinteger[Any]], npt.NDArray[np.signedinteger[Any]]])
100
+
101
+ assert_type(b % i8, np.int64)
102
+ assert_type(f % i8, np.float64)
103
+ assert_type(i8 % i8, np.int64)
104
+ assert_type(f8 % i8, np.float64)
105
+ assert_type(i8 % i4, np.signedinteger[_32Bit | _64Bit])
106
+ assert_type(f8 % i4, np.floating[_32Bit | _64Bit])
107
+ assert_type(i4 % i4, np.int32)
108
+ assert_type(f4 % i4, np.float32)
109
+ assert_type(AR_b % i8, npt.NDArray[np.signedinteger[Any]])
110
+
111
+ assert_type(divmod(b, i8), tuple[np.int64, np.int64])
112
+ assert_type(divmod(f, i8), tuple[np.float64, np.float64])
113
+ assert_type(divmod(i8, i8), tuple[np.int64, np.int64])
114
+ assert_type(divmod(f8, i8), tuple[np.float64, np.float64])
115
+ assert_type(divmod(i4, i8), tuple[np.signedinteger[_32Bit | _64Bit], np.signedinteger[_32Bit | _64Bit]])
116
+ assert_type(divmod(f4, i8), tuple[np.floating[_32Bit | _64Bit], np.floating[_32Bit | _64Bit]])
117
+ assert_type(divmod(i4, i4), tuple[np.int32, np.int32])
118
+ assert_type(divmod(f4, i4), tuple[np.float32, np.float32])
119
+ assert_type(divmod(AR_b, i8), tuple[npt.NDArray[np.signedinteger[Any]], npt.NDArray[np.signedinteger[Any]]])
120
+
121
+ # float
122
+
123
+ assert_type(f8 % b, np.float64)
124
+ assert_type(f8 % f, np.float64)
125
+ assert_type(i8 % f4, np.floating[_32Bit | _64Bit])
126
+ assert_type(f4 % f4, np.float32)
127
+ assert_type(f8 % AR_b, npt.NDArray[np.floating[Any]])
128
+
129
+ assert_type(divmod(f8, b), tuple[np.float64, np.float64])
130
+ assert_type(divmod(f8, f), tuple[np.float64, np.float64])
131
+ assert_type(divmod(f8, f8), tuple[np.float64, np.float64])
132
+ assert_type(divmod(f8, f4), tuple[np.floating[_32Bit | _64Bit], np.floating[_32Bit | _64Bit]])
133
+ assert_type(divmod(f4, f4), tuple[np.float32, np.float32])
134
+ assert_type(divmod(f8, AR_b), tuple[npt.NDArray[np.floating[Any]], npt.NDArray[np.floating[Any]]])
135
+
136
+ assert_type(b % f8, np.float64)
137
+ assert_type(f % f8, np.float64)
138
+ assert_type(f8 % f8, np.float64)
139
+ assert_type(f8 % f8, np.float64)
140
+ assert_type(f4 % f4, np.float32)
141
+ assert_type(AR_b % f8, npt.NDArray[np.floating[Any]])
142
+
143
+ assert_type(divmod(b, f8), tuple[np.float64, np.float64])
144
+ assert_type(divmod(f, f8), tuple[np.float64, np.float64])
145
+ assert_type(divmod(f8, f8), tuple[np.float64, np.float64])
146
+ assert_type(divmod(f4, f8), tuple[np.floating[_32Bit | _64Bit], np.floating[_32Bit | _64Bit]])
147
+ assert_type(divmod(f4, f4), tuple[np.float32, np.float32])
148
+ assert_type(divmod(AR_b, f8), tuple[npt.NDArray[np.floating[Any]], npt.NDArray[np.floating[Any]]])
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/reveal/ndarray_shape_manipulation.pyi ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import sys
2
+ from typing import Any
3
+
4
+ import numpy as np
5
+ import numpy.typing as npt
6
+
7
+ if sys.version_info >= (3, 11):
8
+ from typing import assert_type
9
+ else:
10
+ from typing_extensions import assert_type
11
+
12
+ nd: npt.NDArray[np.int64]
13
+
14
+ # reshape
15
+ assert_type(nd.reshape(), npt.NDArray[np.int64])
16
+ assert_type(nd.reshape(4), npt.NDArray[np.int64])
17
+ assert_type(nd.reshape(2, 2), npt.NDArray[np.int64])
18
+ assert_type(nd.reshape((2, 2)), npt.NDArray[np.int64])
19
+
20
+ assert_type(nd.reshape((2, 2), order="C"), npt.NDArray[np.int64])
21
+ assert_type(nd.reshape(4, order="C"), npt.NDArray[np.int64])
22
+
23
+ # resize does not return a value
24
+
25
+ # transpose
26
+ assert_type(nd.transpose(), npt.NDArray[np.int64])
27
+ assert_type(nd.transpose(1, 0), npt.NDArray[np.int64])
28
+ assert_type(nd.transpose((1, 0)), npt.NDArray[np.int64])
29
+
30
+ # swapaxes
31
+ assert_type(nd.swapaxes(0, 1), npt.NDArray[np.int64])
32
+
33
+ # flatten
34
+ assert_type(nd.flatten(), npt.NDArray[np.int64])
35
+ assert_type(nd.flatten("C"), npt.NDArray[np.int64])
36
+
37
+ # ravel
38
+ assert_type(nd.ravel(), npt.NDArray[np.int64])
39
+ assert_type(nd.ravel("C"), npt.NDArray[np.int64])
40
+
41
+ # squeeze
42
+ assert_type(nd.squeeze(), npt.NDArray[np.int64])
43
+ assert_type(nd.squeeze(0), npt.NDArray[np.int64])
44
+ assert_type(nd.squeeze((0, 2)), npt.NDArray[np.int64])
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/reveal/nditer.pyi ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import sys
2
+ from typing import Any
3
+
4
+ import numpy as np
5
+ import numpy.typing as npt
6
+
7
+ if sys.version_info >= (3, 11):
8
+ from typing import assert_type
9
+ else:
10
+ from typing_extensions import assert_type
11
+
12
+ nditer_obj: np.nditer
13
+
14
+ assert_type(np.nditer([0, 1], flags=["c_index"]), np.nditer)
15
+ assert_type(np.nditer([0, 1], op_flags=[["readonly", "readonly"]]), np.nditer)
16
+ assert_type(np.nditer([0, 1], op_dtypes=np.int_), np.nditer)
17
+ assert_type(np.nditer([0, 1], order="C", casting="no"), np.nditer)
18
+
19
+ assert_type(nditer_obj.dtypes, tuple[np.dtype[Any], ...])
20
+ assert_type(nditer_obj.finished, bool)
21
+ assert_type(nditer_obj.has_delayed_bufalloc, bool)
22
+ assert_type(nditer_obj.has_index, bool)
23
+ assert_type(nditer_obj.has_multi_index, bool)
24
+ assert_type(nditer_obj.index, int)
25
+ assert_type(nditer_obj.iterationneedsapi, bool)
26
+ assert_type(nditer_obj.iterindex, int)
27
+ assert_type(nditer_obj.iterrange, tuple[int, ...])
28
+ assert_type(nditer_obj.itersize, int)
29
+ assert_type(nditer_obj.itviews, tuple[npt.NDArray[Any], ...])
30
+ assert_type(nditer_obj.multi_index, tuple[int, ...])
31
+ assert_type(nditer_obj.ndim, int)
32
+ assert_type(nditer_obj.nop, int)
33
+ assert_type(nditer_obj.operands, tuple[npt.NDArray[Any], ...])
34
+ assert_type(nditer_obj.shape, tuple[int, ...])
35
+ assert_type(nditer_obj.value, tuple[npt.NDArray[Any], ...])
36
+
37
+ assert_type(nditer_obj.close(), None)
38
+ assert_type(nditer_obj.copy(), np.nditer)
39
+ assert_type(nditer_obj.debug_print(), None)
40
+ assert_type(nditer_obj.enable_external_loop(), None)
41
+ assert_type(nditer_obj.iternext(), bool)
42
+ assert_type(nditer_obj.remove_axis(0), None)
43
+ assert_type(nditer_obj.remove_multi_index(), None)
44
+ assert_type(nditer_obj.reset(), None)
45
+
46
+ assert_type(len(nditer_obj), int)
47
+ assert_type(iter(nditer_obj), np.nditer)
48
+ assert_type(next(nditer_obj), tuple[npt.NDArray[Any], ...])
49
+ assert_type(nditer_obj.__copy__(), np.nditer)
50
+ with nditer_obj as f:
51
+ assert_type(f, np.nditer)
52
+ assert_type(nditer_obj[0], npt.NDArray[Any])
53
+ assert_type(nditer_obj[:], tuple[npt.NDArray[Any], ...])
54
+ nditer_obj[0] = 0
55
+ nditer_obj[:] = [0, 1]
LTA_openwebtext_dualt/mini_owt_logdirichlet/.venv_qwen35_uv/lib/python3.12/site-packages/numpy/typing/tests/data/reveal/warnings_and_errors.pyi ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import sys
2
+
3
+ import numpy as np
4
+
5
+ if sys.version_info >= (3, 11):
6
+ from typing import assert_type
7
+ else:
8
+ from typing_extensions import assert_type
9
+
10
+ assert_type(np.ModuleDeprecationWarning(), np.ModuleDeprecationWarning)
11
+ assert_type(np.VisibleDeprecationWarning(), np.VisibleDeprecationWarning)
12
+ assert_type(np.ComplexWarning(), np.ComplexWarning)
13
+ assert_type(np.RankWarning(), np.RankWarning)
14
+ assert_type(np.TooHardError(), np.TooHardError)
15
+ assert_type(np.AxisError("test"), np.AxisError)
16
+ assert_type(np.AxisError(5, 1), np.AxisError)
LTA_openwebtext_dualt/mini_owt_logdirichlet/logs/decode1_4_uniform_online_20260610/lr2e3_online_latest_decode1_4_uniform.log ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ checkpoint=runs/owt_t5_elftokenized_full_len1024_C1_to_1024_pow1_d768_l12_h12_gbs512_8gpu_50ep_lr2e3_ema0p9999_elfopt_t5embed_unfixed_norm_stateprobadd_selfcond_ce_fast_20260606_144231/step_292000.pt
2
+ use_ema=0
3
+ step=292000
4
+ decode_steps=1 4
5
+ n=64 chunk_n=8 gpu=0
6
+ out_base=/e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260610
7
+ [2026-06-09T22:56:14+00:00] infer step=292000 decode=1 -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260610/owt_t5_elftokenized_full_pow1_unfixed_norm_stateprobadd_selfcond_ce_fast_lr2e3_ema0p9999_online_uniform_step292000_sc1p0_decode1_n64
8
+ [2026-06-09T22:56:14+00:00] run decode=1 chunk=0 n=8 seed=123
9
+ [2026-06-09T22:56:18+00:00] done decode=1 chunk=0
10
+ [2026-06-09T22:56:18+00:00] run decode=1 chunk=1 n=8 seed=124
11
+ [2026-06-09T22:56:22+00:00] done decode=1 chunk=1
12
+ [2026-06-09T22:56:22+00:00] run decode=1 chunk=2 n=8 seed=125
13
+ [2026-06-09T22:56:26+00:00] done decode=1 chunk=2
14
+ [2026-06-09T22:56:26+00:00] run decode=1 chunk=3 n=8 seed=126
15
+ [2026-06-09T22:56:30+00:00] done decode=1 chunk=3
16
+ [2026-06-09T22:56:30+00:00] run decode=1 chunk=4 n=8 seed=127
17
+ [2026-06-09T22:56:35+00:00] done decode=1 chunk=4
18
+ [2026-06-09T22:56:35+00:00] run decode=1 chunk=5 n=8 seed=128
19
+ [2026-06-09T22:56:39+00:00] done decode=1 chunk=5
20
+ [2026-06-09T22:56:39+00:00] run decode=1 chunk=6 n=8 seed=129
21
+ [2026-06-09T22:56:43+00:00] done decode=1 chunk=6
22
+ [2026-06-09T22:56:43+00:00] run decode=1 chunk=7 n=8 seed=130
23
+ [2026-06-09T22:56:47+00:00] done decode=1 chunk=7
24
+ merged 64 samples -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260610/owt_t5_elftokenized_full_pow1_unfixed_norm_stateprobadd_selfcond_ce_fast_lr2e3_ema0p9999_online_uniform_step292000_sc1p0_decode1_n64/sc1p0/samples64.txt
25
+ loading scorer /e2e-data/evad-tech-vla/wanghan58/models/flowtext_scorers/gpt2-large-standard dtype=fp32 device=cuda
26
+ run kind ppl mean_entropy distinct_1 distinct_2 top_token_mass eos_rows eos_total ppl_tokens t5_tokens path
27
+ sc1p0 raw_full 1.096572649728622 0.11783789575524715 0.0004955050612302683 0.001132663174288546 0.9815247398598429 0 0 79304 14127 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260610/owt_t5_elftokenized_full_pow1_unfixed_norm_stateprobadd_selfcond_ce_fast_lr2e3_ema0p9999_online_uniform_step292000_sc1p0_decode1_n64/sc1p0
28
+ sc1p0 pre_eos 1.096572649728622 0.11783789575524715 0.0004955050612302683 0.001132663174288546 0.9815247398598429 0 0 79304 14127 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260610/owt_t5_elftokenized_full_pow1_unfixed_norm_stateprobadd_selfcond_ce_fast_lr2e3_ema0p9999_online_uniform_step292000_sc1p0_decode1_n64/sc1p0
29
+ [2026-06-09T22:57:02+00:00] infer step=292000 decode=4 -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260610/owt_t5_elftokenized_full_pow1_unfixed_norm_stateprobadd_selfcond_ce_fast_lr2e3_ema0p9999_online_uniform_step292000_sc1p0_decode4_n64
30
+ [2026-06-09T22:57:02+00:00] run decode=4 chunk=0 n=8 seed=123
31
+ [2026-06-09T22:57:07+00:00] done decode=4 chunk=0
32
+ [2026-06-09T22:57:07+00:00] run decode=4 chunk=1 n=8 seed=124
33
+ [2026-06-09T22:57:11+00:00] done decode=4 chunk=1
34
+ [2026-06-09T22:57:11+00:00] run decode=4 chunk=2 n=8 seed=125
35
+ [2026-06-09T22:57:16+00:00] done decode=4 chunk=2
36
+ [2026-06-09T22:57:16+00:00] run decode=4 chunk=3 n=8 seed=126
37
+ [2026-06-09T22:57:20+00:00] done decode=4 chunk=3
38
+ [2026-06-09T22:57:20+00:00] run decode=4 chunk=4 n=8 seed=127
39
+ [2026-06-09T22:57:25+00:00] done decode=4 chunk=4
40
+ [2026-06-09T22:57:25+00:00] run decode=4 chunk=5 n=8 seed=128
41
+ [2026-06-09T22:57:29+00:00] done decode=4 chunk=5
42
+ [2026-06-09T22:57:29+00:00] run decode=4 chunk=6 n=8 seed=129
43
+ [2026-06-09T22:57:34+00:00] done decode=4 chunk=6
44
+ [2026-06-09T22:57:34+00:00] run decode=4 chunk=7 n=8 seed=130
45
+ [2026-06-09T22:57:39+00:00] done decode=4 chunk=7
46
+ merged 64 samples -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260610/owt_t5_elftokenized_full_pow1_unfixed_norm_stateprobadd_selfcond_ce_fast_lr2e3_ema0p9999_online_uniform_step292000_sc1p0_decode4_n64/sc1p0/samples64.txt
47
+ loading scorer /e2e-data/evad-tech-vla/wanghan58/models/flowtext_scorers/gpt2-large-standard dtype=fp32 device=cuda
48
+ run kind ppl mean_entropy distinct_1 distinct_2 top_token_mass eos_rows eos_total ppl_tokens t5_tokens path
49
+ sc1p0 raw_full 3.9144810000584296 1.655964156095804 0.002519893899204244 0.010636886920077455 0.42931034482758623 8 8 53648 37700 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260610/owt_t5_elftokenized_full_pow1_unfixed_norm_stateprobadd_selfcond_ce_fast_lr2e3_ema0p9999_online_uniform_step292000_sc1p0_decode4_n64/sc1p0
50
+ sc1p0 pre_eos 4.001451443769343 1.6315027573148633 0.002560163850486431 0.010779927774483911 0.4361441237502358 0 0 52447 37107 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260610/owt_t5_elftokenized_full_pow1_unfixed_norm_stateprobadd_selfcond_ce_fast_lr2e3_ema0p9999_online_uniform_step292000_sc1p0_decode4_n64/sc1p0
51
+ [2026-06-09T22:57:50+00:00] done
LTA_openwebtext_dualt/mini_owt_logdirichlet/logs/decode1_4_uniform_online_20260610/lr6e4_online_latest_decode1_4_uniform.log ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ checkpoint=runs/owt_t5_elftokenized_full_len1024_C1_to_1024_pow1_d768_l12_h12_gbs512_8gpu_50ep_lr6e4_ema0p9999_elfopt_t5embed_unfixed_norm_stateprobadd_selfcond_ce_fast_20260606_144245/step_297000.pt
2
+ use_ema=0
3
+ step=297000
4
+ decode_steps=1 4
5
+ n=64 chunk_n=8 gpu=1
6
+ out_base=/e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260610
7
+ [2026-06-09T22:56:14+00:00] infer step=297000 decode=1 -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260610/owt_t5_elftokenized_full_pow1_unfixed_norm_stateprobadd_selfcond_ce_fast_lr6e4_ema0p9999_online_uniform_step297000_sc1p0_decode1_n64
8
+ [2026-06-09T22:56:14+00:00] run decode=1 chunk=0 n=8 seed=123
9
+ [2026-06-09T22:56:18+00:00] done decode=1 chunk=0
10
+ [2026-06-09T22:56:18+00:00] run decode=1 chunk=1 n=8 seed=124
11
+ [2026-06-09T22:56:23+00:00] done decode=1 chunk=1
12
+ [2026-06-09T22:56:23+00:00] run decode=1 chunk=2 n=8 seed=125
13
+ [2026-06-09T22:56:27+00:00] done decode=1 chunk=2
14
+ [2026-06-09T22:56:27+00:00] run decode=1 chunk=3 n=8 seed=126
15
+ [2026-06-09T22:56:31+00:00] done decode=1 chunk=3
16
+ [2026-06-09T22:56:31+00:00] run decode=1 chunk=4 n=8 seed=127
17
+ [2026-06-09T22:56:35+00:00] done decode=1 chunk=4
18
+ [2026-06-09T22:56:35+00:00] run decode=1 chunk=5 n=8 seed=128
19
+ [2026-06-09T22:56:40+00:00] done decode=1 chunk=5
20
+ [2026-06-09T22:56:40+00:00] run decode=1 chunk=6 n=8 seed=129
21
+ [2026-06-09T22:56:44+00:00] done decode=1 chunk=6
22
+ [2026-06-09T22:56:44+00:00] run decode=1 chunk=7 n=8 seed=130
23
+ [2026-06-09T22:56:48+00:00] done decode=1 chunk=7
24
+ merged 64 samples -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260610/owt_t5_elftokenized_full_pow1_unfixed_norm_stateprobadd_selfcond_ce_fast_lr6e4_ema0p9999_online_uniform_step297000_sc1p0_decode1_n64/sc1p0/samples64.txt
25
+ loading scorer /e2e-data/evad-tech-vla/wanghan58/models/flowtext_scorers/gpt2-large-standard dtype=fp32 device=cuda
26
+ run kind ppl mean_entropy distinct_1 distinct_2 top_token_mass eos_rows eos_total ppl_tokens t5_tokens path
27
+ sc1p0 raw_full 1.1010033551856535 0.10100969198731712 0.0004690706537672237 0.0008209217778820218 0.9846379360891234 0 0 82237 17055 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260610/owt_t5_elftokenized_full_pow1_unfixed_norm_stateprobadd_selfcond_ce_fast_lr6e4_ema0p9999_online_uniform_step297000_sc1p0_decode1_n64/sc1p0
28
+ sc1p0 pre_eos 1.1010033551856535 0.10100969198731712 0.0004690706537672237 0.0008209217778820218 0.9846379360891234 0 0 82237 17055 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260610/owt_t5_elftokenized_full_pow1_unfixed_norm_stateprobadd_selfcond_ce_fast_lr6e4_ema0p9999_online_uniform_step297000_sc1p0_decode1_n64/sc1p0
29
+ [2026-06-09T22:57:03+00:00] infer step=297000 decode=4 -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260610/owt_t5_elftokenized_full_pow1_unfixed_norm_stateprobadd_selfcond_ce_fast_lr6e4_ema0p9999_online_uniform_step297000_sc1p0_decode4_n64
30
+ [2026-06-09T22:57:03+00:00] run decode=4 chunk=0 n=8 seed=123
31
+ [2026-06-09T22:57:08+00:00] done decode=4 chunk=0
32
+ [2026-06-09T22:57:08+00:00] run decode=4 chunk=1 n=8 seed=124
33
+ [2026-06-09T22:57:12+00:00] done decode=4 chunk=1
34
+ [2026-06-09T22:57:12+00:00] run decode=4 chunk=2 n=8 seed=125
35
+ [2026-06-09T22:57:16+00:00] done decode=4 chunk=2
36
+ [2026-06-09T22:57:16+00:00] run decode=4 chunk=3 n=8 seed=126
37
+ [2026-06-09T22:57:21+00:00] done decode=4 chunk=3
38
+ [2026-06-09T22:57:21+00:00] run decode=4 chunk=4 n=8 seed=127
39
+ [2026-06-09T22:57:25+00:00] done decode=4 chunk=4
40
+ [2026-06-09T22:57:25+00:00] run decode=4 chunk=5 n=8 seed=128
41
+ [2026-06-09T22:57:30+00:00] done decode=4 chunk=5
42
+ [2026-06-09T22:57:30+00:00] run decode=4 chunk=6 n=8 seed=129
43
+ [2026-06-09T22:57:34+00:00] done decode=4 chunk=6
44
+ [2026-06-09T22:57:34+00:00] run decode=4 chunk=7 n=8 seed=130
45
+ [2026-06-09T22:57:39+00:00] done decode=4 chunk=7
46
+ merged 64 samples -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260610/owt_t5_elftokenized_full_pow1_unfixed_norm_stateprobadd_selfcond_ce_fast_lr6e4_ema0p9999_online_uniform_step297000_sc1p0_decode4_n64/sc1p0/samples64.txt
47
+ loading scorer /e2e-data/evad-tech-vla/wanghan58/models/flowtext_scorers/gpt2-large-standard dtype=fp32 device=cuda
48
+ run kind ppl mean_entropy distinct_1 distinct_2 top_token_mass eos_rows eos_total ppl_tokens t5_tokens path
49
+ sc1p0 raw_full 1.862719202099405 0.4595035966522467 0.0006571908910285801 0.0022008589463387795 0.8955524988537368 4 4 38889 65430 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260610/owt_t5_elftokenized_full_pow1_unfixed_norm_stateprobadd_selfcond_ce_fast_lr6e4_ema0p9999_online_uniform_step297000_sc1p0_decode4_n64/sc1p0
50
+ sc1p0 pre_eos 1.8667632184252236 0.44590311903537977 0.0006596609649459231 0.002193789887088856 0.8989184628365422 0 0 38395 65185 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260610/owt_t5_elftokenized_full_pow1_unfixed_norm_stateprobadd_selfcond_ce_fast_lr6e4_ema0p9999_online_uniform_step297000_sc1p0_decode4_n64/sc1p0
51
+ [2026-06-09T22:57:49+00:00] done
LTA_openwebtext_dualt/mini_owt_logdirichlet/logs/infer_lr3e3_170k_logitnorm64_focused_20260612/gpu3.log ADDED
@@ -0,0 +1,352 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ===== START \2026-06-11T22:34:52+00:00 gpu=3 m=-0.5 s=1.4 gamma=0.25 =====
2
+ checkpoint=runs/owt_t5_elftokenized_full_len1024_C1_to_1024_pow1_d768_l12_h12_gbs512_2x8gpu_50ep_lr3e3_ema0p9999_elfopt_not5_bottleneck128_unfixed_norm_stateprobadd_selfcond_ce_fast_20260609_230847/step_170000.pt
3
+ use_ema=1
4
+ step=170000
5
+ decode_steps=64
6
+ n=64 chunk_n=8 gpu=3
7
+ out_base=/e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612
8
+ [2026-06-11T22:34:52+00:00] infer step=170000 decode=64 -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p5_s1p4_sc1p0_decode64_n64
9
+ [2026-06-11T22:34:52+00:00] run decode=64 chunk=0 n=8 seed=123
10
+ [2026-06-11T22:35:02+00:00] done decode=64 chunk=0
11
+ [2026-06-11T22:35:02+00:00] run decode=64 chunk=1 n=8 seed=124
12
+ [2026-06-11T22:35:12+00:00] done decode=64 chunk=1
13
+ [2026-06-11T22:35:12+00:00] run decode=64 chunk=2 n=8 seed=125
14
+ [2026-06-11T22:35:21+00:00] done decode=64 chunk=2
15
+ [2026-06-11T22:35:21+00:00] run decode=64 chunk=3 n=8 seed=126
16
+ [2026-06-11T22:35:31+00:00] done decode=64 chunk=3
17
+ [2026-06-11T22:35:31+00:00] run decode=64 chunk=4 n=8 seed=127
18
+ [2026-06-11T22:35:41+00:00] done decode=64 chunk=4
19
+ [2026-06-11T22:35:41+00:00] run decode=64 chunk=5 n=8 seed=128
20
+ [2026-06-11T22:35:50+00:00] done decode=64 chunk=5
21
+ [2026-06-11T22:35:50+00:00] run decode=64 chunk=6 n=8 seed=129
22
+ [2026-06-11T22:36:00+00:00] done decode=64 chunk=6
23
+ [2026-06-11T22:36:00+00:00] run decode=64 chunk=7 n=8 seed=130
24
+ [2026-06-11T22:36:10+00:00] done decode=64 chunk=7
25
+ merged 64 samples -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p5_s1p4_sc1p0_decode64_n64/sc1p0/samples64.txt
26
+ loading scorer /e2e-data/evad-tech-vla/wanghan58/models/flowtext_scorers/gpt2-large-standard dtype=fp32 device=cuda
27
+ run kind ppl mean_entropy distinct_1 distinct_2 top_token_mass eos_rows eos_total ppl_tokens t5_tokens path
28
+ sc1p0 raw_full 14.323413610616049 5.009203366356262 0.07500610053684724 0.4101758506565803 0.03593216203025866 62 62 64109 65568 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p5_s1p4_sc1p0_decode64_n64/sc1p0
29
+ sc1p0 pre_eos 16.11984853009216 5.038465452975301 0.07745990736948234 0.4236337571088741 0.03711522102145625 0 0 59907 63478 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p5_s1p4_sc1p0_decode64_n64/sc1p0
30
+ [2026-06-11T22:36:42+00:00] done
31
+ ===== DONE \2026-06-11T22:34:52+00:00 gpu=3 m=-0.5 s=1.4 gamma=0.25 =====
32
+ ===== START \2026-06-11T22:34:52+00:00 gpu=3 m=-0.5 s=1.6 gamma=0.25 =====
33
+ checkpoint=runs/owt_t5_elftokenized_full_len1024_C1_to_1024_pow1_d768_l12_h12_gbs512_2x8gpu_50ep_lr3e3_ema0p9999_elfopt_not5_bottleneck128_unfixed_norm_stateprobadd_selfcond_ce_fast_20260609_230847/step_170000.pt
34
+ use_ema=1
35
+ step=170000
36
+ decode_steps=64
37
+ n=64 chunk_n=8 gpu=3
38
+ out_base=/e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612
39
+ [2026-06-11T22:36:42+00:00] infer step=170000 decode=64 -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p5_s1p6_sc1p0_decode64_n64
40
+ [2026-06-11T22:36:42+00:00] run decode=64 chunk=0 n=8 seed=123
41
+ [2026-06-11T22:36:52+00:00] done decode=64 chunk=0
42
+ [2026-06-11T22:36:52+00:00] run decode=64 chunk=1 n=8 seed=124
43
+ [2026-06-11T22:37:02+00:00] done decode=64 chunk=1
44
+ [2026-06-11T22:37:02+00:00] run decode=64 chunk=2 n=8 seed=125
45
+ [2026-06-11T22:37:11+00:00] done decode=64 chunk=2
46
+ [2026-06-11T22:37:11+00:00] run decode=64 chunk=3 n=8 seed=126
47
+ [2026-06-11T22:37:21+00:00] done decode=64 chunk=3
48
+ [2026-06-11T22:37:21+00:00] run decode=64 chunk=4 n=8 seed=127
49
+ [2026-06-11T22:37:31+00:00] done decode=64 chunk=4
50
+ [2026-06-11T22:37:31+00:00] run decode=64 chunk=5 n=8 seed=128
51
+ [2026-06-11T22:37:41+00:00] done decode=64 chunk=5
52
+ [2026-06-11T22:37:41+00:00] run decode=64 chunk=6 n=8 seed=129
53
+ [2026-06-11T22:37:50+00:00] done decode=64 chunk=6
54
+ [2026-06-11T22:37:50+00:00] run decode=64 chunk=7 n=8 seed=130
55
+ [2026-06-11T22:38:00+00:00] done decode=64 chunk=7
56
+ merged 64 samples -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p5_s1p6_sc1p0_decode64_n64/sc1p0/samples64.txt
57
+ loading scorer /e2e-data/evad-tech-vla/wanghan58/models/flowtext_scorers/gpt2-large-standard dtype=fp32 device=cuda
58
+ run kind ppl mean_entropy distinct_1 distinct_2 top_token_mass eos_rows eos_total ppl_tokens t5_tokens path
59
+ sc1p0 raw_full 15.92467574106307 5.044005213841317 0.07881645608252456 0.42744655201355086 0.03688274430812428 63 63 63695 65532 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p5_s1p6_sc1p0_decode64_n64/sc1p0
60
+ sc1p0 pre_eos 17.980638543037145 5.073226190698163 0.08132027337721646 0.4410481724697249 0.03806179332934396 0 0 59610 63502 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p5_s1p6_sc1p0_decode64_n64/sc1p0
61
+ [2026-06-11T22:38:31+00:00] done
62
+ ===== DONE \2026-06-11T22:34:52+00:00 gpu=3 m=-0.5 s=1.6 gamma=0.25 =====
63
+ ===== START \2026-06-11T22:34:52+00:00 gpu=3 m=-0.5 s=1.8 gamma=0.25 =====
64
+ checkpoint=runs/owt_t5_elftokenized_full_len1024_C1_to_1024_pow1_d768_l12_h12_gbs512_2x8gpu_50ep_lr3e3_ema0p9999_elfopt_not5_bottleneck128_unfixed_norm_stateprobadd_selfcond_ce_fast_20260609_230847/step_170000.pt
65
+ use_ema=1
66
+ step=170000
67
+ decode_steps=64
68
+ n=64 chunk_n=8 gpu=3
69
+ out_base=/e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612
70
+ [2026-06-11T22:38:31+00:00] infer step=170000 decode=64 -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p5_s1p8_sc1p0_decode64_n64
71
+ [2026-06-11T22:38:31+00:00] run decode=64 chunk=0 n=8 seed=123
72
+ [2026-06-11T22:38:41+00:00] done decode=64 chunk=0
73
+ [2026-06-11T22:38:41+00:00] run decode=64 chunk=1 n=8 seed=124
74
+ [2026-06-11T22:38:51+00:00] done decode=64 chunk=1
75
+ [2026-06-11T22:38:51+00:00] run decode=64 chunk=2 n=8 seed=125
76
+ [2026-06-11T22:39:00+00:00] done decode=64 chunk=2
77
+ [2026-06-11T22:39:00+00:00] run decode=64 chunk=3 n=8 seed=126
78
+ [2026-06-11T22:39:10+00:00] done decode=64 chunk=3
79
+ [2026-06-11T22:39:10+00:00] run decode=64 chunk=4 n=8 seed=127
80
+ [2026-06-11T22:39:20+00:00] done decode=64 chunk=4
81
+ [2026-06-11T22:39:20+00:00] run decode=64 chunk=5 n=8 seed=128
82
+ [2026-06-11T22:39:29+00:00] done decode=64 chunk=5
83
+ [2026-06-11T22:39:29+00:00] run decode=64 chunk=6 n=8 seed=129
84
+ [2026-06-11T22:39:39+00:00] done decode=64 chunk=6
85
+ [2026-06-11T22:39:39+00:00] run decode=64 chunk=7 n=8 seed=130
86
+ [2026-06-11T22:39:49+00:00] done decode=64 chunk=7
87
+ merged 64 samples -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p5_s1p8_sc1p0_decode64_n64/sc1p0/samples64.txt
88
+ loading scorer /e2e-data/evad-tech-vla/wanghan58/models/flowtext_scorers/gpt2-large-standard dtype=fp32 device=cuda
89
+ run kind ppl mean_entropy distinct_1 distinct_2 top_token_mass eos_rows eos_total ppl_tokens t5_tokens path
90
+ sc1p0 raw_full 16.511156990368196 5.021153347378399 0.07948354013094638 0.43199230792710847 0.03784930482426018 64 64 63897 65523 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p5_s1p8_sc1p0_decode64_n64/sc1p0
91
+ sc1p0 pre_eos 19.003033006817592 5.056660543916634 0.08224998815296887 0.4470508324645373 0.03917418294975279 0 0 59441 63307 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p5_s1p8_sc1p0_decode64_n64/sc1p0
92
+ [2026-06-11T22:40:18+00:00] done
93
+ ===== DONE \2026-06-11T22:34:52+00:00 gpu=3 m=-0.5 s=1.8 gamma=0.25 =====
94
+ ===== START \2026-06-11T22:34:52+00:00 gpu=3 m=-0.5 s=2.0 gamma=0.25 =====
95
+ checkpoint=runs/owt_t5_elftokenized_full_len1024_C1_to_1024_pow1_d768_l12_h12_gbs512_2x8gpu_50ep_lr3e3_ema0p9999_elfopt_not5_bottleneck128_unfixed_norm_stateprobadd_selfcond_ce_fast_20260609_230847/step_170000.pt
96
+ use_ema=1
97
+ step=170000
98
+ decode_steps=64
99
+ n=64 chunk_n=8 gpu=3
100
+ out_base=/e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612
101
+ [2026-06-11T22:40:18+00:00] infer step=170000 decode=64 -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p5_s2p0_sc1p0_decode64_n64
102
+ [2026-06-11T22:40:18+00:00] run decode=64 chunk=0 n=8 seed=123
103
+ [2026-06-11T22:40:27+00:00] done decode=64 chunk=0
104
+ [2026-06-11T22:40:27+00:00] run decode=64 chunk=1 n=8 seed=124
105
+ [2026-06-11T22:40:37+00:00] done decode=64 chunk=1
106
+ [2026-06-11T22:40:37+00:00] run decode=64 chunk=2 n=8 seed=125
107
+ [2026-06-11T22:40:47+00:00] done decode=64 chunk=2
108
+ [2026-06-11T22:40:47+00:00] run decode=64 chunk=3 n=8 seed=126
109
+ [2026-06-11T22:40:56+00:00] done decode=64 chunk=3
110
+ [2026-06-11T22:40:56+00:00] run decode=64 chunk=4 n=8 seed=127
111
+ [2026-06-11T22:41:06+00:00] done decode=64 chunk=4
112
+ [2026-06-11T22:41:06+00:00] run decode=64 chunk=5 n=8 seed=128
113
+ [2026-06-11T22:41:16+00:00] done decode=64 chunk=5
114
+ [2026-06-11T22:41:16+00:00] run decode=64 chunk=6 n=8 seed=129
115
+ [2026-06-11T22:41:26+00:00] done decode=64 chunk=6
116
+ [2026-06-11T22:41:26+00:00] run decode=64 chunk=7 n=8 seed=130
117
+ [2026-06-11T22:41:35+00:00] done decode=64 chunk=7
118
+ merged 64 samples -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p5_s2p0_sc1p0_decode64_n64/sc1p0/samples64.txt
119
+ loading scorer /e2e-data/evad-tech-vla/wanghan58/models/flowtext_scorers/gpt2-large-standard dtype=fp32 device=cuda
120
+ run kind ppl mean_entropy distinct_1 distinct_2 top_token_mass eos_rows eos_total ppl_tokens t5_tokens path
121
+ sc1p0 raw_full 19.298351448530592 5.110950318985606 0.08453284081022087 0.4514898033947979 0.03542808297589791 62 62 63303 65513 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p5_s2p0_sc1p0_decode64_n64/sc1p0
122
+ sc1p0 pre_eos 22.248318536053922 5.146961055265112 0.08728895054624565 0.46623155505107833 0.036589787649961375 0 0 59118 63433 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p5_s2p0_sc1p0_decode64_n64/sc1p0
123
+ [2026-06-11T22:41:49+00:00] done
124
+ ===== DONE \2026-06-11T22:34:52+00:00 gpu=3 m=-0.5 s=2.0 gamma=0.25 =====
125
+ ===== START \2026-06-11T22:34:52+00:00 gpu=3 m=-0.4 s=1.4 gamma=0.25 =====
126
+ checkpoint=runs/owt_t5_elftokenized_full_len1024_C1_to_1024_pow1_d768_l12_h12_gbs512_2x8gpu_50ep_lr3e3_ema0p9999_elfopt_not5_bottleneck128_unfixed_norm_stateprobadd_selfcond_ce_fast_20260609_230847/step_170000.pt
127
+ use_ema=1
128
+ step=170000
129
+ decode_steps=64
130
+ n=64 chunk_n=8 gpu=3
131
+ out_base=/e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612
132
+ [2026-06-11T22:41:49+00:00] infer step=170000 decode=64 -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p4_s1p4_sc1p0_decode64_n64
133
+ [2026-06-11T22:41:49+00:00] run decode=64 chunk=0 n=8 seed=123
134
+ [2026-06-11T22:41:59+00:00] done decode=64 chunk=0
135
+ [2026-06-11T22:41:59+00:00] run decode=64 chunk=1 n=8 seed=124
136
+ [2026-06-11T22:42:09+00:00] done decode=64 chunk=1
137
+ [2026-06-11T22:42:09+00:00] run decode=64 chunk=2 n=8 seed=125
138
+ [2026-06-11T22:42:19+00:00] done decode=64 chunk=2
139
+ [2026-06-11T22:42:19+00:00] run decode=64 chunk=3 n=8 seed=126
140
+ [2026-06-11T22:42:28+00:00] done decode=64 chunk=3
141
+ [2026-06-11T22:42:28+00:00] run decode=64 chunk=4 n=8 seed=127
142
+ [2026-06-11T22:42:38+00:00] done decode=64 chunk=4
143
+ [2026-06-11T22:42:38+00:00] run decode=64 chunk=5 n=8 seed=128
144
+ [2026-06-11T22:42:48+00:00] done decode=64 chunk=5
145
+ [2026-06-11T22:42:48+00:00] run decode=64 chunk=6 n=8 seed=129
146
+ [2026-06-11T22:42:57+00:00] done decode=64 chunk=6
147
+ [2026-06-11T22:42:57+00:00] run decode=64 chunk=7 n=8 seed=130
148
+ [2026-06-11T22:43:07+00:00] done decode=64 chunk=7
149
+ merged 64 samples -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p4_s1p4_sc1p0_decode64_n64/sc1p0/samples64.txt
150
+ loading scorer /e2e-data/evad-tech-vla/wanghan58/models/flowtext_scorers/gpt2-large-standard dtype=fp32 device=cuda
151
+ run kind ppl mean_entropy distinct_1 distinct_2 top_token_mass eos_rows eos_total ppl_tokens t5_tokens path
152
+ sc1p0 raw_full 15.483493569064475 5.033153009064646 0.08120904800452095 0.43044354838709675 0.036167580529378525 64 64 63185 65473 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p4_s1p4_sc1p0_decode64_n64/sc1p0
153
+ sc1p0 pre_eos 17.4396064742211 5.063226024759697 0.08378514689194301 0.44411871325673397 0.037321901399571304 0 0 59110 63448 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p4_s1p4_sc1p0_decode64_n64/sc1p0
154
+ [2026-06-11T22:43:38+00:00] done
155
+ ===== DONE \2026-06-11T22:34:52+00:00 gpu=3 m=-0.4 s=1.4 gamma=0.25 =====
156
+ ===== START \2026-06-11T22:34:52+00:00 gpu=3 m=-0.4 s=1.6 gamma=0.25 =====
157
+ checkpoint=runs/owt_t5_elftokenized_full_len1024_C1_to_1024_pow1_d768_l12_h12_gbs512_2x8gpu_50ep_lr3e3_ema0p9999_elfopt_not5_bottleneck128_unfixed_norm_stateprobadd_selfcond_ce_fast_20260609_230847/step_170000.pt
158
+ use_ema=1
159
+ step=170000
160
+ decode_steps=64
161
+ n=64 chunk_n=8 gpu=3
162
+ out_base=/e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612
163
+ [2026-06-11T22:43:38+00:00] infer step=170000 decode=64 -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p4_s1p6_sc1p0_decode64_n64
164
+ [2026-06-11T22:43:38+00:00] run decode=64 chunk=0 n=8 seed=123
165
+ [2026-06-11T22:43:48+00:00] done decode=64 chunk=0
166
+ [2026-06-11T22:43:48+00:00] run decode=64 chunk=1 n=8 seed=124
167
+ [2026-06-11T22:43:58+00:00] done decode=64 chunk=1
168
+ [2026-06-11T22:43:58+00:00] run decode=64 chunk=2 n=8 seed=125
169
+ [2026-06-11T22:44:08+00:00] done decode=64 chunk=2
170
+ [2026-06-11T22:44:08+00:00] run decode=64 chunk=3 n=8 seed=126
171
+ [2026-06-11T22:44:18+00:00] done decode=64 chunk=3
172
+ [2026-06-11T22:44:18+00:00] run decode=64 chunk=4 n=8 seed=127
173
+ [2026-06-11T22:44:27+00:00] done decode=64 chunk=4
174
+ [2026-06-11T22:44:27+00:00] run decode=64 chunk=5 n=8 seed=128
175
+ [2026-06-11T22:44:37+00:00] done decode=64 chunk=5
176
+ [2026-06-11T22:44:37+00:00] run decode=64 chunk=6 n=8 seed=129
177
+ [2026-06-11T22:44:47+00:00] done decode=64 chunk=6
178
+ [2026-06-11T22:44:47+00:00] run decode=64 chunk=7 n=8 seed=130
179
+ [2026-06-11T22:44:57+00:00] done decode=64 chunk=7
180
+ merged 64 samples -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p4_s1p6_sc1p0_decode64_n64/sc1p0/samples64.txt
181
+ loading scorer /e2e-data/evad-tech-vla/wanghan58/models/flowtext_scorers/gpt2-large-standard dtype=fp32 device=cuda
182
+ run kind ppl mean_entropy distinct_1 distinct_2 top_token_mass eos_rows eos_total ppl_tokens t5_tokens path
183
+ sc1p0 raw_full 16.66348509590687 5.050779955667837 0.07936846922431547 0.43019495377856426 0.036015559453893675 63 64 63791 65555 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p4_s1p6_sc1p0_decode64_n64/sc1p0
184
+ sc1p0 pre_eos 19.053423138624822 5.082727122518139 0.08202589129440704 0.444621400864108 0.037228590800864096 0 0 59497 63419 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p4_s1p6_sc1p0_decode64_n64/sc1p0
185
+ [2026-06-11T22:45:27+00:00] done
186
+ ===== DONE \2026-06-11T22:34:52+00:00 gpu=3 m=-0.4 s=1.6 gamma=0.25 =====
187
+ ===== START \2026-06-11T22:34:52+00:00 gpu=3 m=-0.4 s=1.8 gamma=0.25 =====
188
+ checkpoint=runs/owt_t5_elftokenized_full_len1024_C1_to_1024_pow1_d768_l12_h12_gbs512_2x8gpu_50ep_lr3e3_ema0p9999_elfopt_not5_bottleneck128_unfixed_norm_stateprobadd_selfcond_ce_fast_20260609_230847/step_170000.pt
189
+ use_ema=1
190
+ step=170000
191
+ decode_steps=64
192
+ n=64 chunk_n=8 gpu=3
193
+ out_base=/e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612
194
+ [2026-06-11T22:45:27+00:00] infer step=170000 decode=64 -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p4_s1p8_sc1p0_decode64_n64
195
+ [2026-06-11T22:45:27+00:00] run decode=64 chunk=0 n=8 seed=123
196
+ [2026-06-11T22:45:37+00:00] done decode=64 chunk=0
197
+ [2026-06-11T22:45:37+00:00] run decode=64 chunk=1 n=8 seed=124
198
+ [2026-06-11T22:45:47+00:00] done decode=64 chunk=1
199
+ [2026-06-11T22:45:47+00:00] run decode=64 chunk=2 n=8 seed=125
200
+ [2026-06-11T22:45:56+00:00] done decode=64 chunk=2
201
+ [2026-06-11T22:45:56+00:00] run decode=64 chunk=3 n=8 seed=126
202
+ [2026-06-11T22:46:06+00:00] done decode=64 chunk=3
203
+ [2026-06-11T22:46:06+00:00] run decode=64 chunk=4 n=8 seed=127
204
+ [2026-06-11T22:46:16+00:00] done decode=64 chunk=4
205
+ [2026-06-11T22:46:16+00:00] run decode=64 chunk=5 n=8 seed=128
206
+ [2026-06-11T22:46:25+00:00] done decode=64 chunk=5
207
+ [2026-06-11T22:46:25+00:00] run decode=64 chunk=6 n=8 seed=129
208
+ [2026-06-11T22:46:35+00:00] done decode=64 chunk=6
209
+ [2026-06-11T22:46:35+00:00] run decode=64 chunk=7 n=8 seed=130
210
+ [2026-06-11T22:46:45+00:00] done decode=64 chunk=7
211
+ merged 64 samples -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p4_s1p8_sc1p0_decode64_n64/sc1p0/samples64.txt
212
+ loading scorer /e2e-data/evad-tech-vla/wanghan58/models/flowtext_scorers/gpt2-large-standard dtype=fp32 device=cuda
213
+ run kind ppl mean_entropy distinct_1 distinct_2 top_token_mass eos_rows eos_total ppl_tokens t5_tokens path
214
+ sc1p0 raw_full 17.12205216748313 5.048381543880808 0.08074211955692533 0.43983155582002104 0.0353818925269293 63 63 63840 65542 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p4_s1p8_sc1p0_decode64_n64/sc1p0
215
+ sc1p0 pre_eos 19.704070943129636 5.0846337107330655 0.08349244922756466 0.454836673504813 0.0365940256584242 0 0 59468 63371 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p4_s1p8_sc1p0_decode64_n64/sc1p0
216
+ [2026-06-11T22:47:14+00:00] done
217
+ ===== DONE \2026-06-11T22:34:52+00:00 gpu=3 m=-0.4 s=1.8 gamma=0.25 =====
218
+ ===== START \2026-06-11T22:34:52+00:00 gpu=3 m=-0.4 s=2.0 gamma=0.25 =====
219
+ checkpoint=runs/owt_t5_elftokenized_full_len1024_C1_to_1024_pow1_d768_l12_h12_gbs512_2x8gpu_50ep_lr3e3_ema0p9999_elfopt_not5_bottleneck128_unfixed_norm_stateprobadd_selfcond_ce_fast_20260609_230847/step_170000.pt
220
+ use_ema=1
221
+ step=170000
222
+ decode_steps=64
223
+ n=64 chunk_n=8 gpu=3
224
+ out_base=/e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612
225
+ [2026-06-11T22:47:14+00:00] infer step=170000 decode=64 -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p4_s2p0_sc1p0_decode64_n64
226
+ [2026-06-11T22:47:14+00:00] run decode=64 chunk=0 n=8 seed=123
227
+ [2026-06-11T22:47:24+00:00] done decode=64 chunk=0
228
+ [2026-06-11T22:47:24+00:00] run decode=64 chunk=1 n=8 seed=124
229
+ [2026-06-11T22:47:34+00:00] done decode=64 chunk=1
230
+ [2026-06-11T22:47:34+00:00] run decode=64 chunk=2 n=8 seed=125
231
+ [2026-06-11T22:47:44+00:00] done decode=64 chunk=2
232
+ [2026-06-11T22:47:44+00:00] run decode=64 chunk=3 n=8 seed=126
233
+ [2026-06-11T22:47:53+00:00] done decode=64 chunk=3
234
+ [2026-06-11T22:47:53+00:00] run decode=64 chunk=4 n=8 seed=127
235
+ [2026-06-11T22:48:03+00:00] done decode=64 chunk=4
236
+ [2026-06-11T22:48:03+00:00] run decode=64 chunk=5 n=8 seed=128
237
+ [2026-06-11T22:48:13+00:00] done decode=64 chunk=5
238
+ [2026-06-11T22:48:13+00:00] run decode=64 chunk=6 n=8 seed=129
239
+ [2026-06-11T22:48:23+00:00] done decode=64 chunk=6
240
+ [2026-06-11T22:48:23+00:00] run decode=64 chunk=7 n=8 seed=130
241
+ [2026-06-11T22:48:32+00:00] done decode=64 chunk=7
242
+ merged 64 samples -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p4_s2p0_sc1p0_decode64_n64/sc1p0/samples64.txt
243
+ loading scorer /e2e-data/evad-tech-vla/wanghan58/models/flowtext_scorers/gpt2-large-standard dtype=fp32 device=cuda
244
+ run kind ppl mean_entropy distinct_1 distinct_2 top_token_mass eos_rows eos_total ppl_tokens t5_tokens path
245
+ sc1p0 raw_full 19.332869239573228 5.078073205362629 0.08395001449518608 0.45138698159846197 0.03645157844947283 63 63 63394 65539 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p4_s2p0_sc1p0_decode64_n64/sc1p0
246
+ sc1p0 pre_eos 22.574710721942342 5.115753742175942 0.08684738163274972 0.46698768550678876 0.037716486951579545 0 0 58973 63341 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p4_s2p0_sc1p0_decode64_n64/sc1p0
247
+ [2026-06-11T22:49:02+00:00] done
248
+ ===== DONE \2026-06-11T22:34:52+00:00 gpu=3 m=-0.4 s=2.0 gamma=0.25 =====
249
+ ===== START \2026-06-11T22:34:52+00:00 gpu=3 m=-0.3 s=1.4 gamma=0.25 =====
250
+ checkpoint=runs/owt_t5_elftokenized_full_len1024_C1_to_1024_pow1_d768_l12_h12_gbs512_2x8gpu_50ep_lr3e3_ema0p9999_elfopt_not5_bottleneck128_unfixed_norm_stateprobadd_selfcond_ce_fast_20260609_230847/step_170000.pt
251
+ use_ema=1
252
+ step=170000
253
+ decode_steps=64
254
+ n=64 chunk_n=8 gpu=3
255
+ out_base=/e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612
256
+ [2026-06-11T22:49:02+00:00] infer step=170000 decode=64 -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p3_s1p4_sc1p0_decode64_n64
257
+ [2026-06-11T22:49:02+00:00] run decode=64 chunk=0 n=8 seed=123
258
+ [2026-06-11T22:49:11+00:00] done decode=64 chunk=0
259
+ [2026-06-11T22:49:11+00:00] run decode=64 chunk=1 n=8 seed=124
260
+ [2026-06-11T22:49:21+00:00] done decode=64 chunk=1
261
+ [2026-06-11T22:49:21+00:00] run decode=64 chunk=2 n=8 seed=125
262
+ [2026-06-11T22:49:31+00:00] done decode=64 chunk=2
263
+ [2026-06-11T22:49:31+00:00] run decode=64 chunk=3 n=8 seed=126
264
+ [2026-06-11T22:49:41+00:00] done decode=64 chunk=3
265
+ [2026-06-11T22:49:41+00:00] run decode=64 chunk=4 n=8 seed=127
266
+ [2026-06-11T22:49:50+00:00] done decode=64 chunk=4
267
+ [2026-06-11T22:49:50+00:00] run decode=64 chunk=5 n=8 seed=128
268
+ [2026-06-11T22:50:00+00:00] done decode=64 chunk=5
269
+ [2026-06-11T22:50:00+00:00] run decode=64 chunk=6 n=8 seed=129
270
+ [2026-06-11T22:50:10+00:00] done decode=64 chunk=6
271
+ [2026-06-11T22:50:10+00:00] run decode=64 chunk=7 n=8 seed=130
272
+ [2026-06-11T22:50:20+00:00] done decode=64 chunk=7
273
+ merged 64 samples -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p3_s1p4_sc1p0_decode64_n64/sc1p0/samples64.txt
274
+ loading scorer /e2e-data/evad-tech-vla/wanghan58/models/flowtext_scorers/gpt2-large-standard dtype=fp32 device=cuda
275
+ run kind ppl mean_entropy distinct_1 distinct_2 top_token_mass eos_rows eos_total ppl_tokens t5_tokens path
276
+ sc1p0 raw_full 16.928129598265656 5.071954158306068 0.0861945144778058 0.4528299003728378 0.03430361372144549 62 62 62556 65445 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p3_s1p4_sc1p0_decode64_n64/sc1p0
277
+ sc1p0 pre_eos 19.28985262070039 5.103880944521719 0.08894075347326258 0.46729325679682077 0.03540283538075789 0 0 58467 63413 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p3_s1p4_sc1p0_decode64_n64/sc1p0
278
+ [2026-06-11T22:50:49+00:00] done
279
+ ===== DONE \2026-06-11T22:34:52+00:00 gpu=3 m=-0.3 s=1.4 gamma=0.25 =====
280
+ ===== START \2026-06-11T22:34:52+00:00 gpu=3 m=-0.3 s=1.6 gamma=0.25 =====
281
+ checkpoint=runs/owt_t5_elftokenized_full_len1024_C1_to_1024_pow1_d768_l12_h12_gbs512_2x8gpu_50ep_lr3e3_ema0p9999_elfopt_not5_bottleneck128_unfixed_norm_stateprobadd_selfcond_ce_fast_20260609_230847/step_170000.pt
282
+ use_ema=1
283
+ step=170000
284
+ decode_steps=64
285
+ n=64 chunk_n=8 gpu=3
286
+ out_base=/e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612
287
+ [2026-06-11T22:50:49+00:00] infer step=170000 decode=64 -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p3_s1p6_sc1p0_decode64_n64
288
+ [2026-06-11T22:50:49+00:00] run decode=64 chunk=0 n=8 seed=123
289
+ [2026-06-11T22:50:59+00:00] done decode=64 chunk=0
290
+ [2026-06-11T22:50:59+00:00] run decode=64 chunk=1 n=8 seed=124
291
+ [2026-06-11T22:51:08+00:00] done decode=64 chunk=1
292
+ [2026-06-11T22:51:08+00:00] run decode=64 chunk=2 n=8 seed=125
293
+ [2026-06-11T22:51:18+00:00] done decode=64 chunk=2
294
+ [2026-06-11T22:51:18+00:00] run decode=64 chunk=3 n=8 seed=126
295
+ [2026-06-11T22:51:28+00:00] done decode=64 chunk=3
296
+ [2026-06-11T22:51:28+00:00] run decode=64 chunk=4 n=8 seed=127
297
+ [2026-06-11T22:51:37+00:00] done decode=64 chunk=4
298
+ [2026-06-11T22:51:37+00:00] run decode=64 chunk=5 n=8 seed=128
299
+ [2026-06-11T22:51:47+00:00] done decode=64 chunk=5
300
+ [2026-06-11T22:51:47+00:00] run decode=64 chunk=6 n=8 seed=129
301
+ [2026-06-11T22:51:57+00:00] done decode=64 chunk=6
302
+ [2026-06-11T22:51:57+00:00] run decode=64 chunk=7 n=8 seed=130
303
+ [2026-06-11T22:52:07+00:00] done decode=64 chunk=7
304
+ merged 64 samples -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p3_s1p6_sc1p0_decode64_n64/sc1p0/samples64.txt
305
+ loading scorer /e2e-data/evad-tech-vla/wanghan58/models/flowtext_scorers/gpt2-large-standard dtype=fp32 device=cuda
306
+ run kind ppl mean_entropy distinct_1 distinct_2 top_token_mass eos_rows eos_total ppl_tokens t5_tokens path
307
+ sc1p0 raw_full 17.3935578124969 5.063727260360727 0.08130180585873696 0.43846553092752033 0.03590346364621655 64 64 63426 65509 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p3_s1p6_sc1p0_decode64_n64/sc1p0
308
+ sc1p0 pre_eos 19.776817910921572 5.090824696528666 0.08390980287105466 0.45255278915852504 0.03706213264839823 0 0 59306 63461 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p3_s1p6_sc1p0_decode64_n64/sc1p0
309
+ [2026-06-11T22:52:21+00:00] done
310
+ ===== DONE \2026-06-11T22:34:52+00:00 gpu=3 m=-0.3 s=1.6 gamma=0.25 =====
311
+ ===== START \2026-06-11T22:34:52+00:00 gpu=3 m=-0.3 s=1.8 gamma=0.25 =====
312
+ checkpoint=runs/owt_t5_elftokenized_full_len1024_C1_to_1024_pow1_d768_l12_h12_gbs512_2x8gpu_50ep_lr3e3_ema0p9999_elfopt_not5_bottleneck128_unfixed_norm_stateprobadd_selfcond_ce_fast_20260609_230847/step_170000.pt
313
+ use_ema=1
314
+ step=170000
315
+ decode_steps=64
316
+ n=64 chunk_n=8 gpu=3
317
+ out_base=/e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612
318
+ [2026-06-11T22:52:21+00:00] infer step=170000 decode=64 -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p3_s1p8_sc1p0_decode64_n64
319
+ [2026-06-11T22:52:21+00:00] run decode=64 chunk=0 n=8 seed=123
320
+ [2026-06-11T22:52:31+00:00] done decode=64 chunk=0
321
+ [2026-06-11T22:52:31+00:00] run decode=64 chunk=1 n=8 seed=124
322
+ [2026-06-11T22:52:40+00:00] done decode=64 chunk=1
323
+ [2026-06-11T22:52:40+00:00] run decode=64 chunk=2 n=8 seed=125
324
+ [2026-06-11T22:52:50+00:00] done decode=64 chunk=2
325
+ [2026-06-11T22:52:50+00:00] run decode=64 chunk=3 n=8 seed=126
326
+ [2026-06-11T22:53:00+00:00] done decode=64 chunk=3
327
+ [2026-06-11T22:53:00+00:00] run decode=64 chunk=4 n=8 seed=127
328
+ [2026-06-11T22:53:09+00:00] done decode=64 chunk=4
329
+ [2026-06-11T22:53:09+00:00] run decode=64 chunk=5 n=8 seed=128
330
+ [2026-06-11T22:53:19+00:00] done decode=64 chunk=5
331
+ [2026-06-11T22:53:19+00:00] run decode=64 chunk=6 n=8 seed=129
332
+ [2026-06-11T22:53:29+00:00] done decode=64 chunk=6
333
+ [2026-06-11T22:53:29+00:00] run decode=64 chunk=7 n=8 seed=130
334
+ [2026-06-11T22:53:39+00:00] done decode=64 chunk=7
335
+ merged 64 samples -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p3_s1p8_sc1p0_decode64_n64/sc1p0/samples64.txt
336
+ loading scorer /e2e-data/evad-tech-vla/wanghan58/models/flowtext_scorers/gpt2-large-standard dtype=fp32 device=cuda
337
+ run kind ppl mean_entropy distinct_1 distinct_2 top_token_mass eos_rows eos_total ppl_tokens t5_tokens path
338
+ sc1p0 raw_full 17.7391049377987 5.071205544786745 0.08228292385167099 0.4488547055502144 0.035754616206317716 64 64 63672 65530 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p3_s1p8_sc1p0_decode64_n64/sc1p0
339
+ sc1p0 pre_eos 20.245708500544218 5.104042020201639 0.08494044242768009 0.4633746671498574 0.03691624125543581 0 0 59522 63468 /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p3_s1p8_sc1p0_decode64_n64/sc1p0
340
+ [2026-06-11T22:53:53+00:00] done
341
+ ===== DONE \2026-06-11T22:34:52+00:00 gpu=3 m=-0.3 s=1.8 gamma=0.25 =====
342
+ ===== START \2026-06-11T22:34:52+00:00 gpu=3 m=-0.3 s=2.0 gamma=0.25 =====
343
+ checkpoint=runs/owt_t5_elftokenized_full_len1024_C1_to_1024_pow1_d768_l12_h12_gbs512_2x8gpu_50ep_lr3e3_ema0p9999_elfopt_not5_bottleneck128_unfixed_norm_stateprobadd_selfcond_ce_fast_20260609_230847/step_170000.pt
344
+ use_ema=1
345
+ step=170000
346
+ decode_steps=64
347
+ n=64 chunk_n=8 gpu=3
348
+ out_base=/e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612
349
+ [2026-06-11T22:53:53+00:00] infer step=170000 decode=64 -> /e2e-data/evad-tech-vla/wanghan58/workspace/LTA_openwebtext_dualt/docs/lta_samples/metrics_20260612/owt_t5_not5_bottleneck128_norm_stateprobadd_selfcond_ce_fast_lr3e3_ema0p9999_logitnorm64_focused_ppl24_ent515_step170000_ema_dgamma0p25_tschedlogit_normal_mn0p3_s2p0_sc1p0_decode64_n64
350
+ [2026-06-11T22:53:53+00:00] run decode=64 chunk=0 n=8 seed=123
351
+ [2026-06-11T22:54:03+00:00] done decode=64 chunk=0
352
+ [2026-06-11T22:54:03+00:00] run decode=64 chunk=1 n=8 seed=124