Enable type annotations for python/ops.

PiperOrigin-RevId: 317779660
tensorflow · Jun 23, 2020 · f05f819 · f05f819
1 parent f241cf1
commit f05f819
Show file tree

Hide file tree

Showing 25 changed files with 25 additions and 0 deletions.
diff --git a/tensorflow_probability/python/internal/backend/numpy/gen/adjoint_registrations.py b/tensorflow_probability/python/internal/backend/numpy/gen/adjoint_registrations.py
@@ -34,6 +34,7 @@
 
 from __future__ import absolute_import
 from __future__ import division
+# [internal] enable type annotations
 from __future__ import print_function
 
 from tensorflow_probability.python.internal.backend.numpy import numpy_math as math_ops

diff --git a/tensorflow_probability/python/internal/backend/numpy/gen/cholesky_registrations.py b/tensorflow_probability/python/internal/backend/numpy/gen/cholesky_registrations.py
@@ -34,6 +34,7 @@
 
 from __future__ import absolute_import
 from __future__ import division
+# [internal] enable type annotations
 from __future__ import print_function
 
 from tensorflow_probability.python.internal.backend.numpy import linalg_impl as linalg_ops

diff --git a/tensorflow_probability/python/internal/backend/numpy/gen/inverse_registrations.py b/tensorflow_probability/python/internal/backend/numpy/gen/inverse_registrations.py
@@ -34,6 +34,7 @@
 
 from __future__ import absolute_import
 from __future__ import division
+# [internal] enable type annotations
 from __future__ import print_function
 
 from tensorflow_probability.python.internal.backend.numpy import numpy_math as math_ops

diff --git a/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator.py b/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator.py
@@ -34,6 +34,7 @@
 
 from __future__ import absolute_import
 from __future__ import division
+# [internal] enable type annotations
 from __future__ import print_function
 
 import abc

diff --git a/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_addition.py b/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_addition.py
@@ -34,6 +34,7 @@
 
 from __future__ import absolute_import
 from __future__ import division
+# [internal] enable type annotations
 from __future__ import print_function
 
 import abc

diff --git a/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_adjoint.py b/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_adjoint.py
@@ -34,6 +34,7 @@
 
 from __future__ import absolute_import
 from __future__ import division
+# [internal] enable type annotations
 from __future__ import print_function
 
 from tensorflow_probability.python.internal.backend.numpy import ops

diff --git a/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_algebra.py b/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_algebra.py
@@ -35,6 +35,7 @@
 
 from __future__ import absolute_import
 from __future__ import division
+# [internal] enable type annotations
 from __future__ import print_function
 
 import itertools

diff --git a/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_block_diag.py b/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_block_diag.py
@@ -34,6 +34,7 @@
 
 from __future__ import absolute_import
 from __future__ import division
+# [internal] enable type annotations
 from __future__ import print_function
 
 from tensorflow_probability.python.internal.backend.numpy import ops as common_shapes

diff --git a/...w_probability/python/internal/backend/numpy/gen/linear_operator_block_lower_triangular.py b/...w_probability/python/internal/backend/numpy/gen/linear_operator_block_lower_triangular.py
@@ -34,6 +34,7 @@
 
 from __future__ import absolute_import
 from __future__ import division
+# [internal] enable type annotations
 from __future__ import print_function
 
 from tensorflow_probability.python.internal.backend.numpy import ops as common_shapes

diff --git a/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_circulant.py b/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_circulant.py
@@ -34,6 +34,7 @@
 
 from __future__ import absolute_import
 from __future__ import division
+# [internal] enable type annotations
 from __future__ import print_function
 
 import numpy as np

diff --git a/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_composition.py b/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_composition.py
@@ -34,6 +34,7 @@
 
 from __future__ import absolute_import
 from __future__ import division
+# [internal] enable type annotations
 from __future__ import print_function
 
 from tensorflow_probability.python.internal.backend.numpy import ops as common_shapes

diff --git a/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_diag.py b/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_diag.py
@@ -34,6 +34,7 @@
 
 from __future__ import absolute_import
 from __future__ import division
