Fix linting errors.

dirac-institute · Jul 31, 2023 · f02a562 · f02a562
1 parent ce53dfb
commit f02a562
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 49 deletions.
diff --git a/docs/notebooks/sinusoidal/sinusoidal_example.ipynb b/docs/notebooks/sinusoidal/sinusoidal_example.ipynb
@@ -9,7 +9,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -20,17 +20,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": null,
    "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "<class 'sorcha.lightcurves.base_lightcurve.SinusoidalLightCurve'>\n"
-     ]
-    }
-   ],
+   "outputs": [],
    "source": [
     "# Update the `LC_METHODS` registration dictionary\n",
     "update_lc_subclasses()\n",
@@ -41,22 +33,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": null,
    "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "       FieldMJD  LCA  Period         Time0\n",
-      "0  60277.351867    1   0.001  60277.351867\n",
-      "1  60289.319749    1   0.001  60277.351867\n",
-      "2  60289.330920    1   0.001  60277.351867\n",
-      "3  60292.334497    1   0.001  60277.351867\n",
-      "4  60292.346208    1   0.001  60277.351867\n"
-     ]
-    }
-   ],
+   "outputs": [],
    "source": [
     "# make a small pandas dataframe with ~5 rows\n",
     "import pandas as pd\n",
@@ -74,22 +53,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": null,
    "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "0   -0.340932\n",
-      "1   -0.934792\n",
-      "2    0.186596\n",
-      "3   -0.026723\n",
-      "4    0.737016\n",
-      "dtype: float64\n"
-     ]
-    }
-   ],
+   "outputs": [],
    "source": [
     "# instantiate the sinusoidal class\n",
     "lc_model = LC_METHODS['sinusoidal']()\n",
@@ -118,8 +84,7 @@
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
    "version": "3.10.12"
-  },
-  "orig_nbformat": 4
+  }
  },
  "nbformat": 4,
  "nbformat_minor": 2

diff --git a/src/sorcha_community_utils/sinusoidal/sinusoidal_lightcurve.py b/src/sorcha_community_utils/sinusoidal/sinusoidal_lightcurve.py
@@ -4,6 +4,7 @@
 import pandas as pd
 import numpy as np
 
+
 class SinusoidalLightCurve(AbstractLightCurve):
     """
     Note: assuming sinusoidal in magnitude instead of flux. Maybe not call LCA?

diff --git a/tests/sinusoidal/test_sinusoidal.py b/tests/sinusoidal/test_sinusoidal.py
@@ -1,17 +1,17 @@
 from sorcha_community_utils.sinusoidal.sinusoidal_lightcurve import SinusoidalLightCurve
-import numpy as np
 import pandas as pd
 
+
 def test_sinusoidal_lightcurve_name():
-    assert 'sinusoidal' == SinusoidalLightCurve.name_id()
+    assert "sinusoidal" == SinusoidalLightCurve.name_id()
 
 
 def test_compute_simple():
     data_dict = {
-    'FieldMJD': [0],
-    'LCA': [1],
-    'Period': [1],
-    'Time0': [0],
+        "FieldMJD": [0],
+        "LCA": [1],
+        "Period": [1],
+        "Time0": [0],
     }
 
     df = pd.DataFrame.from_dict(data_dict)