Bump Version to 0.9.1 (#76)

* bump version to 0.9.1 * fix linting
groundlight · Jun 30, 2023 · 9426378 · 9426378
1 parent bd0f361
commit 9426378
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -9,7 +9,7 @@ packages = [
     {include = "**/*.py", from = "src"},
 ]
 readme = "README.md"
-version = "0.9.0"
+version = "0.9.1"
 
 [tool.poetry.dependencies]
 certifi = "^2021.10.8"

diff --git a/test/integration/test_groundlight.py b/test/integration/test_groundlight.py
@@ -386,14 +386,15 @@ def submit_noisy_image(image, label=None):
     # wait to give enough time to train
     wait_period = 30  # seconds
     num_wait_periods = 4  # 2 minutes total
+    result_confidence = 0.6
     for _ in range(num_wait_periods):
         time.sleep(wait_period)
         new_dog_query = submit_noisy_image(dog)
         new_cat_query = submit_noisy_image(cat)
-        if new_cat_query.result.confidence < 0.6 or new_cat_query.result.label == "YES":
+        if new_cat_query.result.confidence < result_confidence or new_cat_query.result.label == "YES":
             # If the new query is not confident enough, we'll try again
             continue
-        elif new_dog_query.result.confidence < 0.6 or new_dog_query.result.label == "NO":
+        elif new_dog_query.result.confidence < result_confidence or new_dog_query.result.label == "NO":
             # If the new query is not confident enough, we'll try again
             continue
         else: