code: def find_task_splits_and_features_dict():
      """Get the task available (list was pre-computed by `print_data_split_sizes.py`), and get the features for each task."""
      task_splits_and_features = defaultdict(dict)
      data_split_sizes = DATA_SPLITS_SIZES
  
      for task_name, split_sizes in data_split_sizes.items():
          for split_name in split_sizes.keys():
              split_info = split_infos[task_name]
              features_dict = split_info["features"]
              assert (
                  split_info["num_shards"] == 1
              )  # TODO -> handle multiple shards
  
              if not task_splits_and_features[task_name]:
                  task_splits_and_features[task_name] = {
                      "splits": [],
                      "features_dict": features_dict,
                  }
              task_splits_and_features[task_name]["splits"].append(
                  split_name
              )
              assert (
                  features_dict
                  == task_splits_and_features[task_name][
                      "features_dict"
                  ]
              )
      return task_splits_and_features
  
label_desc: Use image node_id 1 for calling a global function with example usage: find_task_splits_and_features_dict() and returns: task_splits_and_features