From f9514a36e2133987cba16f0d247cf3f39c319a63 Mon Sep 17 00:00:00 2001 From: Meg Schwamb Date: Mon, 13 Jan 2025 23:01:29 +0000 Subject: [PATCH] linter updates --- .../activity/lsst_comet/lsst_comet_activity.py | 11 ++++++----- .../lightcurve/sinusoidal/sinusoidal_lightcurve.py | 4 ++-- 2 files changed, 8 insertions(+), 7 deletions(-) diff --git a/src/sorcha_addons/activity/lsst_comet/lsst_comet_activity.py b/src/sorcha_addons/activity/lsst_comet/lsst_comet_activity.py index 90dbf9b..523164d 100644 --- a/src/sorcha_addons/activity/lsst_comet/lsst_comet_activity.py +++ b/src/sorcha_addons/activity/lsst_comet/lsst_comet_activity.py @@ -29,10 +29,10 @@ def __init__( ) -> None: super().__init__(required_column_names) - def checkPhysical(self, df : pd.DataFrame): + def checkPhysical(self, df: pd.DataFrame): if (df.k > 0).any(): raise ValueError("Check complex parameters file. k > 0 is not a physical value") - + def compute( self, df: pd.DataFrame, @@ -101,14 +101,15 @@ def name_id() -> str: """ return "lsst_comet" - def maxBrightness(self, + def maxBrightness( + self, df: pd.DataFrame, observing_filters: List[str], q: List[float], delta: List[float], alpha: List[float], - ): - + ): + self._validate_column_names(df) self.checkPhysical(df) diff --git a/src/sorcha_addons/lightcurve/sinusoidal/sinusoidal_lightcurve.py b/src/sorcha_addons/lightcurve/sinusoidal/sinusoidal_lightcurve.py index f835eee..adfc29a 100644 --- a/src/sorcha_addons/lightcurve/sinusoidal/sinusoidal_lightcurve.py +++ b/src/sorcha_addons/lightcurve/sinusoidal/sinusoidal_lightcurve.py @@ -54,5 +54,5 @@ def name_id() -> str: """ return "sinusoidal" - def maxBrightness(self, df : pd.DataFrame) -> float: - return -df["LCA"] #note this - because magnitudes are weird + def maxBrightness(self, df: pd.DataFrame) -> float: + return -df["LCA"] # note this - because magnitudes are weird