Skip to content

Commit

Permalink
fix: tests check for cost
Browse files Browse the repository at this point in the history
  • Loading branch information
shreyashankar committed Oct 14, 2024
1 parent f65543e commit 82b2654
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/basic/test_pipeline_with_parsing.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@
operations:
- name: map_operation
type: map
bypass_cache: true
prompt: |
Summarize the following text in one sentence: "{{ input.content }}"
output:
Expand Down Expand Up @@ -200,7 +201,6 @@ def test_pipeline_with_custom_parsing():
os.unlink(tmp_input.name)
os.unlink(tmp_output.name)

assert cost > 0, "Total cost was not calculated or is 0"
print(
f"Pipeline with custom parsing executed successfully. Total cost: ${cost:.2f}"
)

0 comments on commit 82b2654

Please sign in to comment.