Fix Model circular import using TYPE_CHECKING and lazy import

Author: aman-coder03

pymc/printing.py

@@ -14,8 +14,8 @@
 
 
 import re
-
 from functools import partial
+from typing import TYPE_CHECKING
 
 from pytensor.compile import SharedVariable
 from pytensor.graph.basic import Constant
@@ -26,7 +26,9 @@
 from pytensor.tensor.random.type import RandomType
 from pytensor.tensor.type_other import NoneTypeT
 
-from pymc.model import Model
+if TYPE_CHECKING:
+    from pymc.model import Model
+
 
 __all__ = [
     "str_for_dist",
@@ -301,6 +303,7 @@ def _default_repr_pretty(obj: TensorVariable | Model, p, cycle):
 try:
     # register our custom pretty printer in ipython shells
     import IPython
+    from pymc.model.core import Model
 
     IPython.lib.pretty.for_type(TensorVariable, _default_repr_pretty)
     IPython.lib.pretty.for_type(Model, _default_repr_pretty)