Merge pull request #227 from merl-dev/update_dependencies

update dependencies

ngboost/helpers.py

@@ -5,13 +5,13 @@
 def Y_from_censored(T, E=None):
     if T is None:
         return None
-    T = check_array(T, ensure_2d=False)
-    T = T.reshape(T.shape[0])
     if T.dtype == [
         ("Event", "?"),
         ("Time", "<f8"),
     ]:  # already processed. Necessary for when d_score() calls score() as in LogNormalCRPScore
         return T
+    T = check_array(T, ensure_2d=False)
+    T = T.reshape(T.shape[0])
     if E is None:
         E = np.ones_like(T)
     else:

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "ngboost"
-version = "0.3.11dev"
+version = "0.3.12dev"
 description = "Library for probabilistic predictions via gradient boosting."
 authors = ["Stanford ML Group <[email protected]>"]
 readme = "README.md"
@@ -14,12 +14,12 @@ license = "Apache License 2.0"
 
 [tool.poetry.dependencies]
 python = ">=3.6, <4.0"
-scikit-learn = ">=0.21, <0.24"
-numpy = ">=1.17, <2.0"
-scipy = ">=1.3, <2.0"
-tqdm = ">=4.4, <5.0"
-lifelines = ">=0.25, <0.29"
-importlib-metadata = "^3.4.0"
+scikit-learn = ">=0.21"
+numpy = ">=1.17"
+scipy = ">=1.3"
+tqdm = ">=4.3"
+lifelines = ">=0.25"
+
 
 [tool.poetry.dev-dependencies]
 pytest = "^6.1.2"