From fbbbcccf664c0c86ceaa41455053fc9564d52678 Mon Sep 17 00:00:00 2001 From: Googler Date: Fri, 15 Nov 2024 02:07:59 -0800 Subject: [PATCH] Automated Code Change PiperOrigin-RevId: 696812241 --- .../experimental/auto_batching/allocation_strategy_test.py | 2 -- tensorflow_probability/python/experimental/auto_batching/dsl.py | 2 -- .../python/experimental/auto_batching/dsl_test.py | 1 - .../python/experimental/auto_batching/frontend_test.py | 1 - .../python/experimental/auto_batching/instructions.py | 1 - .../python/experimental/auto_batching/instructions_test.py | 2 -- .../python/experimental/auto_batching/lowering_test.py | 2 -- .../python/experimental/auto_batching/numpy_backend.py | 1 - .../python/experimental/auto_batching/numpy_backend_test.py | 1 - .../experimental/auto_batching/stack_optimization_test.py | 2 -- .../python/experimental/auto_batching/stackless.py | 1 - .../python/experimental/auto_batching/stackless_test.py | 2 -- .../python/experimental/auto_batching/test_programs.py | 1 - .../python/experimental/auto_batching/tf_backend.py | 1 - .../python/experimental/auto_batching/tf_backend_test.py | 2 -- .../python/experimental/auto_batching/type_inference_test.py | 2 -- .../python/experimental/auto_batching/virtual_machine.py | 2 -- .../python/experimental/auto_batching/virtual_machine_test.py | 2 -- 18 files changed, 28 deletions(-) diff --git a/tensorflow_probability/python/experimental/auto_batching/allocation_strategy_test.py b/tensorflow_probability/python/experimental/auto_batching/allocation_strategy_test.py index a9d282a5da..0cef5d8dff 100644 --- a/tensorflow_probability/python/experimental/auto_batching/allocation_strategy_test.py +++ b/tensorflow_probability/python/experimental/auto_batching/allocation_strategy_test.py @@ -14,8 +14,6 @@ # ============================================================================ """Tests of the allocation strategy optimization pass.""" -# Dependency imports - from tensorflow_probability.python.experimental.auto_batching import allocation_strategy from tensorflow_probability.python.experimental.auto_batching import instructions as inst from tensorflow_probability.python.experimental.auto_batching import test_programs diff --git a/tensorflow_probability/python/experimental/auto_batching/dsl.py b/tensorflow_probability/python/experimental/auto_batching/dsl.py index 78fd0e2ee4..735dff0a33 100644 --- a/tensorflow_probability/python/experimental/auto_batching/dsl.py +++ b/tensorflow_probability/python/experimental/auto_batching/dsl.py @@ -24,8 +24,6 @@ import contextlib import inspect -# Dependency imports - from tensorflow_probability.python.experimental.auto_batching import instructions as inst __all__ = [ diff --git a/tensorflow_probability/python/experimental/auto_batching/dsl_test.py b/tensorflow_probability/python/experimental/auto_batching/dsl_test.py index 661ca0683f..ac1aaff721 100644 --- a/tensorflow_probability/python/experimental/auto_batching/dsl_test.py +++ b/tensorflow_probability/python/experimental/auto_batching/dsl_test.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. # ============================================================================ -# Dependency imports import numpy as np import tensorflow.compat.v2 as tf diff --git a/tensorflow_probability/python/experimental/auto_batching/frontend_test.py b/tensorflow_probability/python/experimental/auto_batching/frontend_test.py index 6cdea79e8d..83823a86fa 100644 --- a/tensorflow_probability/python/experimental/auto_batching/frontend_test.py +++ b/tensorflow_probability/python/experimental/auto_batching/frontend_test.py @@ -16,7 +16,6 @@ import collections -# Dependency imports from absl import logging import numpy as np diff --git a/tensorflow_probability/python/experimental/auto_batching/instructions.py b/tensorflow_probability/python/experimental/auto_batching/instructions.py index 1089fe78b4..75fd5cbdfa 100644 --- a/tensorflow_probability/python/experimental/auto_batching/instructions.py +++ b/tensorflow_probability/python/experimental/auto_batching/instructions.py @@ -19,7 +19,6 @@ import inspect import textwrap -# Dependency imports from absl import logging import numpy as np import six diff --git a/tensorflow_probability/python/experimental/auto_batching/instructions_test.py b/tensorflow_probability/python/experimental/auto_batching/instructions_test.py index 83f8c28a3b..3e73afc955 100644 --- a/tensorflow_probability/python/experimental/auto_batching/instructions_test.py +++ b/tensorflow_probability/python/experimental/auto_batching/instructions_test.py @@ -14,8 +14,6 @@ # ============================================================================ """Tests of the instruction language (and definitional interpreter).""" -# Dependency imports - from tensorflow_probability.python.experimental.auto_batching import instructions from tensorflow_probability.python.experimental.auto_batching import test_programs from tensorflow_probability.python.internal import test_util as tfp_test_util diff --git a/tensorflow_probability/python/experimental/auto_batching/lowering_test.py b/tensorflow_probability/python/experimental/auto_batching/lowering_test.py index 1b0e1c1f57..f924f873c3 100644 --- a/tensorflow_probability/python/experimental/auto_batching/lowering_test.py +++ b/tensorflow_probability/python/experimental/auto_batching/lowering_test.py @@ -14,8 +14,6 @@ # ============================================================================ """Tests of the FunctionCallOp lowering compilation pass.""" -# Dependency imports - from tensorflow_probability.python.experimental.auto_batching import allocation_strategy from tensorflow_probability.python.experimental.auto_batching import lowering from tensorflow_probability.python.experimental.auto_batching import numpy_backend diff --git a/tensorflow_probability/python/experimental/auto_batching/numpy_backend.py b/tensorflow_probability/python/experimental/auto_batching/numpy_backend.py index 26da7dceeb..94dc903185 100644 --- a/tensorflow_probability/python/experimental/auto_batching/numpy_backend.py +++ b/tensorflow_probability/python/experimental/auto_batching/numpy_backend.py @@ -23,7 +23,6 @@ import collections -# Dependency imports import numpy as np from tensorflow_probability.python.experimental.auto_batching import instructions diff --git a/tensorflow_probability/python/experimental/auto_batching/numpy_backend_test.py b/tensorflow_probability/python/experimental/auto_batching/numpy_backend_test.py index a17023c15a..2f7a6e6495 100644 --- a/tensorflow_probability/python/experimental/auto_batching/numpy_backend_test.py +++ b/tensorflow_probability/python/experimental/auto_batching/numpy_backend_test.py @@ -14,7 +14,6 @@ # ============================================================================ """Tests for implementations of batched variables.""" -# Dependency imports import hypothesis as hp from hypothesis import strategies as hps from hypothesis.extra import numpy as hpnp diff --git a/tensorflow_probability/python/experimental/auto_batching/stack_optimization_test.py b/tensorflow_probability/python/experimental/auto_batching/stack_optimization_test.py index 42e81e1316..7c28ac636e 100644 --- a/tensorflow_probability/python/experimental/auto_batching/stack_optimization_test.py +++ b/tensorflow_probability/python/experimental/auto_batching/stack_optimization_test.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. # ============================================================================ -# Dependency imports - from tensorflow_probability.python.experimental.auto_batching import stack_optimization as stack from tensorflow_probability.python.experimental.auto_batching import test_programs from tensorflow_probability.python.internal import test_util as tfp_test_util diff --git a/tensorflow_probability/python/experimental/auto_batching/stackless.py b/tensorflow_probability/python/experimental/auto_batching/stackless.py index fd9489f160..eac5652584 100644 --- a/tensorflow_probability/python/experimental/auto_batching/stackless.py +++ b/tensorflow_probability/python/experimental/auto_batching/stackless.py @@ -22,7 +22,6 @@ import functools import heapq -# Dependency imports from absl import logging import six from tensorflow_probability.python.experimental.auto_batching import instructions as inst diff --git a/tensorflow_probability/python/experimental/auto_batching/stackless_test.py b/tensorflow_probability/python/experimental/auto_batching/stackless_test.py index d5b188126a..2d6d325e6a 100644 --- a/tensorflow_probability/python/experimental/auto_batching/stackless_test.py +++ b/tensorflow_probability/python/experimental/auto_batching/stackless_test.py @@ -14,8 +14,6 @@ # ============================================================================ """Tests of the FunctionCallOp lowering compilation pass.""" -# Dependency imports - import tensorflow.compat.v1 as tf1 from tensorflow_probability.python.experimental.auto_batching import allocation_strategy diff --git a/tensorflow_probability/python/experimental/auto_batching/test_programs.py b/tensorflow_probability/python/experimental/auto_batching/test_programs.py index 8557603cf3..9b432523aa 100644 --- a/tensorflow_probability/python/experimental/auto_batching/test_programs.py +++ b/tensorflow_probability/python/experimental/auto_batching/test_programs.py @@ -14,7 +14,6 @@ # ============================================================================ """Tests of the instruction language (and definitional interpreter).""" -# Dependency imports import numpy as np from tensorflow_probability.python.experimental.auto_batching import instructions diff --git a/tensorflow_probability/python/experimental/auto_batching/tf_backend.py b/tensorflow_probability/python/experimental/auto_batching/tf_backend.py index 5a6d3be091..d195adbf2f 100644 --- a/tensorflow_probability/python/experimental/auto_batching/tf_backend.py +++ b/tensorflow_probability/python/experimental/auto_batching/tf_backend.py @@ -20,7 +20,6 @@ import collections import contextlib -# Dependency imports import six import tensorflow.compat.v1 as tf1 import tensorflow.compat.v2 as tf diff --git a/tensorflow_probability/python/experimental/auto_batching/tf_backend_test.py b/tensorflow_probability/python/experimental/auto_batching/tf_backend_test.py index 62cf451fef..a103862e73 100644 --- a/tensorflow_probability/python/experimental/auto_batching/tf_backend_test.py +++ b/tensorflow_probability/python/experimental/auto_batching/tf_backend_test.py @@ -14,8 +14,6 @@ # ============================================================================ """Tests for the TF implementations of auto-batched VM variables.""" -# Dependency imports - import hypothesis as hp from hypothesis import strategies as hps from hypothesis.extra import numpy as hpnp diff --git a/tensorflow_probability/python/experimental/auto_batching/type_inference_test.py b/tensorflow_probability/python/experimental/auto_batching/type_inference_test.py index cc54773fa0..63b95cdbf7 100644 --- a/tensorflow_probability/python/experimental/auto_batching/type_inference_test.py +++ b/tensorflow_probability/python/experimental/auto_batching/type_inference_test.py @@ -16,8 +16,6 @@ import functools -# Dependency imports - from absl.testing import parameterized import numpy as np diff --git a/tensorflow_probability/python/experimental/auto_batching/virtual_machine.py b/tensorflow_probability/python/experimental/auto_batching/virtual_machine.py index 2c4547b609..d942c5dd7e 100644 --- a/tensorflow_probability/python/experimental/auto_batching/virtual_machine.py +++ b/tensorflow_probability/python/experimental/auto_batching/virtual_machine.py @@ -17,8 +17,6 @@ import contextlib import functools -# Dependency imports - from absl import logging import six diff --git a/tensorflow_probability/python/experimental/auto_batching/virtual_machine_test.py b/tensorflow_probability/python/experimental/auto_batching/virtual_machine_test.py index 7559b9b057..4b80c9bf39 100644 --- a/tensorflow_probability/python/experimental/auto_batching/virtual_machine_test.py +++ b/tensorflow_probability/python/experimental/auto_batching/virtual_machine_test.py @@ -16,8 +16,6 @@ import functools -# Dependency imports - from absl import flags import numpy as np