gigl.common.utils.jupyter_magics#

Attributes#

BLACK

CHARCOAL

gigl_root_dir

Classes#

GraphVisualizer

Used to build and visualize graph which is user configured in a yaml file.

GraphVisualizerLayoutMode

Generic enumeration.

PbVisualizer

PbVisualizerFromOutput

Generic enumeration.

Functions#

change_working_dir_to_gigl_root()

Can be used inside notebooks to change the working directory to the GIGL root directory.

show_colored_unified_diff(f1_lines, f2_lines, f1_name, ...)

show_task_config_colored_unified_diff(f1_uri, f2_uri, ...)

Displays a colored unified diff of two task config files.

sort_yaml_dict_recursively(obj)

Module Contents#

class gigl.common.utils.jupyter_magics.GraphVisualizer[source]#

Used to build and visualize graph which is user configured in a yaml file.

static assign_edge_color(name)[source]#

Assign an edge color to a name based on deterministic hash and a fixed palette (optimized for white background).

Parameters:

name (str)

Return type:

str

static assign_node_color(name)[source]#

Assign a node color to a name based on deterministic hash and a fixed palette.

Parameters:

name (str)

Return type:

str

static visualize_graph(data, seed=42, layout_mode=GraphVisualizerLayoutMode.BIPARTITE, subgraph_node_to_unenumerated_node_id_map=None, pos_edges=None, global_root_node=None)[source]#

Warning: This is mostly just AI slop, but it serves the purpose for now. Visualize a graph.

Parameters:

  • data (torch_geometric.data.HeteroData) – The HeteroData object to visualize.

  • seed – The seed for the random number generator - fix it to ensure reproducibility in visualizations

  • layout_mode – Either GraphVisualizerLayoutMode.HOMOGENEOUS or GraphVisualizerLayoutMode.BIPARTITE

  • subgraph_node_to_global_node_mapping – A mapping from local node indices to global node indices.

  • subgraph_node_to_unenumerated_node_id_map (Optional[gigl.common.collections.frozen_dict.FrozenDict[gigl.src.common.types.graph_data.Node, gigl.src.common.types.graph_data.Node]])

  • pos_edges (Optional[dict[tuple[str, str, str], list[tuple[int, int]]]])

  • global_root_node (Optional[tuple[int, str]])

edge_colors = ['#1565C0', '#43A047', '#000000'][source]#
node_colors = ['#64B5F6', '#E57373', '#81C784', '#FFD54F', '#BA68C8', '#4DB6AC', '#F06292', '#A1887F', '#FFB74D'][source]#
class gigl.common.utils.jupyter_magics.GraphVisualizerLayoutMode[source]#

Bases: enum.Enum

Generic enumeration.

Derive from this class to define new enumerations.

BIPARTITE = 'bipartite'[source]#
HOMOGENEOUS = 'homogeneous'[source]#
class gigl.common.utils.jupyter_magics.PbVisualizer(frozen_task_config)[source]#
Parameters:

frozen_task_config (gigl.src.common.types.pb_wrappers.gbml_config.GbmlConfigPbWrapper)

find_node_pb(unenumerated_node_id, unenumerated_node_type, from_output, pb_type)[source]#
Parameters:
Return type:

Optional[Union[snapchat.research.gbml.training_samples_schema_pb2.NodeAnchorBasedLinkPredictionSample, snapchat.research.gbml.training_samples_schema_pb2.RootedNodeNeighborhood]]

plot_pb(pb, layout_mode=GraphVisualizerLayoutMode.BIPARTITE)[source]#
Parameters:
enumerated_node_to_unenumerated_node_id_map: dict[tuple[int, int], tuple[str, int]][source]#
frozen_task_config[source]#
unenumerated_node_id_to_enumerated_node_id_map: dict[tuple[str, int], tuple[int, int]][source]#
class gigl.common.utils.jupyter_magics.PbVisualizerFromOutput[source]#

Bases: enum.Enum

Generic enumeration.

Derive from this class to define new enumerations.

SGS = 'sgs'[source]#
SPLIT_TEST = 'split_test'[source]#
SPLIT_TRAIN = 'split_train'[source]#
SPLIT_VAL = 'split_val'[source]#
gigl.common.utils.jupyter_magics.change_working_dir_to_gigl_root()[source]#

Can be used inside notebooks to change the working directory to the GIGL root directory. This is useful for ensuring that relative imports and file paths work correctly no matter where the notebook is located.

gigl.common.utils.jupyter_magics.show_colored_unified_diff(f1_lines, f2_lines, f1_name, f2_name)[source]#
gigl.common.utils.jupyter_magics.show_task_config_colored_unified_diff(f1_uri, f2_uri, f1_name, f2_name)[source]#

Displays a colored unified diff of two task config files. :param f1_uri: URI of the first file. :type f1_uri: Uri :param f2_uri: URI of the second file. :type f2_uri: Uri

Parameters:
gigl.common.utils.jupyter_magics.sort_yaml_dict_recursively(obj)[source]#
Parameters:

obj (dict)

Return type:

dict

gigl.common.utils.jupyter_magics.BLACK = '#000000'[source]#
gigl.common.utils.jupyter_magics.CHARCOAL = '#36454F'[source]#
gigl.common.utils.jupyter_magics.gigl_root_dir[source]#