From 02d48caf17cad04afc19bbc0aae946ca594aab7a Mon Sep 17 00:00:00 2001
From: Leighton Chen <lechen@microsoft.com>
Date: Thu, 20 Aug 2020 14:02:55 -0700
Subject: [PATCH] Move samplers to SDK package (#1023)

---
 .../src/opentelemetry/exporter/datadog/exporter.py             | 3 ++-
 .../tests/test_datadog_exporter.py                             | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/exporter/opentelemetry-exporter-datadog/src/opentelemetry/exporter/datadog/exporter.py b/exporter/opentelemetry-exporter-datadog/src/opentelemetry/exporter/datadog/exporter.py
index 49dab7c68..c23288442 100644
--- a/exporter/opentelemetry-exporter-datadog/src/opentelemetry/exporter/datadog/exporter.py
+++ b/exporter/opentelemetry-exporter-datadog/src/opentelemetry/exporter/datadog/exporter.py
@@ -21,6 +21,7 @@ from ddtrace.internal.writer import AgentWriter
 from ddtrace.span import Span as DatadogSpan
 
 import opentelemetry.trace as trace_api
+from opentelemetry.sdk.trace import sampling
 from opentelemetry.sdk.trace.export import SpanExporter, SpanExportResult
 from opentelemetry.trace.status import StatusCanonicalCode
 
@@ -246,7 +247,7 @@ def _get_sampling_rate(span):
     return (
         span.sampler.rate
         if ctx.trace_flags.sampled
-        and isinstance(span.sampler, trace_api.sampling.ProbabilitySampler)
+        and isinstance(span.sampler, sampling.ProbabilitySampler)
         else None
     )
 
diff --git a/exporter/opentelemetry-exporter-datadog/tests/test_datadog_exporter.py b/exporter/opentelemetry-exporter-datadog/tests/test_datadog_exporter.py
index 45ce9417e..47a0280c0 100644
--- a/exporter/opentelemetry-exporter-datadog/tests/test_datadog_exporter.py
+++ b/exporter/opentelemetry-exporter-datadog/tests/test_datadog_exporter.py
@@ -23,6 +23,7 @@ from ddtrace.internal.writer import AgentWriter
 from opentelemetry import trace as trace_api
 from opentelemetry.exporter import datadog
 from opentelemetry.sdk import trace
+from opentelemetry.sdk.trace import sampling
 from opentelemetry.sdk.util.instrumentation import InstrumentationInfo
 
 
@@ -497,7 +498,7 @@ class TestDatadogSpanExporter(unittest.TestCase):
             is_remote=False,
             trace_flags=trace_api.TraceFlags(trace_api.TraceFlags.SAMPLED),
         )
-        sampler = trace_api.sampling.ProbabilitySampler(0.5)
+        sampler = sampling.ProbabilitySampler(0.5)
 
         span = trace.Span(
             name="sampled", context=context, parent=None, sampler=sampler