fix the ipython_display side effect

mrmustard/lab_dev/circuit_components.py

@@ -28,6 +28,7 @@
 import numpy as np
 from numpy.typing import ArrayLike
 import ipywidgets as widgets
+from IPython import get_ipython, InteractiveShell
 from IPython.display import display
 
 from mrmustard import settings, math, widgets as mmwidgets
@@ -724,6 +725,9 @@ def __truediv__(self, other: Scalar) -> CircuitComponent:
         return self._from_attributes(Representation(self.ansatz / other, self.wires), self.name)
 
     def _ipython_display_(self):
+        if isinstance(get_ipython(), InteractiveShell):
+            print(self)
+            return
         # both reps might return None
         rep_fn = mmwidgets.fock if isinstance(self.ansatz, ArrayAnsatz) else mmwidgets.bargmann
         rep_widget = rep_fn(self.ansatz)

mrmustard/lab_dev/states/dm.py

@@ -22,6 +22,7 @@
 from itertools import product
 import warnings
 import numpy as np
+from IPython import get_ipython, InteractiveShell
 from IPython.display import display
 
 from mrmustard import math, settings, widgets
@@ -375,6 +376,9 @@ def quadrature_distribution(self, quad: RealVector, phi: float = 0.0) -> Complex
         return self.quadrature(quad, phi)
 
     def _ipython_display_(self):  # pragma: no cover
+        if isinstance(get_ipython(), InteractiveShell):
+            print(self)
+            return
         is_fock = isinstance(self.ansatz, ArrayAnsatz)
         display(widgets.state(self, is_ket=False, is_fock=is_fock))
 

mrmustard/lab_dev/states/ket.py

@@ -22,6 +22,7 @@
 from itertools import product
 import warnings
 import numpy as np
+from IPython import get_ipython, InteractiveShell
 from IPython.display import display
 
 from mrmustard import math, settings, widgets
@@ -335,6 +336,9 @@ def quadrature_distribution(self, quad: RealVector, phi: float = 0.0) -> Complex
         return math.abs(self.quadrature(quad, phi)) ** 2
 
     def _ipython_display_(self):  # pragma: no cover
+        if isinstance(get_ipython(), InteractiveShell):
+            print(self)
+            return
         is_fock = isinstance(self.ansatz, ArrayAnsatz)
         display(widgets.state(self, is_ket=True, is_fock=is_fock))
 

mrmustard/physics/ansatz/array_ansatz.py

@@ -24,6 +24,7 @@
 import numpy as np
 from numpy.typing import ArrayLike
 
+from IPython import get_ipython, InteractiveShell
 from IPython.display import display
 
 from mrmustard import math, widgets
@@ -207,6 +208,9 @@ def _generate_ansatz(self):
             self.array = [self._fn(**self._kwargs)]
 
     def _ipython_display_(self):
+        if isinstance(get_ipython(), InteractiveShell):
+            print(self)
+            return
         w = widgets.fock(self)
         if w is None:
             print(repr(self))

mrmustard/physics/ansatz/polyexp_ansatz.py

@@ -29,6 +29,7 @@
 from matplotlib import colors
 import matplotlib.pyplot as plt
 
+from IPython import get_ipython, InteractiveShell
 from IPython.display import display
 
 from mrmustard.utils.typing import (
@@ -568,6 +569,9 @@ def _generate_ansatz(self):
             self.c = c
 
     def _ipython_display_(self):
+        if isinstance(get_ipython(), InteractiveShell):
+            print(self)
+            return
         display(widgets.bargmann(self))
 
     def _order_batch(self):

mrmustard/physics/wires.py

@@ -18,6 +18,7 @@
 from functools import cached_property
 import numpy as np
 
+from IPython import get_ipython, InteractiveShell
 from IPython.display import display
 
 from mrmustard import widgets
@@ -546,4 +547,7 @@ def __repr__(self) -> str:
         return f"Wires{self.args}"
 
     def _ipython_display_(self):
+        if isinstance(get_ipython(), InteractiveShell):
+            print(self)
+            return
         display(widgets.wires(self))