Added Pipfile packages that were missing and updated regression tests

esbmc · Feb 11, 2024 · 405aa41 · 405aa41
1 parent 3c3b315
commit 405aa41
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/Pipfile b/Pipfile
@@ -35,6 +35,8 @@ py = "*" # Dependency of pytest-regtest
 build = "*"
 twine = "*"
 hatch = "*"
+transformers = "*"
+torch = "*"
 
 [requires]
 python_version = "3.11"
diff --git a/tests/regtest/_regtest_outputs/test_base_chat_interface.test_send_message.out b/tests/regtest/_regtest_outputs/test_base_chat_interface.test_send_message.out
@@ -1,3 +1,3 @@
 (SystemMessage(content='System message'), AIMessage(content='OK'))
 [HumanMessage(content='Test 1'), AIMessage(content='OK 1'), HumanMessage(content='Test 2'), AIMessage(content='OK 2'), HumanMessage(content='Test 3'), AIMessage(content='OK 3')]
-[ChatResponse(message=AIMessage(content='OK 1'), total_tokens=1024, finish_reason=<FinishReason.stop: 1>), ChatResponse(message=AIMessage(content='OK 2'), total_tokens=1024, finish_reason=<FinishReason.stop: 1>), ChatResponse(message=AIMessage(content='OK 3'), total_tokens=1024, finish_reason=<FinishReason.stop: 1>)]
+[ChatResponse(message=AIMessage(content='OK 1'), total_tokens=15, finish_reason=<FinishReason.stop: 1>), ChatResponse(message=AIMessage(content='OK 2'), total_tokens=23, finish_reason=<FinishReason.stop: 1>), ChatResponse(message=AIMessage(content='OK 3'), total_tokens=31, finish_reason=<FinishReason.stop: 1>)]