Utilities with hypergraph dictionary input

hypergraph_utils.py

from typing import Any
from collections.abc import Hashable, Iterable
from itertools import combinations

def calc_adjacency_dict(hypergraph_dict: dict[Hashable, Iterable[Hashable]]) -> dict[Hashable, set[Hashable]]:
    key_list = list(hypergraph_dict.keys())
    hypergraph_set_dict = {key: set(hypergraph_dict[key]) for key in key_list}
    return_dict = {key: set() for key in key_list}
    for key1, key2 in combinations(key_list, 2):
        if hypergraph_set_dict[key1] & hypergraph_set_dict[key2]:
            return_dict[key1].add(key2)
            return_dict[key2].add(key1)
    return return_dict

print("Hello world!")
my_hypergraph = {11: [1, 2, 3], 22: [3], 33: [2]}
print(calc_adjacency_dict(my_hypergraph))