Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Jan 22, 2024
1 parent 101b4fa commit dad48f6
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
2 changes: 1 addition & 1 deletion ehrapy/tools/__init__.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from ehrapy.tools._sa import anova_glm, glm, kmf, ols, test_kmf_logrank, test_nested_f_statistic, cox_ph
from ehrapy.tools._sa import anova_glm, cox_ph, glm, kmf, ols, test_kmf_logrank, test_nested_f_statistic
from ehrapy.tools._scanpy_tl_api import * # noqa: F403
from ehrapy.tools.causal._dowhy import causal_inference
from ehrapy.tools.feature_ranking._rank_features_groups import filter_rank_features_groups, rank_features_groups
Expand Down
6 changes: 3 additions & 3 deletions ehrapy/tools/_sa.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,11 +2,11 @@

from typing import TYPE_CHECKING, Literal

import anndata as ehrapy_ad
import numpy as np # This package is implicitly used
import pandas as pd
import statsmodels.api as sm
import statsmodels.formula.api as smf
import anndata as ehrapy_ad
from lifelines import CoxPHFitter, KaplanMeierFitter
from lifelines.statistics import StatisticalResult, logrank_test
from scipy import stats
Expand Down Expand Up @@ -269,7 +269,7 @@ def cox_ph(adata: AnnData, duration_col: str, event_col: str, entry_col: str = N
"""Fit the Cox’s proportional hazard for the survival function.
See https://lifelines.readthedocs.io/en/latest/fitters/regression/CoxPHFitter.html
Args:
adata: anndata object with necessary columns duration_col and event_col (see below)
duration_col: the name of the column in anndata that contains the subjects’ lifetimes.
Expand All @@ -286,7 +286,7 @@ def cox_ph(adata: AnnData, duration_col: str, event_col: str, entry_col: str = N
>>> # So we need to flip `censor_fl` when pass `censor_fl` to KaplanMeierFitter
>>> adata[:, ['censor_flg']].X = np.where(adata[:, ['censor_flg']].X == 0, 1, 0)
>>> cph = ep.tl.cox_ph(adata, "mort_day_censored", "censor_flg")
"""
df = ehrapy_ad.anndata_to_df(adata)
df = df[[duration_col, event_col, entry_col]]
Expand Down

0 comments on commit dad48f6

Please sign in to comment.