Update docstring

jvivian · Dec 7, 2023 · 62e154f · 62e154f
1 parent b61c887
commit 62e154f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/covid19_drdfm/dfm.py b/covid19_drdfm/dfm.py
@@ -74,8 +74,8 @@ def run_model(df: pd.DataFrame, state: str, outdir: Path):  # -> sm.tsa.DynamicF
         state (str): Two-letter state code to process
         outdir (str): Output directory for model CSV files
 
-    Returns:
-        sm.tsa.DynamicFactor: Dynamic Factor Model
+    # Returns:
+        # sm.tsa.DynamicFactor: Dynamic Factor Model
 
     """
     df = state_process(df, state)

diff --git a/covid19_drdfm/processing.py b/covid19_drdfm/processing.py
@@ -148,7 +148,7 @@ def diff_vars(df: pd.DataFrame, cols: list[str], log: bool = False) -> pd.DataFr
 
     Args:
         df (pd.DataFrame): Input DataFrame
-        vars (List[str]): List of columns to difference
+        cols (List[str]): List of columns to difference
         log bool: Whether to take the log(difference) or not
 
     Returns: