Skip to content

Commit ae3b6a7

Browse files
committed
Remove references to third party cloudpickle.
1 parent f310803 commit ae3b6a7

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

sdks/python/apache_beam/ml/anomaly/specifiable_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525

2626
from parameterized import parameterized
2727

28+
from apache_beam.internal.cloudpickle import cloudpickle
2829
from apache_beam.ml.anomaly.specifiable import _FALLBACK_SUBSPACE
2930
from apache_beam.ml.anomaly.specifiable import _KNOWN_SPECIFIABLE
3031
from apache_beam.ml.anomaly.specifiable import Spec
@@ -353,7 +354,6 @@ def test_on_pickle(self):
353354
self.assertEqual(FooForPickle.counter, 1)
354355
self.assertEqual(new_foo_2.__dict__, foo.__dict__)
355356

356-
import cloudpickle
357357
FooForPickle.counter = 0
358358
foo = FooForPickle(456)
359359
self.assertEqual(FooForPickle.counter, 0)

sdks/python/apache_beam/typehints/schemas_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,14 +30,14 @@
3030
from typing import Optional
3131
from typing import Sequence
3232

33-
import cloudpickle
3433
import dill
3534
import numpy as np
3635
from hypothesis import given
3736
from hypothesis import settings
3837
from parameterized import parameterized
3938
from parameterized import parameterized_class
4039

40+
from apache_beam.internal.cloudpickle import cloudpickle
4141
from apache_beam.portability import common_urns
4242
from apache_beam.portability.api import schema_pb2
4343
from apache_beam.typehints import row_type

0 commit comments

Comments
 (0)