pre-commit auto fixes

Author: andrewfulton9

RELEASE.md

@@ -45,7 +45,7 @@
 *   Enable passing `tf.saved_model.SaveOptions` to model saving functionality.
 *   Census and sentiment examples updated to only use Keras instead of
     estimator.
-*   Depends on `apache-beam[gcp]>=2.53.0,<3` for Python 3.11 and on 
+*   Depends on `apache-beam[gcp]>=2.53.0,<3` for Python 3.11 and on
     `apache-beam[gcp]>=2.47.0,<3` for 3.9 and 3.10.
 *   Depends on `protobuf>=4.25.2,<5` for Python 3.11 and on `protobuf>3.20.3,<5`
     for 3.9 and 3.10.
@@ -1513,4 +1513,3 @@ the generated vocab_filename on a downstream component.
 * Update tensorflow_transform to use `tf.saved_model` APIs.
 * Add default values on example proto coder.
 * Various performance and stability improvements.
-

docs/build_tft_beam_docs.py

@@ -25,51 +25,53 @@
 ```
 
 """
-from absl import app
-from absl import flags
 
-from tensorflow_docs.api_generator import doc_controls
-from tensorflow_docs.api_generator import generate_lib
-from tensorflow_docs.api_generator import public_api
-import tensorflow_transform.beam as tft_beam
+from absl import app, flags
+from tensorflow_docs.api_generator import doc_controls, generate_lib, public_api
 
+import tensorflow_transform.beam as tft_beam
 
-flags.DEFINE_string('output_dir', '/tmp/tft_beam_api/',
-                    'The path to output the files to')
+flags.DEFINE_string(
+    "output_dir", "/tmp/tft_beam_api/", "The path to output the files to"
+)
 
 flags.DEFINE_string(
-    'code_url_prefix',
-    'https://github.com/tensorflow/transform/tree/master/tensorflow_transform',
-    'The url prefix for links to code.')
+    "code_url_prefix",
+    "https://github.com/tensorflow/transform/tree/master/tensorflow_transform",
+    "The url prefix for links to code.",
+)
 
-flags.DEFINE_bool('search_hints', True,
-                  'Include metadata search hints in the generated files')
+flags.DEFINE_bool(
+    "search_hints", True, "Include metadata search hints in the generated files"
+)
 
-flags.DEFINE_string('site_path', 'tfx/transform/api_docs/python',
-                    'Path prefix in the _toc.yaml')
+flags.DEFINE_string(
+    "site_path", "tfx/transform/api_docs/python", "Path prefix in the _toc.yaml"
+)
 
 FLAGS = flags.FLAGS
 
 
 def main(args):
-  if args[1:]:
-    raise ValueError('Unrecognized Command line args', args[1:])
+    if args[1:]:
+        raise ValueError("Unrecognized Command line args", args[1:])
 
-  doc_controls.do_not_generate_docs(tft_beam.analyzer_impls)
+    doc_controls.do_not_generate_docs(tft_beam.analyzer_impls)
 
-  doc_generator = generate_lib.DocGenerator(
-      root_title='TFT-Beam',
-      py_modules=[('tft_beam', tft_beam)],
-      code_url_prefix=FLAGS.code_url_prefix + '/beam',
-      search_hints=FLAGS.search_hints,
-      site_path=FLAGS.site_path,
-      callbacks=[
-          public_api.explicit_package_contents_filter,
-          public_api.local_definitions_filter
-      ])
+    doc_generator = generate_lib.DocGenerator(
+        root_title="TFT-Beam",
+        py_modules=[("tft_beam", tft_beam)],
+        code_url_prefix=FLAGS.code_url_prefix + "/beam",
+        search_hints=FLAGS.search_hints,
+        site_path=FLAGS.site_path,
+        callbacks=[
+            public_api.explicit_package_contents_filter,
+            public_api.local_definitions_filter,
+        ],
+    )
 
-  doc_generator.build(FLAGS.output_dir)
+    doc_generator.build(FLAGS.output_dir)
 
 
-if __name__ == '__main__':
-  app.run(main)
+if __name__ == "__main__":
+    app.run(main)

docs/build_tft_docs.py

@@ -25,45 +25,46 @@
 ```
 
 """
-from absl import app
-from absl import flags
 
-from tensorflow_docs.api_generator import generate_lib
-from tensorflow_docs.api_generator import public_api
-import tensorflow_transform as transform
+from absl import app, flags
+from tensorflow_docs.api_generator import generate_lib, public_api
 
+import tensorflow_transform as transform
 
-flags.DEFINE_string('output_dir', '/tmp/tft_api/',
-                    'The path to output the files to')
+flags.DEFINE_string("output_dir", "/tmp/tft_api/", "The path to output the files to")
 
 flags.DEFINE_string(
-    'code_url_prefix',
-    'https://github.com/tensorflow/transform/tree/master/tensorflow_transform',
-    'The url prefix for links to code.')
+    "code_url_prefix",
+    "https://github.com/tensorflow/transform/tree/master/tensorflow_transform",
+    "The url prefix for links to code.",
+)
 
-flags.DEFINE_bool('search_hints', True,
-                  'Include metadata search hints in the generated files')
+flags.DEFINE_bool(
+    "search_hints", True, "Include metadata search hints in the generated files"
+)
 
-flags.DEFINE_string('site_path', 'tfx/transform/api_docs/python',
-                    'Path prefix in the _toc.yaml')
+flags.DEFINE_string(
+    "site_path", "tfx/transform/api_docs/python", "Path prefix in the _toc.yaml"
+)
 
 FLAGS = flags.FLAGS
 
 
 def main(args):
-  if args[1:]:
-    raise ValueError('Unrecognized Command line args', args[1:])
+    if args[1:]:
+        raise ValueError("Unrecognized Command line args", args[1:])
 
-  doc_generator = generate_lib.DocGenerator(
-      root_title='TF-Transform',
-      py_modules=[('tft', transform)],
-      code_url_prefix=FLAGS.code_url_prefix,
-      search_hints=FLAGS.search_hints,
-      site_path=FLAGS.site_path,
-      callbacks=[public_api.explicit_package_contents_filter])
+    doc_generator = generate_lib.DocGenerator(
+        root_title="TF-Transform",
+        py_modules=[("tft", transform)],
+        code_url_prefix=FLAGS.code_url_prefix,
+        search_hints=FLAGS.search_hints,
+        site_path=FLAGS.site_path,
+        callbacks=[public_api.explicit_package_contents_filter],
+    )
 
-  doc_generator.build(FLAGS.output_dir)
+    doc_generator.build(FLAGS.output_dir)
 
 
-if __name__ == '__main__':
-  app.run(main)
+if __name__ == "__main__":
+    app.run(main)