Internal change

PiperOrigin-RevId: 478540818
Change-Id: I18a2da69735645e8eb6130f13d29a69c1ae06179

Author: Googler

tensorflow_compression/python/entropy_models/BUILD

@@ -95,6 +95,7 @@ py_library(
 
 py_test(
     name = "universal_test",
+    timeout = "long",
     srcs = ["universal_test.py"],
     shard_count = 3,
     deps = [