PR feedback.

ray-project · Nov 7, 2022 · bfdbd7e · bfdbd7e
1 parent 144cb13
commit bfdbd7e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/ray/data/_internal/arrow_block.py b/python/ray/data/_internal/arrow_block.py
@@ -78,7 +78,7 @@ def __getitem__(self, key: str) -> Any:
 
  schema = self._row.schema
  if isinstance(
- schema.field(schema.get_field_index(key)).type,
+ schema.field(key).type,
  (ArrowTensorType, ArrowVariableShapedTensorType),
  ):
  # Build a tensor row.