9de86aeb8e02e50d79fd859d23104cdb4e950bf3,stellargraph/layer/graphsage.py,DirectedGraphSAGE,_get_sizes_from_keywords,#DirectedGraphSAGE#,1116

Before Change

             kwargs: The additional keyword arguments.
        
        self.in_samples = kwargs.get("in_samples")
        self.out_samples = kwargs.get("out_samples")
        self.input_feature_size = kwargs.get("input_dim")
        if (
            self.in_samples is None
            or self.out_samples is None
            or self.input_feature_size is None
        ):
            raise ValueError(
                "If generator is not provided, in_samples, out_samples and input_dim must be specified."
            )
        if len(self.in_samples) != self.max_hops:
            raise ValueError(
                "Mismatched lengths: in-node sample sizes {} versus layer sizes {}".format(
                    self.in_samples, self.layer_sizes

After Change

        Args:
             kwargs: The additional keyword arguments.
        
        try:
            self.in_samples = kwargs["in_samples"]
            self.out_samples = kwargs["out_samples"]
            self.input_feature_size = kwargs["input_dim"]
            self.multiplicity = kwargs["multiplicity"]

        except KeyError:
            raise KeyError(
                "If generator is not provided, in_samples, out_samples, "
                "input_dim, and multiplicity must be specified."
            )

        if len(self.in_samples) != self.max_hops:
            raise ValueError(
                "Mismatched lengths: in-node sample sizes {} versus layer sizes {}".format(
                    self.in_samples, self.layer_sizes

Instances