fix tests

Author: andrewfulton9

.github/workflows/test.yml

@@ -14,7 +14,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.10", "3.11"]
+        python-version: ["3.9", "3.10", "3.11"]
 
     steps:
     - name: Checkout
@@ -29,7 +29,9 @@ jobs:
     - name: Install built wheel
       shell: bash
       run: |
-        pip install --find-links ./dist tensorflow-data-validation[test]
+        PYTHON_VERSION_TAG="cp$(echo ${{ matrix.python-version }} | sed 's/\.//')"
+        WHEEL_FILE=$(ls dist/*${PYTHON_VERSION_TAG}*.whl)
+        pip install "${WHEEL_FILE}[test]"
 
     - name: Run Test
       run: |

g3doc/custom_data_validation.md

@@ -31,7 +31,6 @@ sh_binary(
     srcs = ["move_generated_files.sh"],
     data = select({
         "//conditions:default": [
-            "//tensorflow_data_validation/anomalies/proto:custom_validation_config_proto_py_pb2",
             "//tensorflow_data_validation/anomalies/proto:validation_config_proto_py_pb2",
             "//tensorflow_data_validation/anomalies/proto:validation_metadata_proto_py_pb2",
             "//tensorflow_data_validation/pywrap:tensorflow_data_validation_extension.so",

tensorflow_data_validation/BUILD

@@ -425,37 +425,6 @@ cc_test(
     ],
 )
 
-cc_library(
-    name = "custom_validation",
-    srcs = ["custom_validation.cc"],
-    hdrs = ["custom_validation.h"],
-    deps = [
-        ":path",
-        ":schema",
-        ":status_util",
-        "//tensorflow_data_validation/anomalies/proto:custom_validation_config_proto_cc_pb2",
-        "@com_github_tensorflow_metadata//tensorflow_metadata/proto/v0:metadata_v0_proto_cc_pb2",
-        "@com_google_absl//absl/base:log_severity",
-        "@com_google_absl//absl/container:flat_hash_map",
-        "@com_google_absl//absl/log",
-        "@com_google_absl//absl/status",
-        "@com_google_absl//absl/strings",
-        "@com_google_absl//absl/types:optional",
-    ],
-)
-
-cc_test(
-    name = "custom_validation_test",
-    srcs = ["custom_validation_test.cc"],
-    deps = [
-        ":custom_validation",
-        ":test_util",
-        "@com_github_tensorflow_metadata//tensorflow_metadata/proto/v0:metadata_v0_proto_cc_pb2",
-        "@com_google_absl//absl/types:optional",
-        "@com_google_googletest//:gtest_main",
-    ],
-)
-
 cc_library(
     name = "telemetry",
     srcs = ["telemetry.cc"],