Fix reset test (#247)

* actually reset detector during test --------- Co-authored-by: Auto-format Bot <[email protected]>
groundlight · Sep 4, 2024 · 3eb19be · 3eb19be
1 parent 5813c60
commit 3eb19be
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -9,7 +9,7 @@ packages = [
     {include = "**/*.py", from = "src"},
 ]
 readme = "README.md"
-version = "0.17.5"
+version = "0.17.6"
 
 [tool.poetry.dependencies]
 # For certifi, use ">=" instead of "^" since it upgrades its "major version" every year, not really following semver

diff --git a/test/unit/test_detector_reset.py b/test/unit/test_detector_reset.py
@@ -29,5 +29,6 @@ def test_reset_training(gl_experimental: ExperimentalApi):
     # If we reset a detector, we should have low confidence after the reset
     low_confidence_threshold = 0.6
     det = gl_experimental.get_or_create_detector("Test Detector for Resets", "is this a cat?")
+    gl_experimental.reset_detector(det.id)
     iq = gl_experimental.submit_image_query(det, "test/assets/cat.jpeg")
     assert iq.result.confidence < low_confidence_threshold