code: def main():
      ds = tfds.as_numpy(
          create_point_tracking_dataset(shuffle_buffer_size=None)
      )
      for i, data in enumerate(ds):
          disp = plot_tracks(
              data["video"] * 0.5 + 0.5,
              data["target_points"],
              data["occluded"],
          )
          media.write_video(f"{i}.mp4", disp, fps=10)
          if i > 10:
              break
  
imports: import functools
  import itertools
  import matplotlib.pyplot
  import mediapy
  import numpy
  import tensorflow.compat.v1
  import tensorflow_datasets
  from tensorflow_graphics.geometry.transformation import rotation_matrix_3d
label_desc: Use image node_id 15 for calling a global function with example usage: main() without return types