From e236d9edc5904c1a86fb0714a2a3613e32dbba89 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 2 Sep 2024 21:46:07 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- docs/benchmarks.ipynb | 9 ++++----- docs/tutorials/derivative.ipynb | 14 +++++++------- docs/tutorials/geometry.ipynb | 7 ++++--- docs/tutorials/intro.ipynb | 2 +- docs/tutorials/kernels.ipynb | 5 +++-- docs/tutorials/likelihoods.ipynb | 5 +++-- docs/tutorials/means.ipynb | 7 ++++--- docs/tutorials/mixture.ipynb | 3 +-- docs/tutorials/modeling.ipynb | 10 ++++------ docs/tutorials/multivariate.ipynb | 6 ++++-- docs/tutorials/quasisep-custom.ipynb | 2 +- docs/tutorials/quasisep.ipynb | 4 ++-- docs/tutorials/quickstart.ipynb | 5 ++--- docs/tutorials/transforms.ipynb | 9 +++++---- 14 files changed, 45 insertions(+), 43 deletions(-) diff --git a/docs/benchmarks.ipynb b/docs/benchmarks.ipynb index 6908583f..a1b0f8a4 100644 --- a/docs/benchmarks.ipynb +++ b/docs/benchmarks.ipynb @@ -118,13 +118,12 @@ "source": [ "from functools import partial\n", "\n", - "import numpy as np\n", - "import matplotlib.pyplot as plt\n", - "\n", + "import celerite2\n", + "import george\n", "import jax\n", + "import matplotlib.pyplot as plt\n", + "import numpy as np\n", "\n", - "import george\n", - "import celerite2\n", "import tinygp\n", "\n", "jax.config.update(\"jax_enable_x64\", True)\n", diff --git a/docs/tutorials/derivative.ipynb b/docs/tutorials/derivative.ipynb index ca9ec089..b408543a 100644 --- a/docs/tutorials/derivative.ipynb +++ b/docs/tutorials/derivative.ipynb @@ -60,9 +60,8 @@ "metadata": {}, "outputs": [], "source": [ - "import numpy as np\n", "import matplotlib.pyplot as plt\n", - "\n", + "import numpy as np\n", "\n", "X = np.linspace(0.0, 5 * np.pi, 50)\n", "y = np.concatenate(\n", @@ -97,10 +96,11 @@ "metadata": {}, "outputs": [], "source": [ - "import tinygp\n", "import jax\n", "import jax.numpy as jnp\n", "\n", + "import tinygp\n", + "\n", "jax.config.update(\"jax_enable_x64\", True)\n", "\n", "\n", @@ -169,10 +169,10 @@ " )[0]\n", "\n", " plt.figure()\n", - " plt.plot(dt, k00, label=\"$\\mathrm{cov}(f,\\,f)$\", lw=1)\n", - " plt.plot(dt, k01, label=\"$\\mathrm{cov}(f,\\,\\dot{f})$\", lw=1)\n", - " plt.plot(dt, k10, label=\"$\\mathrm{cov}(\\dot{f},\\,f)$\", lw=1)\n", - " plt.plot(dt, k11, label=\"$\\mathrm{cov}(\\dot{f},\\,\\dot{f})$\", lw=1)\n", + " plt.plot(dt, k00, label=r\"$\\mathrm{cov}(f,\\,f)$\", lw=1)\n", + " plt.plot(dt, k01, label=r\"$\\mathrm{cov}(f,\\,\\dot{f})$\", lw=1)\n", + " plt.plot(dt, k10, label=r\"$\\mathrm{cov}(\\dot{f},\\,f)$\", lw=1)\n", + " plt.plot(dt, k11, label=r\"$\\mathrm{cov}(\\dot{f},\\,\\dot{f})$\", lw=1)\n", " plt.legend()\n", " plt.xlabel(r\"$\\Delta t$\")\n", " plt.xlim(dt.min(), dt.max())\n", diff --git a/docs/tutorials/geometry.ipynb b/docs/tutorials/geometry.ipynb index b155db2f..0f581032 100644 --- a/docs/tutorials/geometry.ipynb +++ b/docs/tutorials/geometry.ipynb @@ -47,11 +47,12 @@ "metadata": {}, "outputs": [], "source": [ - "import numpy as np\n", - "import matplotlib.pyplot as plt\n", "import jax\n", "import jax.numpy as jnp\n", - "from tinygp import kernels, GaussianProcess\n", + "import matplotlib.pyplot as plt\n", + "import numpy as np\n", + "\n", + "from tinygp import GaussianProcess, kernels\n", "\n", "jax.config.update(\"jax_enable_x64\", True)\n", "\n", diff --git a/docs/tutorials/intro.ipynb b/docs/tutorials/intro.ipynb index 0ad9d41c..a0383c27 100644 --- a/docs/tutorials/intro.ipynb +++ b/docs/tutorials/intro.ipynb @@ -82,8 +82,8 @@ "metadata": {}, "outputs": [], "source": [ - "import numpy as np\n", "import matplotlib.pyplot as plt\n", + "import numpy as np\n", "\n", "\n", "def plot_kernel(kernel, **kwargs):\n", diff --git a/docs/tutorials/kernels.ipynb b/docs/tutorials/kernels.ipynb index 9f3f263b..e706c1cb 100644 --- a/docs/tutorials/kernels.ipynb +++ b/docs/tutorials/kernels.ipynb @@ -48,10 +48,11 @@ "metadata": {}, "outputs": [], "source": [ - "import tinygp\n", "import jax\n", "import jax.numpy as jnp\n", "\n", + "import tinygp\n", + "\n", "\n", "class SpectralMixture(tinygp.kernels.Kernel):\n", " weight: jax.Array\n", @@ -85,8 +86,8 @@ "metadata": {}, "outputs": [], "source": [ - "import numpy as np\n", "import matplotlib.pyplot as plt\n", + "import numpy as np\n", "\n", "\n", "def build_gp(theta):\n", diff --git a/docs/tutorials/likelihoods.ipynb b/docs/tutorials/likelihoods.ipynb index dd2ff088..a46c2f60 100644 --- a/docs/tutorials/likelihoods.ipynb +++ b/docs/tutorials/likelihoods.ipynb @@ -54,8 +54,8 @@ "metadata": {}, "outputs": [], "source": [ - "import numpy as np\n", "import matplotlib.pyplot as plt\n", + "import numpy as np\n", "\n", "random = np.random.default_rng(203618)\n", "x = np.linspace(-3, 3, 20)\n", @@ -90,7 +90,8 @@ "import jax.numpy as jnp\n", "import numpyro\n", "import numpyro.distributions as dist\n", - "from tinygp import kernels, GaussianProcess\n", + "\n", + "from tinygp import GaussianProcess, kernels\n", "\n", "jax.config.update(\"jax_enable_x64\", True)\n", "\n", diff --git a/docs/tutorials/means.ipynb b/docs/tutorials/means.ipynb index 8176a00f..cd3808ef 100644 --- a/docs/tutorials/means.ipynb +++ b/docs/tutorials/means.ipynb @@ -52,10 +52,11 @@ "outputs": [], "source": [ "from functools import partial\n", - "import numpy as np\n", - "import matplotlib.pyplot as plt\n", + "\n", "import jax\n", "import jax.numpy as jnp\n", + "import matplotlib.pyplot as plt\n", + "import numpy as np\n", "\n", "jax.config.update(\"jax_enable_x64\", True)\n", "\n", @@ -172,7 +173,7 @@ "metadata": {}, "outputs": [], "source": [ - "from tinygp import kernels, GaussianProcess\n", + "from tinygp import GaussianProcess, kernels\n", "\n", "\n", "def build_gp(params):\n", diff --git a/docs/tutorials/mixture.ipynb b/docs/tutorials/mixture.ipynb index 0090ba82..73ccdf90 100644 --- a/docs/tutorials/mixture.ipynb +++ b/docs/tutorials/mixture.ipynb @@ -66,9 +66,8 @@ "source": [ "import jax\n", "import jax.numpy as jnp\n", - "\n", - "import numpy as np\n", "import matplotlib.pyplot as plt\n", + "import numpy as np\n", "\n", "from tinygp import GaussianProcess, kernels, transforms\n", "\n", diff --git a/docs/tutorials/modeling.ipynb b/docs/tutorials/modeling.ipynb index ba983ac8..71a7f7c3 100644 --- a/docs/tutorials/modeling.ipynb +++ b/docs/tutorials/modeling.ipynb @@ -67,8 +67,8 @@ "metadata": {}, "outputs": [], "source": [ - "import numpy as np\n", "import matplotlib.pyplot as plt\n", + "import numpy as np\n", "\n", "random = np.random.default_rng(42)\n", "\n", @@ -120,15 +120,13 @@ "metadata": {}, "outputs": [], "source": [ - "from tinygp import kernels, GaussianProcess\n", - "\n", + "import flax.linen as nn\n", "import jax\n", "import jax.numpy as jnp\n", - "\n", - "import flax.linen as nn\n", + "import optax\n", "from flax.linen.initializers import zeros\n", "\n", - "import optax\n", + "from tinygp import GaussianProcess, kernels\n", "\n", "\n", "class GPModule(nn.Module):\n", diff --git a/docs/tutorials/multivariate.ipynb b/docs/tutorials/multivariate.ipynb index 6009611d..9c15c75e 100644 --- a/docs/tutorials/multivariate.ipynb +++ b/docs/tutorials/multivariate.ipynb @@ -74,8 +74,9 @@ "outputs": [], "source": [ "import jax\n", - "import numpy as np\n", "import jax.numpy as jnp\n", + "import numpy as np\n", + "\n", "from tinygp import kernels\n", "\n", "jax.config.update(\"jax_enable_x64\", True)\n", @@ -117,8 +118,8 @@ "metadata": {}, "outputs": [], "source": [ - "import numpy as np\n", "import matplotlib.pyplot as plt\n", + "import numpy as np\n", "\n", "random = np.random.default_rng(48392)\n", "X = random.uniform(-5, 5, (100, 2))\n", @@ -163,6 +164,7 @@ "outputs": [], "source": [ "import jaxopt\n", + "\n", "from tinygp import GaussianProcess, kernels, transforms\n", "\n", "\n", diff --git a/docs/tutorials/quasisep-custom.ipynb b/docs/tutorials/quasisep-custom.ipynb index d1e7629b..465cd3e2 100644 --- a/docs/tutorials/quasisep-custom.ipynb +++ b/docs/tutorials/quasisep-custom.ipynb @@ -131,8 +131,8 @@ "metadata": {}, "outputs": [], "source": [ - "import numpy as np\n", "import matplotlib.pyplot as plt\n", + "import numpy as np\n", "\n", "random = np.random.default_rng(394)\n", "t = np.sort(random.uniform(0, 10, 700))\n", diff --git a/docs/tutorials/quasisep.ipynb b/docs/tutorials/quasisep.ipynb index 80283176..73fbaa43 100644 --- a/docs/tutorials/quasisep.ipynb +++ b/docs/tutorials/quasisep.ipynb @@ -57,8 +57,8 @@ "metadata": {}, "outputs": [], "source": [ - "import numpy as np\n", "import matplotlib.pyplot as plt\n", + "import numpy as np\n", "\n", "random = np.random.default_rng(42)\n", "\n", @@ -108,7 +108,7 @@ "import jax\n", "import jax.numpy as jnp\n", "\n", - "from tinygp import kernels, GaussianProcess\n", + "from tinygp import GaussianProcess, kernels\n", "\n", "jax.config.update(\"jax_enable_x64\", True)\n", "\n", diff --git a/docs/tutorials/quickstart.ipynb b/docs/tutorials/quickstart.ipynb index a35671d4..244f4f53 100644 --- a/docs/tutorials/quickstart.ipynb +++ b/docs/tutorials/quickstart.ipynb @@ -48,8 +48,8 @@ "metadata": {}, "outputs": [], "source": [ - "import numpy as np\n", "import matplotlib.pyplot as plt\n", + "import numpy as np\n", "from statsmodels.datasets import co2\n", "\n", "data = co2.load_pandas().data\n", @@ -102,8 +102,7 @@ "import jax\n", "import jax.numpy as jnp\n", "\n", - "from tinygp import kernels, GaussianProcess\n", - "\n", + "from tinygp import GaussianProcess, kernels\n", "\n", "jax.config.update(\"jax_enable_x64\", True)\n", "\n", diff --git a/docs/tutorials/transforms.ipynb b/docs/tutorials/transforms.ipynb index 220329a0..6942d97f 100644 --- a/docs/tutorials/transforms.ipynb +++ b/docs/tutorials/transforms.ipynb @@ -61,8 +61,8 @@ "metadata": {}, "outputs": [], "source": [ - "import numpy as np\n", "import matplotlib.pyplot as plt\n", + "import numpy as np\n", "\n", "random = np.random.default_rng(567)\n", "\n", @@ -99,12 +99,13 @@ "metadata": {}, "outputs": [], "source": [ + "import flax.linen as nn\n", "import jax\n", - "import optax\n", "import jax.numpy as jnp\n", - "import flax.linen as nn\n", + "import optax\n", "from flax.linen.initializers import zeros\n", - "from tinygp import kernels, transforms, GaussianProcess" + "\n", + "from tinygp import GaussianProcess, kernels, transforms" ] }, {