|
36 | 36 | from apache_beam.testing.test_pipeline import TestPipeline |
37 | 37 | from apache_beam.testing.util import assert_that |
38 | 38 | from apache_beam.testing.util import equal_to |
39 | | -from apache_beam.typehints.schemas import LogicalType |
40 | | -from apache_beam.typehints.schemas import MillisInstant |
41 | 39 | from apache_beam.utils.timestamp import Timestamp |
42 | 40 |
|
43 | 41 | # pylint: disable=wrong-import-order, wrong-import-position, ungrouped-imports |
@@ -241,10 +239,6 @@ def test_xlang_jdbc_write_read(self, database): |
241 | 239 |
|
242 | 240 | config = self.jdbc_configs[database] |
243 | 241 |
|
244 | | - # Register MillisInstant logical type to override the mapping from Timestamp |
245 | | - # originally handled by MicrosInstant. |
246 | | - LogicalType.register_logical_type(MillisInstant) |
247 | | - |
248 | 242 | with TestPipeline() as p: |
249 | 243 | p.not_use_test_runner_api = True |
250 | 244 | _ = ( |
@@ -355,10 +349,6 @@ def custom_row_equals(expected, actual): |
355 | 349 | classpath=config['classpath'], |
356 | 350 | )) |
357 | 351 |
|
358 | | - # Register MillisInstant logical type to override the mapping from Timestamp |
359 | | - # originally handled by MicrosInstant. |
360 | | - LogicalType.register_logical_type(MillisInstant) |
361 | | - |
362 | 352 | # Run read pipeline with custom schema |
363 | 353 | with TestPipeline() as p: |
364 | 354 | p.not_use_test_runner_api = True |
|
0 commit comments