diff --git a/examples/05_matrix_operations.ipynb b/examples/05_matrix_operations.ipynb index 926d1734..53963e29 100644 --- a/examples/05_matrix_operations.ipynb +++ b/examples/05_matrix_operations.ipynb @@ -6,6 +6,7 @@ "metadata": {}, "outputs": [], "source": [ + "from packaging import version\n", "import pyerrors as pe\n", "import numpy as np\n", "import scipy" @@ -343,15 +344,16 @@ } ], "source": [ - "e, v = pe.linalg.eigh(matrix)\n", - "for (i), entry in np.ndenumerate(e):\n", - " entry.gamma_method()\n", - "print('Eigenvalues:')\n", - "print(e)\n", - "for (i, j), entry in np.ndenumerate(v):\n", - " entry.gamma_method()\n", - "print('Eigenvectors:')\n", - "print(v)" + "if version.parse(np.__version__) >= version.parse("1.25.0"):\n", + " e, v = pe.linalg.eigh(matrix)\n", + " for (i), entry in np.ndenumerate(e):\n", + " entry.gamma_method()\n", + " print('Eigenvalues:')\n", + " print(e)\n", + " for (i, j), entry in np.ndenumerate(v):\n", + " entry.gamma_method()\n", + " print('Eigenvectors:')\n", + " print(v)" ] }, { @@ -378,9 +380,10 @@ } ], "source": [ - "for i in range(2):\n", - " print('Check eigenvector', i + 1)\n", - " print(matrix @ v[:, i] - v[:, i] * e[i])" + "if version.parse(np.__version__) >= version.parse("1.25.0"):\n", + " for i in range(2):\n", + " print('Check eigenvector', i + 1)\n", + " print(matrix @ v[:, i] - v[:, i] * e[i])" ] }, {