Size column can be renamed.

scverse · Mar 28, 2020 · 937ec6d · 937ec6d
1 parent 77a79b8
commit 937ec6d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/sctcrpy/_plotting/_vdj_usage.py b/sctcrpy/_plotting/_vdj_usage.py
@@ -18,7 +18,7 @@ def vdj_usage(
     ],
     for_cells: Union[None, list, np.ndarray, pd.Series] = None,
     cell_weights: Union[None, str, list, np.ndarray, pd.Series] = None,
-    # size_column: str = "cell_weights",
+    size_column: str = "cell_weights",
     fraction_base: Union[None, str] = None,
     ax: Union[plt.axes, None] = None,
     bar_clip: int = 5,
@@ -69,8 +69,6 @@ def vdj_usage(
     Axes object. 
     """
 
-    size_column = "cell_weights"
-
     # Execute the tool
     df = tl.vdj_usage(
         adata,

diff --git a/sctcrpy/_tools/_vdj_usage.py b/sctcrpy/_tools/_vdj_usage.py
@@ -17,6 +17,7 @@ def vdj_usage(
     ),
     for_cells: Union[None, list, np.ndarray, pd.Series] = None,
     cell_weights: Union[None, str, list, np.ndarray, pd.Series] = None,
+    size_column: str = "cell_weights",
     fraction_base: Union[None, str] = None,
     as_dict: bool = False,
 ) -> Union[AnnData, dict]:
@@ -54,7 +55,7 @@ def vdj_usage(
     """
 
     # Preproces the data table (remove unnecessary rows and columns)
-    size_column = "cell_weights"
+
     if for_cells is None:
         for_cells = adata.obs.loc[
             ~_is_na(adata.obs.loc[:, target_cols]).all(axis="columns"), target_cols