Skip to content

Commit

Permalink
add comment to test
Browse files Browse the repository at this point in the history
Signed-off-by: Ishaan Mehta <45380942+ishaan-mehta@users.noreply.github.com>
  • Loading branch information
ishaan-mehta committed Apr 4, 2024
1 parent b4e20eb commit 4a658af
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions tests/langchain/test_langchain_model_export.py
Original file line number Diff line number Diff line change
Expand Up @@ -2566,5 +2566,9 @@ def test_runnable_with_message_history_predict():
code_paths=[],
)
loaded_model = mlflow.pyfunc.load_model(model_info.model_uri)

# functionally an assert that the params were passed, as RunnableWithMessageHistory throws
# an error if the expected configurable fields are not passed
result = loaded_model.predict([input_example], params_example)

assert result == [TEST_CONTENT]

0 comments on commit 4a658af

Please sign in to comment.