diff --git a/components/input-number/index.jsx b/components/input-number/index.jsx
index ad14782ea6..c9d919698b 100644
--- a/components/input-number/index.jsx
+++ b/components/input-number/index.jsx
@@ -1,5 +1,6 @@
-import InputNumber from 'rc-input-number';
import React from 'react';
+import classNames from 'classnames';
+import InputNumber from 'rc-input-number';
export default React.createClass({
getDefaultProps() {
@@ -9,12 +10,13 @@ export default React.createClass({
};
},
render() {
- let sizeClass = '';
- if (this.props.size === 'large') {
- sizeClass = 'ant-input-number-lg';
- } else if (this.props.size === 'small') {
- sizeClass = 'ant-input-number-sm';
- }
- return ;
+ const {className, size, ...other} = this.props;
+ const inputNumberClass = classNames({
+ 'ant-input-number-lg': size === 'large',
+ 'ant-input-number-sm': size === 'small',
+ [className]: !!className,
+ });
+
+ return ;
}
});