Switching from list comprehension to np.fromiter converts this to an nd.array and resolves mypy error

Ok3ks · Ok3ks · commit 8838710cae09 · 2025-07-17T07:58:54.000+01:00
diff --git a/pymc/sampling/mcmc.py b/pymc/sampling/mcmc.py
@@ -760,7 +760,7 @@ def joined_blas_limiter():
             UserWarning,
         )
     rngs = get_random_generator(random_seed).spawn(chains)
-    random_seed_list = [rng.integers(2**30) for rng in rngs]
+    random_seed_list = np.fromiter((rng.integers(2**30) for rng in rngs), dtype=np.int64)
 
     if not discard_tuned_samples and not return_inferencedata and not isinstance(trace, ZarrTrace):
         warnings.warn(
diff --git a/scripts/run_mypy.py b/scripts/run_mypy.py
@@ -44,7 +44,6 @@
 pymc/model/transform/conditioning.py
 pymc/pytensorf.py
 pymc/sampling/jax.py
-pymc/sampling/mcmc.py
 """
 
 

Original file line number	Diff line number	Diff line change
`@@ -760,7 +760,7 @@ def joined_blas_limiter():`
`760`	`760`	`UserWarning,`
`761`	`761`	`)`
`762`	`762`	`rngs = get_random_generator(random_seed).spawn(chains)`
`763`		`- random_seed_list = [rng.integers(2**30) for rng in rngs]`
	`763`	`+ random_seed_list = np.fromiter((rng.integers(2**30) for rng in rngs), dtype=np.int64)`
`764`	`764`
`765`	`765`	`if not discard_tuned_samples and not return_inferencedata and not isinstance(trace, ZarrTrace):`
`766`	`766`	`warnings.warn(`