SYMBOL INDEX (105 symbols across 11 files)

FILE: talk_like_a_graph/graph_generators.py
  function generate_graphs (line 21) | def generate_graphs(
  function remove_graph_data (line 142) | def remove_graph_data(graph: nx.Graph) -> nx.Graph:
  function randomize_directions (line 152) | def randomize_directions(graph: nx.Graph) -> nx.DiGraph:
  function remove_directions (line 166) | def remove_directions(graph: nx.Graph) -> nx.Graph:

FILE: talk_like_a_graph/graph_generators_runner.py
  function write_graphs (line 43) | def write_graphs(graphs: list[nx.Graph], output_dir: str) -> None:
  function main (line 56) | def main(argv: Sequence[str]) -> None:

FILE: talk_like_a_graph/graph_generators_test.py
  class GraphGenerationTest (line 6) | class GraphGenerationTest(absltest.TestCase, parameterized.TestCase):
    method test_number_of_graphs (line 34) | def test_number_of_graphs(self, algorithm, directed, k):
    method test_directions (line 110) | def test_directions(self, algorithm, directed):

FILE: talk_like_a_graph/graph_metrics.py
  function yes_no_accuracy (line 10) | def yes_no_accuracy(

FILE: talk_like_a_graph/graph_metrics_test.py
  class GraphTasksTest (line 5) | class GraphTasksTest(absltest.TestCase):
    method test_yes_no_correct (line 7) | def test_yes_no_correct(self):
    method test_yes_no_ambiguous (line 22) | def test_yes_no_ambiguous(self):
    method test_yes_no_indeterminate (line 37) | def test_yes_no_indeterminate(self):
    method test_yes_no_accuracy_raises_on_ambiguous_target (line 52) | def test_yes_no_accuracy_raises_on_ambiguous_target(self):
    method test_yes_no_accuracy_raises_on_indeterminate_target (line 59) | def test_yes_no_accuracy_raises_on_indeterminate_target(self):

FILE: talk_like_a_graph/graph_tasks.py
  class GraphTask (line 11) | class GraphTask:
    method __init__ (line 14) | def __init__(self):
    method prepare_examples_dict (line 18) | def prepare_examples_dict(
    method create_few_shot_example (line 26) | def create_few_shot_example(
  class CycleCheck (line 32) | class CycleCheck(GraphTask):
    method __init__ (line 35) | def __init__(self):
    method prepare_examples_dict (line 40) | def prepare_examples_dict(
    method create_few_shot_example (line 69) | def create_few_shot_example(
    method choose_few_shot_examples (line 98) | def choose_few_shot_examples(
  class EdgeExistence (line 119) | class EdgeExistence(GraphTask):
    method __init__ (line 122) | def __init__(self):
    method prepare_examples_dict (line 126) | def prepare_examples_dict(
    method create_few_shot_example (line 161) | def create_few_shot_example(
  class NodeCount (line 192) | class NodeCount(GraphTask):
    method __init__ (line 195) | def __init__(self):
    method prepare_examples_dict (line 200) | def prepare_examples_dict(
    method get_nodes_string (line 223) | def get_nodes_string(self, name_dict: dict[int, str], nnodes: int) -> ...
    method create_few_shot_example (line 230) | def create_few_shot_example(
  class NodeDegree (line 247) | class NodeDegree(GraphTask):
    method __init__ (line 250) | def __init__(self):
    method prepare_examples_dict (line 254) | def prepare_examples_dict(
    method get_edge_string (line 282) | def get_edge_string(
    method create_few_shot_example (line 299) | def create_few_shot_example(
  class EdgeCount (line 319) | class EdgeCount(GraphTask):
    method __init__ (line 322) | def __init__(self):
    method prepare_examples_dict (line 327) | def prepare_examples_dict(
    method get_edges_string (line 350) | def get_edges_string(
    method create_few_shot_example (line 360) | def create_few_shot_example(
  class ConnectedNodes (line 376) | class ConnectedNodes(GraphTask):
    method __init__ (line 379) | def __init__(self):
    method prepare_examples_dict (line 383) | def prepare_examples_dict(
    method get_connected_nodes (line 416) | def get_connected_nodes(
    method create_few_shot_example (line 433) | def create_few_shot_example(
  class DisconnectedNodes (line 464) | class DisconnectedNodes(GraphTask):
    method __init__ (line 467) | def __init__(self):
    method prepare_examples_dict (line 471) | def prepare_examples_dict(
    method get_disconnected_nodes (line 508) | def get_disconnected_nodes(
    method create_few_shot_example (line 537) | def create_few_shot_example(
  class Reachability (line 574) | class Reachability(GraphTask):
    method __init__ (line 577) | def __init__(self):
    method prepare_examples_dict (line 581) | def prepare_examples_dict(
    method create_few_shot_example (line 614) | def create_few_shot_example(
  class ShortestPath (line 655) | class ShortestPath(GraphTask):
    method __init__ (line 658) | def __init__(self):
    method prepare_examples_dict (line 662) | def prepare_examples_dict(
    method create_few_shot_example (line 703) | def create_few_shot_example(
  class TriangleCounting (line 751) | class TriangleCounting(GraphTask):
    method __init__ (line 754) | def __init__(self):
    method prepare_examples_dict (line 759) | def prepare_examples_dict(
    method create_few_shot_example (line 786) | def create_few_shot_example(
  class MaximumFlow (line 829) | class MaximumFlow(GraphTask):
    method __init__ (line 832) | def __init__(self):
    method prepare_examples_dict (line 836) | def prepare_examples_dict(
    method create_few_shot_example (line 870) | def create_few_shot_example(
  function has_edge_weights (line 912) | def has_edge_weights(graph):
  function add_edge_weight (line 919) | def add_edge_weight(graph):
  class NodeClassification (line 928) | class NodeClassification(GraphTask):
    method __init__ (line 931) | def __init__(self):
    method prepare_examples_dict (line 943) | def prepare_examples_dict(
    method create_few_shot_example (line 988) | def create_few_shot_example(

FILE: talk_like_a_graph/graph_tasks_generator.py
  function zero_shot (line 35) | def zero_shot(
  function few_shot (line 69) | def few_shot(
  function generate_random_sbm_graph (line 116) | def generate_random_sbm_graph(random_state: np.random.RandomState):
  function main (line 129) | def main(argv: Sequence[str]) -> None:

FILE: talk_like_a_graph/graph_tasks_utils.py
  function laplacian_pos_embedding (line 19) | def laplacian_pos_embedding(graph: nx.Graph, units: int = 4) -> nx.Graph:
  function to_tfgnn (line 33) | def to_tfgnn(graph: nx.Graph, node_ids: list[int]) -> tfgnn.GraphTensor:
  function create_example_feature (line 116) | def create_example_feature(
  function load_graphs (line 179) | def load_graphs(
  function prepare_examples (line 204) | def prepare_examples(
  function create_zero_shot_task (line 247) | def create_zero_shot_task(
  function write_examples (line 267) | def write_examples(examples: list[example_pb2.Example], output_path: str):
  function prepare_few_shots (line 273) | def prepare_few_shots(
  function choose_few_shot_examples (line 291) | def choose_few_shot_examples(
  function create_few_shot_task (line 304) | def create_few_shot_task(

FILE: talk_like_a_graph/graph_text_encoders.py
  function create_node_string (line 8) | def create_node_string(name_dict, nnodes: int) -> str:
  function nx_encoder (line 17) | def nx_encoder(graph: nx.Graph, _: dict[int, str], edge_type="id") -> str:
  function adjacency_encoder (line 44) | def adjacency_encoder(graph: nx.Graph, name_dict: dict[int, str]) -> str:
  function friendship_encoder (line 65) | def friendship_encoder(graph: nx.Graph, name_dict: dict[int, str]) -> str:
  function coauthorship_encoder (line 80) | def coauthorship_encoder(graph: nx.Graph, name_dict: dict[int, str]) -> ...
  function incident_encoder (line 99) | def incident_encoder(graph: nx.Graph, name_dict: dict[int, str]) -> str:
  function social_network_encoder (line 125) | def social_network_encoder(graph: nx.Graph, name_dict: dict[int, str]) -...
  function expert_encoder (line 143) | def expert_encoder(graph: nx.Graph, name_dict: dict[int, str]) -> str:
  function nodes_to_text (line 156) | def nodes_to_text(graph, encoding_type):
  function get_tlag_node_encoder (line 180) | def get_tlag_node_encoder(graph, encoder_name):
  function with_ids (line 224) | def with_ids(graph: nx.Graph, node_encoder: str) -> nx.Graph:
  function encode_graph (line 229) | def encode_graph(

FILE: talk_like_a_graph/graph_text_encoders_test.py
  class GraphTextEncodersTest (line 20) | class GraphTextEncodersTest(absltest.TestCase, parameterized.TestCase):
    method test_encoders (line 65) | def test_encoders(self, encoding_method, expected_result):

FILE: talk_like_a_graph/name_dictionaries.py
  function create_name_dict (line 265) | def create_name_dict(graph, name: str, nnodes: int = 20) -> dict[int, str]: