polish code (#2561)
polish code (#2561)


sql-machine-learning/sqlflow,sql-machine-learning/sqlflow,sql-machine-learning/sqlflow,sql-machine-learning/sqlflow,sql-machine-learning/sqlflow,sql-machine-learning/sqlflow