Skip to content

Commit 4170099

Browse files
author
Ubuntu
committed
adding a test
1 parent be17f93 commit 4170099

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

test/integration/test_groundlight.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -128,6 +128,19 @@ def test_create_detector_with_pipeline_config(gl: Groundlight):
128128
assert isinstance(_detector, Detector)
129129

130130

131+
def test_create_detector_with_edge_pipeline_config(gl: Groundlight):
132+
name = f"Test edge-pipeline-config {datetime.utcnow()}"
133+
query = "Is there a dog (edge-config)?"
134+
_detector = gl.create_detector(
135+
name=name,
136+
query=query,
137+
pipeline_config="never-review",
138+
edge_pipeline_config="generic-cached-timm-efficientnetv2s-knn",
139+
)
140+
assert str(_detector)
141+
assert isinstance(_detector, Detector)
142+
143+
131144
def test_create_detector_with_confidence_threshold(gl: Groundlight):
132145
# "never-review" is a special model that always returns the same result with 100% confidence.
133146
# It's useful for testing.

0 commit comments

Comments
 (0)