From 81a4fd82f71a58d31356ac1a71a65c1675e6a462 Mon Sep 17 00:00:00 2001
From: Steve Chien <schien@google.com>
Date: Wed, 13 Dec 2023 19:04:32 -0800
Subject: [PATCH] Temporarily disable two unit tests while fixing a problem
 with multiprocessor setup.

PiperOrigin-RevId: 590782049
---
 tensorflow_privacy/privacy/keras_models/BUILD | 3 +++
 tensorflow_privacy/privacy/optimizers/BUILD   | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/tensorflow_privacy/privacy/keras_models/BUILD b/tensorflow_privacy/privacy/keras_models/BUILD
index 7380084..a4ae42c 100644
--- a/tensorflow_privacy/privacy/keras_models/BUILD
+++ b/tensorflow_privacy/privacy/keras_models/BUILD
@@ -36,6 +36,9 @@ py_test(
 py_test(
     name = "dp_keras_model_distributed_test",
     srcs = ["dp_keras_model_distributed_test.py"],
+    tags = [
+        "manual",
+    ],
     deps = [
         ":dp_keras_model",
         "//tensorflow_privacy/privacy/fast_gradient_clipping:layer_registry",
diff --git a/tensorflow_privacy/privacy/optimizers/BUILD b/tensorflow_privacy/privacy/optimizers/BUILD
index 30b834b..af02ea8 100644
--- a/tensorflow_privacy/privacy/optimizers/BUILD
+++ b/tensorflow_privacy/privacy/optimizers/BUILD
@@ -113,6 +113,9 @@ py_test(
     srcs = ["dp_optimizer_keras_sparse_distributed_test.py"],
     python_version = "PY3",
     srcs_version = "PY3",
+    tags = [
+        "manual",
+    ],
     deps = [":dp_optimizer_keras_sparse"],
 )