Method parse_val

rat-sql-gap/seq2struct/grammars/spider.py:146–172

(self, val)

    #

    def parse_val(self, val):
        if isinstance(val, str):
            if not self.include_literals:
                return {'_type': 'Terminal'}
            return {
                    '_type': 'String',
                    's': val,
            }
        elif isinstance(val, list):
            return {
                    '_type': 'ColUnit',
                    'c': self.parse_col_unit(val),
            }
        elif isinstance(val, float):
            if not self.include_literals:
                return {'_type': 'Terminal'}
            return {
                    '_type': 'Number',
                    'f': val,
            }
        elif isinstance(val, dict):
            return {
                    '_type': 'ValSql',
                    's': self.parse_sql(val),
            }
        else:
            raise ValueError(val)

    def parse_col_unit(self, col_unit):
        agg_id, col_id, is_distinct = col_unit