Skip to content

Commit 3376417

Browse files
authored
Merge pull request apache#8975 Use Beam's abstraction of pickler instead of dill in coder tests.
2 parents 4ba9989 + e2025ff commit 3376417

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

sdks/python/apache_beam/coders/coders_test_common.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -24,10 +24,9 @@
2424
import unittest
2525
from builtins import range
2626

27-
import dill
28-
2927
from apache_beam.coders import proto2_coder_test_messages_pb2 as test_message
3028
from apache_beam.coders import coders
29+
from apache_beam.internal import pickler
3130
from apache_beam.runners import pipeline_context
3231
from apache_beam.transforms import window
3332
from apache_beam.transforms.window import GlobalWindow
@@ -102,7 +101,7 @@ def check_coder(self, coder, *values, **kwargs):
102101
coder.get_impl().estimate_size(v))
103102
self.assertEqual(coder.get_impl().get_estimated_size_and_observables(v),
104103
(coder.get_impl().estimate_size(v), []))
105-
copy1 = dill.loads(dill.dumps(coder))
104+
copy1 = pickler.loads(pickler.dumps(coder))
106105
copy2 = coders.Coder.from_runner_api(coder.to_runner_api(context), context)
107106
for v in values:
108107
self.assertEqual(v, copy1.decode(copy2.encode(v)))

0 commit comments

Comments
 (0)