Merge commit for internal changes

tensorflow · Jul 30, 2024 · f522583 · f522583
2 parents 3660d33 + 544ec1c
commit f522583
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -27,7 +27,7 @@
 # This version number should always be that of the *next* (unreleased) version.
 # Immediately after uploading a package to PyPI, you should increment the
 # version number and push to gitHub.
-__version__ = "2.1.0"
+__version__ = "2.1.1"
 
 if "--release" in sys.argv:
   sys.argv.remove("--release")

diff --git a/tensorflow_lattice/python/lattice_lib.py b/tensorflow_lattice/python/lattice_lib.py
@@ -2482,7 +2482,7 @@ def verify_hyperparameters(lattice_sizes,
                        (weights_shape, lattice_sizes, expected_num_weights))
 
   if input_shape is not None:
-    if not isinstance(input_shape, list):
+    if isinstance(input_shape[-1], int):
       if input_shape[-1] != len(lattice_sizes):
         raise ValueError("Last dimension of input shape must have same number "
                          "of elements as 'lattice_sizes'. 'input shape': %s, "