Skip to content

Commit b7183af

Browse files
committed
Clean up reasoning tests
1 parent 889e926 commit b7183af

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

tests/integration/test_knowledge_engines/test_reasoning.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
1-
"""Core tests."""
1+
"""Core tests for reasoning with knowledge engines."""
2+
23
import logging
34
import unittest
45
from typing import Iterator, List
@@ -145,7 +146,7 @@ def test_reason(self):
145146
results = []
146147
for task in self.tasks():
147148
result = reasoner.reason(task)
148-
print(yaml.dump(result.dict(), sort_keys=False))
149+
print(yaml.dump(result.model_dump(), sort_keys=False))
149150
print(result.prompt)
150151
results.append(result)
151152
ReasonerResultSet(results=[result])
@@ -163,7 +164,7 @@ def test_reason_with_explanations(self):
163164
task.include_explanations = True
164165
result = reasoner.reason(task)
165166
print(result.prompt)
166-
print(yaml.dump(result.dict(), sort_keys=False))
167+
print(yaml.dump(result.model_dump(), sort_keys=False))
167168
results.append(result)
168169
for result in results:
169170
print(
@@ -179,7 +180,7 @@ def test_reason_with_chain_of_thought(self):
179180
task.chain_of_thought = True
180181
result = reasoner.reason(task)
181182
print(result.prompt)
182-
print(yaml.dump(result.dict(), sort_keys=False))
183+
print(yaml.dump(result.model_dump(), sort_keys=False))
183184
results.append(result)
184185
for result in results:
185186
print(

0 commit comments

Comments
 (0)