No tensorflow 2 (#3548)

Author: veblush

python/python_requirements.in

@@ -30,6 +30,7 @@ absl_py
 ai-edge-litert
 bitarray
 mako
+matplotlib
 numpy
 pillow
 prettyprinter

python/python_requirements_3_10.txt

@@ -109,7 +109,6 @@ py_test(
         "metadata_py",
         "//:tflite_micro_shim",
         requirement("flatbuffers"),
-        requirement("tensorflow"),
     ],
 )
 
@@ -162,7 +161,6 @@ py_test(
         "//tensorflow/lite/python:schema_py",
         requirement("bitarray"),
         requirement("numpy"),
-        requirement("tensorflow"),
     ],
 )
 
@@ -184,7 +182,6 @@ py_test(
     deps = [
         ":model_facade",
         ":test_models",
-        requirement("tensorflow"),
     ],
 )
 
@@ -204,7 +201,6 @@ py_test(
     target_compatible_with = INCOMPATIBLE_WITH_WINDOWS,
     deps = [
         ":spec",
-        requirement("tensorflow"),
     ],
 )
 
@@ -224,7 +220,6 @@ py_test(
     deps = [
         ":spec",
         ":spec_builder",
-        requirement("tensorflow"),
     ],
 )
 
@@ -247,7 +242,6 @@ py_test(
     deps = [
         ":test_models",
         "//tensorflow/lite/python:schema_py",
-        requirement("tensorflow"),
     ],
 )
 

python/python_requirements_3_11.txt

@@ -15,7 +15,7 @@
 import bitarray
 import bitarray.util
 import numpy as np
-import tensorflow as tf
+import unittest
 
 from tflite_micro.tensorflow.lite.micro.compression import compress
 from tflite_micro.tensorflow.lite.micro.compression import metadata_py_generated as schema
@@ -25,7 +25,7 @@
 from tflite_micro.tensorflow.lite.python import schema_py_generated as tflite
 
 
-class TestPackIndices(tf.test.TestCase):
+class TestPackIndices(unittest.TestCase):
 
   def test_basic_case(self):
     indices = np.array([1, 2, 3])
@@ -112,7 +112,7 @@ def test_bitwidth_7(self):
     self.assertEqual(result, expected_bytes)
 
 
-class TestPackLookupTables(tf.test.TestCase):
+class TestPackLookupTables(unittest.TestCase):
 
   def test_int16_positive(self):
     tables = [np.array([0x1234, 0x5678], dtype='<i2')]
@@ -361,7 +361,7 @@ def test_multiple_tables_with_padding(self):
 # yapf: enable
 
 
-class TestsCompression(tf.test.TestCase):
+class TestsCompression(unittest.TestCase):
   """Tests with the uncompressed model."""
 
   @classmethod
@@ -435,7 +435,7 @@ def test_no_axis(self):
                       lambda: compress.compress(self.flatbuffer, specs))
 
 
-class TestLutCompressedArray(tf.test.TestCase):
+class TestLutCompressedArray(unittest.TestCase):
 
   def test_bitwidth(self):
     """Bitwidth is determined from index values."""
@@ -453,7 +453,7 @@ def test_bitwidth(self):
     self.assertEqual(a.index_bitwidth, 1)
 
 
-class TestCompressedModel(tf.test.TestCase):
+class TestCompressedModel(unittest.TestCase):
   """Test the compressed model."""
 
   @classmethod
@@ -538,7 +538,7 @@ def test_compressed_uint8(self):
     self.assertEqual(indices, expected_indices)
 
     expected_values = np.array(range(16), dtype="<u1")
-    self.assertAllEqual(values, expected_values)
+    np.testing.assert_array_equal(values, expected_values)
 
   def test_compressed_int8(self):
     bitwidth, indices, values = self._get_compressed(subgraph=0, tensor=1)
@@ -555,7 +555,7 @@ def test_compressed_int8(self):
     self.assertEqual(indices, expected_indices)
 
     expected_values = np.array(range(-16, 0), dtype="<i1")
-    self.assertAllEqual(values, expected_values)
+    np.testing.assert_array_equal(values, expected_values)
 
   def test_compressed_int16(self):
     bitwidth, indices, values = self._get_compressed(subgraph=0, tensor=2)
@@ -572,7 +572,7 @@ def test_compressed_int16(self):
     self.assertEqual(indices, expected_indices)
 
     expected_values = np.array(range(-1616, -1600), dtype="<i2")
-    self.assertAllEqual(values, expected_values)
+    np.testing.assert_array_equal(values, expected_values)
 
   def test_compressed_int32(self):
     bitwidth, indices, values = self._get_compressed(subgraph=0, tensor=3)
@@ -589,7 +589,7 @@ def test_compressed_int32(self):
     self.assertEqual(indices, expected_indices)
 
     expected_values = np.array(range(-160_016, -160_000), dtype="<i4")
-    self.assertAllEqual(values, expected_values)
+    np.testing.assert_array_equal(values, expected_values)
 
   def test_axis_1(self):
     """Compression along quanitzation_dimension == 1."""
@@ -607,7 +607,7 @@ def test_axis_1(self):
     self.assertEqual(indices, expected_indices)
 
     expected_values = np.array(range(1, 21), dtype=np.dtype("<i2"))
-    self.assertAllEqual(values, expected_values)
+    np.testing.assert_array_equal(values, expected_values)
 
   def test_axis_0(self):
     """Compression along quanitzation_dimension == 0."""
@@ -626,7 +626,7 @@ def test_axis_0(self):
     self.assertEqual(indices, expected_indices)
 
     expected_values = np.array(range(1, 21), dtype=np.dtype("<i2"))
-    self.assertAllEqual(values, expected_values)
+    np.testing.assert_array_equal(values, expected_values)
 
   def test_per_tensor(self):
     """Compression with one value table per tensor."""
@@ -645,8 +645,8 @@ def test_per_tensor(self):
     self.assertEqual(indices, expected_indices)
 
     expected_values = np.array(range(1, 5), dtype=np.dtype("<i2"))
-    self.assertAllEqual(values, expected_values)
+    np.testing.assert_array_equal(values, expected_values)
 
 
 if __name__ == "__main__":
-  tf.test.main()
+  unittest.main()

python/python_requirements_3_12.txt

@@ -18,7 +18,7 @@
 
 from dataclasses import dataclass
 import flatbuffers
-import tensorflow as tf
+import unittest
 
 # `.*_generated` is the name of the module created by the Bazel rule
 # `flatbuffer_py_library' based on the schema.
@@ -48,7 +48,7 @@ class _LutTensor:
 _EXPECTED_SCHEMA_VERSION = 1
 
 
-class TestReadEqualsWrite(tf.test.TestCase):
+class TestReadEqualsWrite(unittest.TestCase):
 
   def setUp(self):
     """Sets up the test by creating a flatbuffer using the metadata schema.
@@ -108,4 +108,4 @@ def testPrintFlatbufferLen(self):
 
 
 if __name__ == "__main__":
-  tf.test.main()
+  unittest.main()

python/python_requirements_3_13.txt

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 import numpy as np
-import tensorflow as tf
+import unittest
 from tflite_micro.tensorflow.lite.python import schema_py_generated as tflite
 from tflite_micro.tensorflow.lite.micro.compression import model_facade
 from tflite_micro.tensorflow.lite.micro.compression import test_models
@@ -96,7 +96,7 @@
 }
 
 
-class TestModelFacade(tf.test.TestCase):
+class TestModelFacade(unittest.TestCase):
 
   def setUp(self):
     self.flatbuffer = test_models.build(TEST_MODEL)
@@ -116,7 +116,7 @@ def testMetadata(self):
     self.assertNotIn("metadata2", self.facade.metadata)
 
 
-class TestTensors(tf.test.TestCase):
+class TestTensors(unittest.TestCase):
 
   def setUp(self):
     flatbuffer = test_models.build(TEST_MODEL)
@@ -135,10 +135,10 @@ def testNameIsString(self):
   def testTensors(self):
     for id, attrs in self.test_tensors:
       tensor = self.facade.subgraphs[0].tensors[id]
-      self.assertAllEqual(tensor.shape, attrs["shape"])
+      np.testing.assert_array_equal(tensor.shape, attrs["shape"])
       data = TEST_MODEL["buffers"][attrs["buffer"]]
-      self.assertAllEqual(tensor.array, data.reshape(tensor.shape))
+      np.testing.assert_array_equal(tensor.array, data.reshape(tensor.shape))
 
 
 if __name__ == "__main__":
-  tf.test.main()
+  unittest.main()

tensorflow/lite/micro/compression/BUILD

@@ -14,13 +14,13 @@
 #
 """Tests for the compression spec builder."""
 
-import tensorflow as tf
+import unittest
 
 from tflite_micro.tensorflow.lite.micro.compression import spec
 from tflite_micro.tensorflow.lite.micro.compression import spec_builder
 
 
-class SpecBuilderTest(tf.test.TestCase):
+class SpecBuilderTest(unittest.TestCase):
 
   def test_basic_builder_pattern(self):
     """Test basic fluent builder usage."""
@@ -109,4 +109,4 @@ def test_builder_produces_same_type_as_parse_yaml(self):
 
 
 if __name__ == "__main__":
-  tf.test.main()
+  unittest.main()