Skip to content

Commit

Permalink
fix test
Browse files Browse the repository at this point in the history
  • Loading branch information
Erik Bernhardsson committed Mar 31, 2020
1 parent 174d38a commit 86d4864
Showing 1 changed file with 0 additions and 4 deletions.
4 changes: 0 additions & 4 deletions test_convoys.py
Original file line number Diff line number Diff line change
Expand Up @@ -86,10 +86,6 @@ def test_output_shapes(c=0.3, lambd=0.1, n=1000, k=5):
assert model.predict_ci([[X[0]], [X[1]]], [[0, 1, 2]], ci=0.8) \
.shape == (2, 3, 3)

# Assert old interface still works
assert model.cdf(X[0], 0).shape == ()
assert model.cdf(X[0], 0, ci=0.8).shape == (3,)

# Fit model without ci (should be the same)
model = convoys.regression.Exponential(ci=False)
model.fit(X, B, T)
Expand Down

0 comments on commit 86d4864

Please sign in to comment.