configure ruff

Author: andrewfulton9

ruff.toml

@@ -39,9 +39,6 @@ select = [
     "NPY",
 ]
 
-[isort]
-profile = "black"
-
 ignore = [
     "D104", # Missing docstring in public package
     "D100", # Missing docstring in public module
@@ -60,6 +57,12 @@ ignore = [
     "D206", # Docstrings should be indented with spaces; unnecessary when running ruff-format
     "E501", # Line length too long; unnecessary when running ruff-format
     "W191", # Indentation contains tabs; unnecessary when running ruff-format
+
+    # REMOVE AFTER FIXING
+    "ANN001", # Missing type annotation for function argument `args`
+    "ANN202", # Missing Missing return type annotation for private function
+    "D103", # Missing docstring in public function
+    "D101", # Missing docstring in public class
 ]
 
 

setup.py

@@ -18,15 +18,15 @@
 import shutil
 import subprocess
 
-import setuptools
-from setuptools import find_packages
-from setuptools import setup
-
 # pylint: disable=g-bad-import-order
 # It is recommended to import setuptools prior to importing distutils to avoid
 # using legacy behavior from distutils.
 # https://setuptools.readthedocs.io/en/latest/history.html#v48-0-0
 from distutils.command import build
+
+import setuptools
+from setuptools import find_packages, setup
+
 # pylint: enable=g-bad-import-order
 
 

tensorflow_metadata/proto/__init__.py

@@ -13,10 +13,12 @@
 # limitations under the License.
 """Init module for tf.Metadata protos."""
 
-from tensorflow_metadata.proto.v0 import anomalies_pb2  # noqa: F401
-from tensorflow_metadata.proto.v0 import derived_feature_pb2  # noqa: F401
-from tensorflow_metadata.proto.v0 import metric_pb2  # noqa: F401
-from tensorflow_metadata.proto.v0 import path_pb2  # noqa: F401
-from tensorflow_metadata.proto.v0 import problem_statement_pb2  # noqa: F401
-from tensorflow_metadata.proto.v0 import schema_pb2  # noqa: F401
-from tensorflow_metadata.proto.v0 import statistics_pb2  # noqa: F401
+from tensorflow_metadata.proto.v0 import (
+    anomalies_pb2,
+    derived_feature_pb2,
+    metric_pb2,
+    path_pb2,
+    problem_statement_pb2,
+    schema_pb2,
+    statistics_pb2,
+)

tensorflow_metadata/python/proto_test.py

@@ -13,11 +13,8 @@
 # limitations under the License.
 """Tests for tensorflow_metadata.python.proto."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 from absl.testing import absltest
+
 from tensorflow_metadata.proto.v0 import schema_pb2
 
 

tools/build_tfmd_docs.py

@@ -25,10 +25,9 @@
 
 """
 
-from absl import app
-from absl import flags
-
+from absl import app, flags
 from tensorflow_docs.api_generator import generate_lib
+
 import tensorflow_metadata as tfmd
 
 # `.proto` (which contains all the classes) is not imported by default