Fix bug in preprocess_spec.py: Remove list() that fails if tensor r…

…ank is unknown. PiperOrigin-RevId: 627656872
google · Apr 24, 2024 · b172f5a · b172f5a
1 parent 613dcbb
commit b172f5a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/clu/preprocess_spec.py b/clu/preprocess_spec.py
@@ -375,7 +375,7 @@ def _describe_features(features: Features) -> str:
   description = {}
   for k, v in features.items():
     if isinstance(v, (tf.Tensor, tf.RaggedTensor, tf.SparseTensor)):
-      description[k] = f"{v.dtype.name}{list(v.shape)}"
+      description[k] = f"{v.dtype.name}{v.shape}"
     elif isinstance(v, dict):
       description[k] = _describe_features(v)
     else: