Remove usage of deprecated TensorInfo.dtype call.

PiperOrigin-RevId: 521972189
tensorflow · Apr 5, 2023 · 91bf06f · 91bf06f
1 parent cff72fb
commit 91bf06f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow_datasets/core/features/feature.py b/tensorflow_datasets/core/features/feature.py
@@ -122,7 +122,7 @@ def copy_from(cls, tensor_info: TensorInfo) -> TensorInfo:
     """Copy constructor."""
     return cls(
         shape=tensor_info.shape,
-        dtype=tensor_info.dtype,
+        dtype=tensor_info.np_dtype,
         default_value=tensor_info.default_value,
         sequence_rank=tensor_info.sequence_rank,
         dataset_lvl=tensor_info.dataset_lvl,