Internal change

PiperOrigin-RevId: 521463747

Author: tomvdw

tensorflow_datasets/core/dataset_builder.py

@@ -57,6 +57,7 @@
 import termcolor
 
 if typing.TYPE_CHECKING:
+  import apache_beam as beam
   from apache_beam.runners import runner
   from array_record.python import array_record_data_source