Adding cupy as dependency. Copied MBLS from astropy into repo. Upda…

…ted comparison notebook.

docs/notebooks/mbls-comparison.ipynb

@@ -79,6 +79,45 @@
     "print(f\"Frequency with peak power: {frequency[np.argmax(power)]}\")"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Reproduce the results above using the Mapals implementation of Multiband Lomb Scargle"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from mapals.lombscargle_multiband import LombScargleMultiband as MapalsLombScargleMultiband"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "model = MapalsLombScargleMultiband(t, y, bands, dy, nterms_base=1, nterms_band=1)\n",
+    "frequency, power = model.autopower(method=\"flexible\")\n",
+    "freq_maxpower = frequency[np.argmax(power)]\n",
+    "t_phase = np.linspace(0, 1 / freq_maxpower, 100)\n",
+    "y_fit = model.model(t_phase, freq_maxpower)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "plt.plot(frequency, power)\n",
+    "print(f\"Frequency with peak power: {frequency[np.argmax(power)]}\")"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,

pyproject.toml

@@ -17,6 +17,7 @@ requires-python = ">=3.9"
 dependencies = [
     "cuda-python",
     "numpy",
+    "cupy-cuda12x", # assuming CUDA toolkit>=12.0
 ]
 
 [project.urls]

src/mapals/lombscargle_multiband/__init__.py

@@ -0,0 +1,3 @@
+# Licensed under a 3-clause BSD style license - see LICENSE.rst
+
+from .core import LombScargleMultiband