tree 446ae1c74dcebfbfc2a7587a93c91a92825eecc0
parent 58f34704f95dfabfdc09283ee3dbc094e914c6e4
author Nikhil Kak <nkak@vmware.com> 1605146027 -0800
committer kaknikhil <nkak@vmware.com> 1607474340 -0800

Fix predict test cases

JIRA: MADLIB-1438

Removed the mocking of keras sessions in predict since they would globally mock the set_session and clear_session functions which made our
fit/eval transition test cases fail.

This wasn't an issue before because fit/eval were also mocking these
functions but now we don't need to mock them anywhere.

Also fixed test_predict_error_should_clear_sd by making it fail after we
populate SD. Previously it was failing before SD was set. Also setting
normalizing_const to 0 doesn't make it fail so we set current_seg_id to
-1 to inject failure

Co-authored-by: Ekta Khanna <ekhanna@vmware.com>