[Data] [Docs] Adding in missing typing imports (ray-project#44216)

Adds more imports that were missing in ray-project#44203. Example failure here: https://buildkite.com/ray-project/postmerge/builds/3659#018e615b-eeeb-4a7a-8aed-b5cd139eb64b

Signed-off-by: Matthew Owen <[email protected]>

ci/pipeline/determine_tests_to_run.py

@@ -290,6 +290,7 @@ def get_commit_range():
  changed_file.endswith(".py")
  or changed_file.endswith(".ipynb")
  or changed_file.endswith("BUILD")
+ or changed_file.endswith(".rst")
  ):
  RAY_CI_DOC_AFFECTED = 1
  # Else, this affects only a rst file or so. In that case,

doc/source/data/transforming-data.rst

@@ -54,6 +54,8 @@ input and output a dictionary with keys of strings and values of any type. For e
 
 .. testcode::
 
+ from typing import Any, Dict
+
  def fn(row: Dict[str, Any]) -> Dict[str, Any]:
  # access row data
  value = row["col1"]
@@ -95,6 +97,8 @@ dictionaries that have the same type as the input, for example:
 
 .. testcode::
 
+ from typing import Any, Dict, List
+
  def fn(row: Dict[str, Any]) -> List[Dict[str, Any]]:
  # access row data
  value = row["col1"]
@@ -198,6 +202,7 @@ In this case, your function would look like:
 
 .. testcode::
 
+ from typing import Dict, Iterator
  import numpy as np
 
  def fn(batch: Dict[str, np.ndarray]) -> Iterator[Dict[str, np.ndarray]]: