Add tile_min_size_floating.ipynb

Intermediate test is simplified, because now phi-center covers only half of the face and we do not expect m- and p- directions to be equal
lincc-frameworks · Sep 11, 2024 · 22d47a7 · 22d47a7
1 parent 0e0d068
commit 22d47a7
Show file tree

Hide file tree

Showing 4 changed files with 421 additions and 22 deletions.
diff --git a/docs/notebooks/tile_min_size_1by1.ipynb b/docs/notebooks/tile_min_size_1by1.ipynb
@@ -14,8 +14,8 @@
    "id": "f4bb6871af102d56",
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2024-07-17T21:26:11.700772Z",
-     "start_time": "2024-07-17T21:26:11.697782Z"
+     "end_time": "2024-09-11T14:45:32.084217Z",
+     "start_time": "2024-09-11T14:45:32.082Z"
     }
    },
    "outputs": [],
@@ -37,27 +37,23 @@
    "id": "57fa7fa66ab3d5c0",
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2024-07-17T21:26:12.634248Z",
-     "start_time": "2024-07-17T21:26:11.701790Z"
+     "end_time": "2024-09-11T14:45:32.090478Z",
+     "start_time": "2024-09-11T14:45:32.088273Z"
     }
    },
    "outputs": [],
    "source": [
-    "import os\n",
-    "from functools import partial\n",
-    "\n",
     "import jax\n",
     "import jax.numpy as jnp\n",
     "import matplotlib.pyplot as plt\n",
     "import optax\n",
-    "from numpyro.infer import MCMC, NUTS\n",
     "\n",
     "from healpix_geometry_analysis.coordinates import HealpixCoordinates\n",
     "from healpix_geometry_analysis.geometry.tile import TileGeometry\n",
-    "from healpix_geometry_analysis.problems.numpyro_sampler import NumpyroSamplerProblem\n",
     "from healpix_geometry_analysis.problems.optax_optimizer import OptaxOptimizerProblem\n",
     "from healpix_geometry_analysis.enable_x64 import enable_x64\n",
     "\n",
+    "# Use float64 with Jax\n",
     "enable_x64()"
    ]
   },
@@ -75,8 +71,8 @@
    "id": "3bc13fcf4eb9c747",
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2024-07-17T21:26:12.655671Z",
-     "start_time": "2024-07-17T21:26:12.634963Z"
+     "end_time": "2024-09-11T14:45:32.098947Z",
+     "start_time": "2024-09-11T14:45:32.097078Z"
     }
    },
    "outputs": [],
@@ -98,8 +94,8 @@
    "id": "69b3011e7a38f3b9",
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2024-07-17T21:26:13.074963Z",
-     "start_time": "2024-07-17T21:26:12.656748Z"
+     "end_time": "2024-09-11T14:45:32.108669Z",
+     "start_time": "2024-09-11T14:45:32.104440Z"
     }
    },
    "outputs": [],
@@ -114,8 +110,8 @@
    "id": "c859164b16234847",
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2024-07-17T21:26:13.414059Z",
-     "start_time": "2024-07-17T21:26:13.075581Z"
+     "end_time": "2024-09-11T14:45:32.302155Z",
+     "start_time": "2024-09-11T14:45:32.187751Z"
     }
    },
    "outputs": [],
@@ -143,8 +139,8 @@
    "id": "22d4a055b369364c",
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2024-07-17T21:26:13.416602Z",
-     "start_time": "2024-07-17T21:26:13.414815Z"
+     "end_time": "2024-09-11T14:45:32.310509Z",
+     "start_time": "2024-09-11T14:45:32.308439Z"
     }
    },
    "outputs": [],
@@ -197,8 +193,8 @@
    "id": "8a340519783e6a84",
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2024-07-17T21:26:21.587926Z",
-     "start_time": "2024-07-17T21:26:13.417116Z"
+     "end_time": "2024-09-11T14:45:40.133132Z",
+     "start_time": "2024-09-11T14:45:32.316469Z"
     }
    },
    "outputs": [],