Merge pull request #1 from cycl0n3sab3r/cycl0n3sab3r-patch-1

Update plot_util.py
This commit is contained in:
Demelza Robinson 2023-05-26 09:34:56 +12:00 committed by GitHub
commit 5aae3cfc08
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -72,7 +72,7 @@ def helper_predict_with_model(self, Xgrid, plot_raw, apply_link, percentiles, wh
if 'Y_metadata' not in predict_kw:
predict_kw['Y_metadata'] = {}
if 'output_index' not in predict_kw['Y_metadata']:
predict_kw['Y_metadata']['output_index'] = Xgrid[:,-1:].astype(np.int)
predict_kw['Y_metadata']['output_index'] = Xgrid[:,-1:].astype(np.int64)
mu, _ = self.predict(Xgrid, **predict_kw)