93a71bd6c3f3c731473bc0ddd9b894a252738d01,src/backends/cgpm_backend.py,CGPM_Backend,json_ready_models,#CGPM_Backend#Any#Any#,926

Before Change

        import operator, itertools
        // Sort (name, code, value) list by variable name
        sorted_cats = sorted(cursor, key=operator.itemgetter(0))
        // Group triples by variable name -> list? of (name, group)
        groups = itertools.groupby(sorted_cats, key=operator.itemgetter(0))

        return {"column-statistical-types": stattypes,
                "categories": {name: {triple[1]:triple[2] for triple in group} for name, group in groups},
                "models": []}

    def _unique_rowid(self, rowids):
        if len(set(rowids)) != 1:
            raise ValueError("Multiple-row query: %r" % (list(set(rowids)),))

After Change

                     """, (generator_id, population_id))}

        raw_categories = \
            sorted(bdb.sql_execute("""
                       SELECT colno, code, value FROM bayesdb_cgpm_category
                       WHERE generator_id = ?
                   """, (generator_id,)))
        assert len(raw_categories) > 0

        // compare core.bayesdb_variable_name
        cat_variables = {colno: name for (colno, name) in
                         bdb.sql_execute("""
                             SELECT colno, name FROM bayesdb_variable
                             WHERE (generator_id IS NULL OR generator_id = ?)
                             AND population_id = ?
                         """, (generator_id, population_id))}
        assert len(cat_variables) > 0

        // Doing the join in python because i can"t figure out how to
        // do it in sqlite SQL
        import operator, itertools
        groups = \
            {colno: group for (colno, group) in 
                itertools.groupby(raw_categories, key=operator.itemgetter(0))}
        assert len(groups) > 0
        assert cat_variables[groups.keys()[0]]

Instances