Internal change

PiperOrigin-RevId: 657694234
Change-Id: Ia0f5dfdf51b8d9396d0cb4f631e4fdd5ddb3049f

Author: TensorFlow Lattice Authors

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")

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, "