similar issues embeddings

getsentry · Aug 3, 2024 · 054be2d · 054be2d
1 parent 808fd0f
commit 054be2d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/sentry/api/endpoints/test_group_similar_issues_embeddings.py b/tests/sentry/api/endpoints/test_group_similar_issues_embeddings.py
@@ -5,13 +5,13 @@
 import orjson
 from urllib3.response import HTTPResponse
 
+from sentry import options
 from sentry.api.endpoints.group_similar_issues_embeddings import (
     GroupSimilarIssuesEmbeddingsEndpoint,
 )
 from sentry.api.serializers.base import serialize
 from sentry.conf.server import SEER_SIMILAR_ISSUES_URL
 from sentry.models.group import Group
-from sentry.seer.similarity.similar_issues import SIMILARITY_REQUEST_METRIC_SAMPLE_RATE
 from sentry.seer.similarity.types import SeerSimilarIssueData, SimilarIssuesEmbeddingsResponse
 from sentry.testutils.cases import APITestCase
 from sentry.testutils.helpers.eventprocessing import save_new_event
@@ -244,7 +244,7 @@ def test_simple(self, mock_logger, mock_seer_request, mock_metrics_incr):
         )
         mock_metrics_incr.assert_any_call(
             "seer.similar_issues_request",
-            sample_rate=SIMILARITY_REQUEST_METRIC_SAMPLE_RATE,
+            sample_rate=options.get("seer.similarity.metrics_sample_rate"),
             tags={
                 "response_status": 200,
                 "outcome": "matching_group_found",
@@ -356,7 +356,7 @@ def test_incomplete_return_data(self, mock_seer_request, mock_logger, mock_metri
         )
         mock_metrics_incr.assert_any_call(
             "seer.similar_issues_request",
-            sample_rate=SIMILARITY_REQUEST_METRIC_SAMPLE_RATE,
+            sample_rate=options.get("seer.similarity.metrics_sample_rate"),
             tags={
                 "response_status": 200,
                 "outcome": "error",
@@ -409,7 +409,7 @@ def test_nonexistent_group(
 
         mock_metrics_incr.assert_any_call(
             "seer.similar_issues_request",
-            sample_rate=SIMILARITY_REQUEST_METRIC_SAMPLE_RATE,
+            sample_rate=options.get("seer.similarity.metrics_sample_rate"),
             tags={
                 "response_status": 200,
                 "outcome": "error",