diff --git a/polyclonal/plot.py b/polyclonal/plot.py index 4510e93..0a1d735 100644 --- 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 index 3b137aa..1dc6033 100644 --- 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 index 180a16f..748bc17 100644 --- 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"