Skip to content

Commit

Permalink
style: fix
Browse files Browse the repository at this point in the history
  • Loading branch information
tazlin committed Jun 3, 2024
1 parent c1bc46d commit adfbd9a
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion tests/test_image.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
import json

import requests

TEST_MODELS = ["Fustercluck", "AlbedoBase XL (SDXL)"]
Expand Down Expand Up @@ -76,7 +77,7 @@ def test_simple_image_gen(api_key: str, HORDE_URL: str, CIVERSION: str) -> None:
retrieve_req = requests.get(f"{protocol}://{HORDE_URL}/api/v2/generate/status/{req_id}", headers=headers)
assert retrieve_req.ok, retrieve_req.text
retrieve_results = retrieve_req.json()
print(json.dumps(retrieve_results,indent=4))
print(json.dumps(retrieve_results, indent=4))
assert len(retrieve_results["generations"]) == 1
gen = retrieve_results["generations"][0]
assert len(gen["gen_metadata"]) == 0
Expand Down

0 comments on commit adfbd9a

Please sign in to comment.