Ci lint

.github/actions/setup-env/action.yml

@@ -25,4 +25,3 @@ runs:
       run: |
         python -m pip install --upgrade pip
         pip install ${{ inputs.package-root-dir }}[test]
-

.github/workflows/ci-lint.yml

@@ -0,0 +1,21 @@
+name: pre-commit
+
+on:
+  pull_request:
+  push:
+     branches: [master]
+
+jobs:
+  pre-commit:
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@v4.1.7
+      with:
+        # Ensure the full history is fetched
+        # This is required to run pre-commit on a specific set of commits
+        # TODO: Remove this when all the pre-commit issues are fixed
+        fetch-depth: 0
+    - uses: actions/setup-python@v5.1.1
+      with:
+        python-version: 3.13
+    - uses: pre-commit/action@v3.0.1

.pre-commit-config.yaml

@@ -0,0 +1,39 @@
+# pre-commit is a tool to perform a predefined set of tasks manually and/or
+# automatically before git commits are made.
+#
+# Config reference: https://pre-commit.com/#pre-commit-configyaml---top-level
+#
+# Common tasks
+#
+# - Register git hooks: pre-commit install --install-hooks
+# - Run on all files:   pre-commit run --all-files
+#
+# These pre-commit hooks are run as CI.
+#
+# NOTE: if it can be avoided, add configs/args in pyproject.toml or below instead of creating a new `.config.file`.
+# https://pre-commit.ci/#configuration
+ci:
+  autoupdate_schedule: monthly
+  autofix_commit_msg: |
+    [pre-commit.ci] Apply automatic pre-commit fixes
+
+repos:
+  # general
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v4.6.0
+    hooks:
+      - id: end-of-file-fixer
+        exclude: '\.svg$'
+      - id: trailing-whitespace
+        exclude: '\.svg$'
+      - id: check-json
+      - id: check-yaml
+        args: [--allow-multiple-documents, --unsafe]
+      - id: check-toml
+
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.5.6
+    hooks:
+      - id: ruff
+        args: ["--fix"]
+      - id: ruff-format

docs/index.md