+# [internal] enable type annotations
 from __future__ import print_function
 
 from tensorflow_probability.python.internal.backend.numpy import ops

diff --git a/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_full_matrix.py b/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_full_matrix.py
@@ -34,6 +34,7 @@
 
 from __future__ import absolute_import
 from __future__ import division
+# [internal] enable type annotations
 from __future__ import print_function
 
 from tensorflow_probability.python.internal.backend.numpy import dtype as dtypes

diff --git a/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_householder.py b/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_householder.py
@@ -34,6 +34,7 @@
 
 from __future__ import absolute_import
 from __future__ import division
+# [internal] enable type annotations
 from __future__ import print_function
 
 from tensorflow_probability.python.internal.backend.numpy import errors

diff --git a/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_identity.py b/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_identity.py
@@ -34,6 +34,7 @@
 
 from __future__ import absolute_import
 from __future__ import division
+# [internal] enable type annotations
 from __future__ import print_function
 
 import numpy as np

diff --git a/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_inversion.py b/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_inversion.py
@@ -34,6 +34,7 @@
 
 from __future__ import absolute_import
 from __future__ import division
+# [internal] enable type annotations
 from __future__ import print_function
 
 from tensorflow_probability.python.internal.backend.numpy import ops

diff --git a/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_kronecker.py b/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_kronecker.py
@@ -34,6 +34,7 @@
 
 from __future__ import absolute_import
 from __future__ import division
+# [internal] enable type annotations
 from __future__ import print_function
 
 from tensorflow_probability.python.internal.backend.numpy import ops as common_shapes

diff --git a/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_low_rank_update.py b/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_low_rank_update.py
@@ -34,6 +34,7 @@
 
 from __future__ import absolute_import
 from __future__ import division
+# [internal] enable type annotations
 from __future__ import print_function
 
 from tensorflow_probability.python.internal.backend.numpy import ops

diff --git a/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_lower_triangular.py b/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_lower_triangular.py
@@ -34,6 +34,7 @@
 
 from __future__ import absolute_import
 from __future__ import division
+# [internal] enable type annotations
 from __future__ import print_function
 
 from tensorflow_probability.python.internal.backend.numpy import ops

diff --git a/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_toeplitz.py b/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_toeplitz.py
@@ -34,6 +34,7 @@
 
 from __future__ import absolute_import
 from __future__ import division
+# [internal] enable type annotations
 from __future__ import print_function
 
 from tensorflow_probability.python.internal.backend.numpy import dtype as dtypes

diff --git a/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_util.py b/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_util.py
@@ -34,6 +34,7 @@
 
 from __future__ import absolute_import
 from __future__ import division
+# [internal] enable type annotations
 from __future__ import print_function
 
 import numpy as np

diff --git a/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_zeros.py b/tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_zeros.py
@@ -34,6 +34,7 @@
 
 from __future__ import absolute_import
 from __future__ import division
+# [internal] enable type annotations
 from __future__ import print_function
 
 import numpy as np

diff --git a/tensorflow_probability/python/internal/backend/numpy/gen/matmul_registrations.py b/tensorflow_probability/python/internal/backend/numpy/gen/matmul_registrations.py
@@ -34,6 +34,7 @@
 
 from __future__ import absolute_import
 from __future__ import division
+# [internal] enable type annotations
 from __future__ import print_function
 
 from tensorflow_probability.python.internal.backend.numpy.gen import linear_operator

diff --git a/tensorflow_probability/python/internal/backend/numpy/gen/registrations_util.py b/tensorflow_probability/python/internal/backend/numpy/gen/registrations_util.py
@@ -34,6 +34,7 @@
 
 from __future__ import absolute_import
 from __future__ import division
+# [internal] enable type annotations
 from __future__ import print_function
 
 

diff --git a/tensorflow_probability/python/internal/backend/numpy/gen/solve_registrations.py b/tensorflow_probability/python/internal/backend/numpy/gen/solve_registrations.py
@@ -34,6 +34,7 @@
 
 from __future__ import absolute_import
 from __future__ import division
+# [internal] enable type annotations
 from __future__ import print_function
 
 from tensorflow_probability.python.internal.backend.numpy.gen import linear_operator