c676395da1a5ca321ba75f6a8025eefa4b99483e,pgmpy/models/ClusterGraph.py,ClusterGraph,get_cardinality,#ClusterGraph#Any#,211

Before Change

        for factor in self.factors:
            for variable, cardinality in zip(factor.scope(), factor.cardinality):
                cardinalities[variable] = cardinality
        if check_cardinality and len(set((x for clique in self.nodes() for x in clique))) != len(cardinalities):
            raise ValueError("Factors for all the variables not defined.")
        return cardinalities

After Change

        >>> student.get_cardinality(node="Alice")
        2
        
        if node:
            for factor in self.factors:
                for variable, cardinality in zip(factor.scope(), factor.cardinality):
                    if node == variable:
                        return cardinality

        else:
	        cardinalities = defaultdict(int)
	        for factor in self.factors:
	            for variable, cardinality in zip(factor.scope(), factor.cardinality):
	                cardinalities[variable] = cardinality
	        return cardinalities

    def get_partition_function(self):
        r
        Returns the partition function for a given undirected graph.

Instances