Use built-in prompt instead of click.confirm to remove dependency.

PiperOrigin-RevId: 692173947

Author: marcenacp

setup.py

@@ -63,7 +63,6 @@
     # platform except 'x86_64'. See
     # https://github.com/google/array_record/issues/71
     'array_record>=0.5.0;platform_system=="Linux"',
-    'click',
     'dm-tree',
     'etils[edc,enp,epath,epy,etree]>=1.6.0;python_version<"3.11"',
     'etils[edc,enp,epath,epy,etree]>=1.9.1;python_version>="3.11"',

tensorflow_datasets/core/split_builder.py

@@ -28,7 +28,6 @@
 from etils import epath
 from etils import epy
 from tensorflow_datasets.core.utils.lazy_imports_utils import apache_beam as beam
-from tensorflow_datasets.core.utils.lazy_imports_utils import click
 from tensorflow_datasets.core.utils.lazy_imports_utils import psutil
 
 with epy.lazy_imports():
@@ -341,13 +340,14 @@ def beam_pipeline(self) -> 'beam.Pipeline':
 
       total_memory = psutil.virtual_memory().total
       if self._dataset_size >= total_memory:
-        if not click.confirm(
+        value = input(
             (
                 f'The dataset is {self._dataset_size} in size, but your machine'
                 f' has only {utils.Size(total_memory)} of memory. Continue?'
+                '[Y/n] > '
             ),
-            default=True,
-        ):
+        )
+        if value.lower() in ('n', 'no'):
           sys.exit(1)
 
     beam_options = (

tensorflow_datasets/core/utils/lazy_imports_utils.py

@@ -109,7 +109,6 @@ def datasets_error_callback(module_name: Exception):
 
 with epy.lazy_imports():
   import apache_beam
-  import click
   import pandas
   import psutil
   import pyarrow