def copy_for_dialect()

in evalbench/dataset/evalinput.py [0:0]

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

    def copy_for_dialect(self, dialect: str):
        copy = self.copy()
        copy.dialects = [dialect]
        if isinstance(self.golden_sql, dict):
            copy.golden_sql = self.golden_sql.get(dialect, [])
        if isinstance(self.eval_query, dict):
            copy.eval_query = self.eval_query.get(dialect, [])
        if isinstance(self.setup_sql, dict):
            copy.setup_sql = self.setup_sql.get(dialect, [])
        if isinstance(self.cleanup_sql, dict):
            copy.cleanup_sql = self.cleanup_sql.get(dialect, [])
        return copy