def __init__()

in source/sagemaker/sagemaker_graph_entity_resolution/dgl_entity_resolution/model.py [0:0]

• 5 lines of code
• 3 McCabe index (conditional complexity)

    def __init__(self, in_sizes, out_sizes, etypes):
        super(HeteroRGCNLayer, self).__init__()
        # W_r for each relation
        self.weight = nn.ModuleDict({
                name: nn.Linear(in_size, out_size) for name, in_size, out_size in zip(etypes, in_sizes, out_sizes)
            })