Fix imports

apache · Sep 30, 2024 · 7737e8b · 7737e8b
1 parent d0ad1ff
commit 7737e8b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sdks/python/apache_beam/runners/worker/statesampler_fast.pxd b/sdks/python/apache_beam/runners/worker/statesampler_fast.pxd
@@ -18,7 +18,7 @@
 cimport cython
 
 from apache_beam.metrics.execution cimport MetricsContainer
-from apache_beam.runners.worker.statesampler_interface import StateSamplerInterface
+from apache_beam.runners.worker.statesampler_interface cimport StateSamplerInterface
 
 from cpython cimport pythread
 from libc.stdint cimport int32_t, int64_t

diff --git a/sdks/python/apache_beam/runners/worker/statesampler_fast.pyx b/sdks/python/apache_beam/runners/worker/statesampler_fast.pyx
@@ -36,7 +36,7 @@ import threading
 
 from apache_beam.utils.counters import CounterName
 from apache_beam.metrics.execution cimport MetricsContainer
-from apache_beam.runners.worker.statesampler_interface import StateSamplerInterface
+from apache_beam.runners.worker.statesampler_interface cimport StateSamplerInterface
 
 cimport cython
 from cpython cimport pythread