Split sampling.py into sampling.py and sampling_forward.py (#6257)

Co-authored-by: Ricardo Vieira <[email protected]>

Author: michaelosthege

.github/workflows/tests.yml

@@ -59,6 +59,8 @@ jobs:
             pymc/tests/distributions/test_censored.py
             pymc/tests/distributions/test_simulator.py
             pymc/tests/distributions/test_truncated.py
+            pymc/tests/test_sampling_forward.py
+            pymc/tests/stats/test_convergence.py
 
           - |
             pymc/tests/tuning/test_scaling.py

pymc/__init__.py

@@ -68,6 +68,7 @@ def __set_compiler_flags():
 from pymc.plots import *
 from pymc.printing import *
 from pymc.sampling import *
+from pymc.sampling_forward import *
 from pymc.smc import *
 from pymc.stats import *
 from pymc.step_methods import *

pymc/parallel_sampling.py

@@ -21,7 +21,7 @@
 import traceback
 
 from collections import namedtuple
-from typing import TYPE_CHECKING, Dict, List, Sequence
+from typing import Dict, List, Sequence
 
 import cloudpickle
 import numpy as np
@@ -31,10 +31,7 @@
 from pymc import aesaraf
 from pymc.blocking import DictToArrayBijection
 from pymc.exceptions import SamplingError
-
-# Avoid circular import
-if TYPE_CHECKING:
-    from pymc.sampling import RandomSeed
+from pymc.util import RandomSeed
 
 logger = logging.getLogger("pymc")