feat: new periodic search

lgrcia · Jul 24, 2024 · 4640fbc · 4640fbc
2 parents f9087f7 + 6356b40
commit 4640fbc
Show file tree

Hide file tree

Showing 18 changed files with 359 additions and 193 deletions.
diff --git a/.vscode/settings.json b/.vscode/settings.json
@@ -4,7 +4,7 @@
     "editor.defaultFormatter": "ms-python.black-formatter",
     "editor.formatOnSave": true,
     "editor.codeActionsOnSave": {
-      "source.organizeImports": true
+      "source.organizeImports": "explicit"
     },
   },
   "python.testing.pytestArgs": [

diff --git a/docs/index.md b/docs/index.md
@@ -70,6 +70,7 @@ citation.md
 :maxdepth: 1
 :caption: Tutorials
 
+notebooks/tutorials/GP_optimization.ipynb
 notebooks/tutorials/ground_based.ipynb
 notebooks/tutorials/tess_search.ipynb
 notebooks/tutorials/exocomet.ipynb

diff --git a/docs/notebooks/combined.ipynb b/docs/notebooks/combined.ipynb
diff --git a/docs/notebooks/motivation.ipynb b/docs/notebooks/motivation.ipynb
diff --git a/docs/notebooks/multi.ipynb b/docs/notebooks/multi.ipynb
@@ -125,7 +125,7 @@
                             {
                                    "data": {
                                           "application/vnd.jupyter.widget-view+json": {
-                                                 "model_id": "238ee5574a104a0e83a212127bae0cb0",
+                                                 "model_id": "9238d5daa59846f4b61da59fa9b1d2e4",
                                                  "version_major": 2,
                                                  "version_minor": 0
                                           },
@@ -198,7 +198,7 @@
                             {
                                    "data": {
                                           "application/vnd.jupyter.widget-view+json": {
-                                                 "model_id": "c6bc551524ce4ad5bd4613f31dfc0ff1",
+                                                 "model_id": "3b9a579ec7614654a44be64ae9c89d48",
                                                  "version_major": 2,
                                                  "version_minor": 0
                                           },
@@ -346,7 +346,7 @@
                             {
                                    "data": {
                                           "application/vnd.jupyter.widget-view+json": {
-                                                 "model_id": "1641a1072c2740fbac93079acf292b00",
+                                                 "model_id": "08170c8a85be489ba96efa0a6618d5ec",
                                                  "version_major": 2,
                                                  "version_minor": 0
                                           },
@@ -380,7 +380,7 @@
               },
               {
                      "cell_type": "code",
-                     "execution_count": 10,
+                     "execution_count": 9,
                      "metadata": {
                             "tags": [
                                    "hide-input"

diff --git a/docs/notebooks/periodic.ipynb b/docs/notebooks/periodic.ipynb
diff --git a/docs/notebooks/single.ipynb b/docs/notebooks/single.ipynb
diff --git a/docs/notebooks/star.ipynb b/docs/notebooks/star.ipynb
@@ -16,7 +16,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 14,
+   "execution_count": 1,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -36,7 +36,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 16,
+   "execution_count": 2,
    "metadata": {},
    "outputs": [
     {
@@ -74,7 +74,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 18,
+   "execution_count": 3,
    "metadata": {},
    "outputs": [
     {
@@ -93,7 +93,7 @@
        "       0.9370029 , 0.95199615, 0.96747702, 0.98346971, 1.        ])"
       ]
      },
-     "execution_count": 18,
+     "execution_count": 3,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -113,7 +113,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 20,
+   "execution_count": 4,
    "metadata": {},
    "outputs": [
     {
@@ -122,7 +122,7 @@
        "0.5242080656950339"
       ]
      },
-     "execution_count": 20,
+     "execution_count": 4,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -148,7 +148,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.5"
+   "version": "3.10.14"
   }
  },
  "nbformat": 4,

diff --git a/docs/notebooks/templates.ipynb b/docs/notebooks/templates.ipynb
diff --git a/docs/notebooks/tutorials/GP_optimization.ipynb b/docs/notebooks/tutorials/GP_optimization.ipynb
diff --git a/docs/notebooks/tutorials/exocomet.ipynb b/docs/notebooks/tutorials/exocomet.ipynb
@@ -265,7 +265,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 12,
+   "execution_count": 7,
    "metadata": {},
    "outputs": [
     {
@@ -314,7 +314,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "4bf9da14b72f4c3ca5738a9519784e0f",
+       "model_id": "3b05ee65d3c944f39a6c545145c44932",
        "version_major": 2,
        "version_minor": 0
       },
@@ -389,7 +389,7 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "100%|██████████| 17458/17458 [00:27<00:00, 643.46it/s]\n"
+      "100%|██████████| 17458/17458 [00:22<00:00, 771.74it/s]\n"
      ]
     }
    ],
@@ -410,7 +410,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 13,
+   "execution_count": 11,
    "metadata": {
     "tags": [
      "hide-input"

diff --git a/docs/notebooks/tutorials/ground_based.ipynb b/docs/notebooks/tutorials/ground_based.ipynb
diff --git a/docs/notebooks/tutorials/tess_search.ipynb b/docs/notebooks/tutorials/tess_search.ipynb
@@ -38,7 +38,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 12,
+   "execution_count": 2,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -73,7 +73,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 13,
+   "execution_count": 3,
    "metadata": {
     "tags": [
      "hide-input"
@@ -122,7 +122,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 4,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -154,7 +154,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": 5,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -180,7 +180,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 6,
    "metadata": {
     "tags": [
      "hide-input"
@@ -241,13 +241,13 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": 7,
    "metadata": {},
    "outputs": [
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "92a760786a57463991b35cf850635521",
+       "model_id": "763fbcb712f44f988ec4dd9f81ef3089",
        "version_major": 2,
        "version_minor": 0
       },
@@ -281,7 +281,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": 8,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -300,7 +300,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": 9,
    "metadata": {},
    "outputs": [
     {
@@ -314,7 +314,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "0be5fc78f77c40b598a3d44c9920036e",
+       "model_id": "eaa75804cc2d42b8bf8c7aea19338d97",
        "version_major": 2,
        "version_minor": 0
       },
@@ -324,6 +324,14 @@
      },
      "metadata": {},
      "output_type": "display_data"
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "/opt/homebrew/Caskroom/miniforge/base/envs/nuance/lib/python3.10/site-packages/multiprocess/popen_fork.py:66: RuntimeWarning: os.fork() was called. os.fork() is incompatible with multithreaded code, and JAX is multithreaded, so this will likely lead to a deadlock.\n",
+      "  self.pid = os.fork()\n"
+     ]
     }
    ],
    "source": [
@@ -341,7 +349,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": 10,
    "metadata": {},
    "outputs": [
     {