code linting / formatting

jbloomlab · Jan 30, 2024 · aa66832 · aa66832
1 parent 03a5576
commit aa66832
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 0 deletions.
diff --git a/polyclonal/plot.py b/polyclonal/plot.py
@@ -594,6 +594,7 @@ def lineplot_and_heatmap(
     -------
     altair.Chart
         Interactive plot.
+
     """
     if addtl_tooltip_stats is None:
         addtl_tooltip_stats = []

diff --git a/polyclonal/polyclonal.py b/polyclonal/polyclonal.py
@@ -2867,6 +2867,7 @@ def filter_variants_by_seen_muts(
         variants_df : pandas.DataFrame
             Copy of input dataframe, with rows of variants
             that have unseen mutations removed.
+
         """
         variants_df = variants_df.copy()
 
@@ -3154,6 +3155,7 @@ def mut_escape_corr(self, ref_poly):
         -------
         corr_df : pandas.DataFrame
             Pairwise epitope correlations for escape.
+
         """
         if self.mut_escape_df is None or ref_poly.mut_escape_df is None:
             raise ValueError("Both objects must have `mut_escape_df` initialized.")

diff --git a/polyclonal/polyclonal_collection.py b/polyclonal/polyclonal_collection.py
@@ -851,6 +851,7 @@ def mut_escape_corr_heatmap(
             Plot squared correlation (eg, :math:`R^2` rather :math:`R`).
         **kwargs
             Keyword args for :func:`polyclonal.plot.corr_heatmap`
+
         """
         corr_label = {"pearson": "r", "kendall": "tau", "spearman": "rho"}[method]
         corr2_label = f"{corr_label}2"