diff --git a/pyproject.toml b/pyproject.toml index c40d452..bfabdd7 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -19,8 +19,8 @@ requires-python = ">=3.8" dependencies = [ "numpy", "cvxpy", - "scikit-survival", - "scikit-learn", + "scikit-survival==0.18.0", + "scikit-learn==1.1.2", "pandas", "tqdm", "seaborn",