@@ -64,7 +64,7 @@ options {
 -   ![Fairness Indicators on the TensorFlow blog](images/tf_full_color_primary_icon.svg)
 
     ### [Fairness Indicators on the TensorFlow blog](https://blog.tensorflow.org/2019/12/fairness-indicators-fair-ML-systems.html)
-    
+
     ---
 
     [Read on the TensorFlow blog](https://blog.tensorflow.org/2019/12/fairness-indicators-fair-ML-systems.html)

docs/javascripts/mathjax.js

@@ -11,7 +11,7 @@ window.MathJax = {
   }
 };
 
-document$.subscribe(() => { 
+document$.subscribe(() => {
   MathJax.startup.output.clearCache()
   MathJax.typesetClear()
   MathJax.texReset()

docs/tutorials/_toc.yaml

@@ -13,4 +13,3 @@ toc:
   path: /responsible_ai/fairness_indicators/tutorials/Fairness_Indicators_Pandas_Case_Study
 - title: FaceSSD example Colab
   path: /responsible_ai/fairness_indicators/tutorials/Facessd_Fairness_Indicators_Example_Colab
-

fairness_indicators/example_model.py

@@ -21,15 +21,15 @@
 
 from typing import Any
 
-from fairness_indicators import fairness_indicators_metrics  # pylint: disable=unused-import
-from tensorflow import keras
 import tensorflow.compat.v1 as tf
 import tensorflow_model_analysis as tfma
+from tensorflow import keras
 
+from fairness_indicators import fairness_indicators_metrics  # noqa: F401
 
-TEXT_FEATURE = 'comment_text'
-LABEL = 'toxicity'
-SLICE = 'slice'
+TEXT_FEATURE = "comment_text"
+LABEL = "toxicity"
+SLICE = "slice"
 FEATURE_MAP = {
     LABEL: tf.io.FixedLenFeature([], tf.float32),
     TEXT_FEATURE: tf.io.FixedLenFeature([], tf.string),
@@ -38,74 +38,75 @@
 
 
 class ExampleParser(keras.layers.Layer):
-  """A Keras layer that parses the tf.Example."""
+    """A Keras layer that parses the tf.Example."""
+
+    def __init__(self, input_feature_key):
+        self._input_feature_key = input_feature_key
+        self.input_spec = keras.layers.InputSpec(shape=(1,), dtype=tf.string)
+        super().__init__()
 
-  def __init__(self, input_feature_key):
-    self._input_feature_key = input_feature_key
-    self.input_spec = keras.layers.InputSpec(shape=(1,), dtype=tf.string)
-    super().__init__()
+    def compute_output_shape(self, input_shape: Any):
+        return [1, 1]
 
-  def compute_output_shape(self, input_shape: Any):
-    return [1, 1]
+    def call(self, serialized_examples):
+        def get_feature(serialized_example):
+            parsed_example = tf.io.parse_single_example(
+                serialized_example, features=FEATURE_MAP
+            )
+            return parsed_example[self._input_feature_key]
 
-  def call(self, serialized_examples):
-    def get_feature(serialized_example):
-      parsed_example = tf.io.parse_single_example(
-          serialized_example, features=FEATURE_MAP
-      )
-      return parsed_example[self._input_feature_key]
-    serialized_examples = tf.cast(serialized_examples, tf.string)
-    return tf.map_fn(get_feature, serialized_examples)
+        serialized_examples = tf.cast(serialized_examples, tf.string)
+        return tf.map_fn(get_feature, serialized_examples)
 
 
 class Reshaper(keras.layers.Layer):
-  """A Keras layer that reshapes the input."""
+    """A Keras layer that reshapes the input."""
 
-  def call(self, inputs):
-    return tf.reshape(inputs, (1, 32))
+    def call(self, inputs):
+        return tf.reshape(inputs, (1, 32))
 
 
 class Caster(keras.layers.Layer):
-  """A Keras layer that reshapes the input."""
+    """A Keras layer that reshapes the input."""
 
-  def call(self, inputs):
-    return tf.cast(inputs, tf.float32)
+    def call(self, inputs):
+        return tf.cast(inputs, tf.float32)
 
 
 def get_example_model(input_feature_key: str):
-  """Returns a Keras model for testing."""
-  parser = ExampleParser(input_feature_key)
-  text_vectorization = keras.layers.TextVectorization(
-      max_tokens=32,
-      output_mode='int',
-      output_sequence_length=32,
-  )
-  text_vectorization.adapt(
-      ['nontoxic', 'toxic comment', 'test comment', 'abc', 'abcdef', 'random']
-  )
-  dense1 = keras.layers.Dense(
-      32,
-      activation=None,
-      use_bias=True,
-      kernel_initializer='glorot_uniform',
-      bias_initializer='zeros',
-  )
-  dense2 = keras.layers.Dense(
-      1,
-      activation=None,
-      use_bias=False,
-      kernel_initializer='glorot_uniform',
-      bias_initializer='zeros',
-  )
-
-  inputs = tf.keras.Input(shape=(), dtype=tf.string)
-  parsed_example = parser(inputs)
-  text_vector = text_vectorization(parsed_example)
-  text_vector = Reshaper()(text_vector)
-  text_vector = Caster()(text_vector)
-  output1 = dense1(text_vector)
-  output2 = dense2(output1)
-  return tf.keras.Model(inputs=inputs, outputs=output2)
+    """Returns a Keras model for testing."""
+    parser = ExampleParser(input_feature_key)
+    text_vectorization = keras.layers.TextVectorization(
+        max_tokens=32,
+        output_mode="int",
+        output_sequence_length=32,
+    )
+    text_vectorization.adapt(
+        ["nontoxic", "toxic comment", "test comment", "abc", "abcdef", "random"]
+    )
+    dense1 = keras.layers.Dense(
+        32,
+        activation=None,
+        use_bias=True,
+        kernel_initializer="glorot_uniform",
+        bias_initializer="zeros",
+    )
+    dense2 = keras.layers.Dense(
+        1,
+        activation=None,
+        use_bias=False,
+        kernel_initializer="glorot_uniform",
+        bias_initializer="zeros",
+    )
+
+    inputs = tf.keras.Input(shape=(), dtype=tf.string)
+    parsed_example = parser(inputs)
+    text_vector = text_vectorization(parsed_example)
+    text_vector = Reshaper()(text_vector)
+    text_vector = Caster()(text_vector)
+    output1 = dense1(text_vector)
+    output2 = dense2(output1)
+    return tf.keras.Model(inputs=inputs, outputs=output2)
 
 
 def evaluate_model(
@@ -114,23 +115,23 @@ def evaluate_model(
     tfma_eval_result_path,
     eval_config,
 ):
-  """Evaluate Model using Tensorflow Model Analysis.
-
-  Args:
-    classifier_model_path: Trained classifier model to be evaluted.
-    validate_tf_file_path: File containing validation TFRecordDataset.
-    tfma_eval_result_path: Path to export tfma-related eval path.
-    eval_config: tfma eval_config.
-  """
-
-  eval_shared_model = tfma.default_eval_shared_model(
-      eval_saved_model_path=classifier_model_path, eval_config=eval_config
-  )
-
-  # Run the fairness evaluation.
-  tfma.run_model_analysis(
-      eval_shared_model=eval_shared_model,
-      data_location=validate_tf_file_path,
-      output_path=tfma_eval_result_path,
-      eval_config=eval_config,
-  )
+    """Evaluate Model using Tensorflow Model Analysis.
+
+    Args:
+    ----
+      classifier_model_path: Trained classifier model to be evaluted.
+      validate_tf_file_path: File containing validation TFRecordDataset.
+      tfma_eval_result_path: Path to export tfma-related eval path.
+      eval_config: tfma eval_config.
+    """
+    eval_shared_model = tfma.default_eval_shared_model(
+        eval_saved_model_path=classifier_model_path, eval_config=eval_config
+    )
+
+    # Run the fairness evaluation.
+    tfma.run_model_analysis(
+        eval_shared_model=eval_shared_model,
+        data_location=validate_tf_file_path,
+        output_path=tfma_eval_result_path,
+        eval_config=eval_config,
+    )