uber / causalml

Uplift modeling and causal inference with machine learning algorithms
Other
4.87k stars 756 forks source link

Columns and DataType Not Explicitly Set on line 154 of test_sensitivity.py #715

Open CodeSmileBot opened 7 months ago

CodeSmileBot commented 7 months ago

Hello!

I found an AI-Specific Code smell in your project. The smell is called: Columns and DataType Not Explicitly Set

You can find more information about it in this paper: https://dl.acm.org/doi/abs/10.1145/3522664.3528620.

According to the paper, the smell is described as follows:

Problem If the columns are not selected explicitly, it is not easy for developers to know what to expect in the downstream data schema. If the datatype is not set explicitly, it may silently continue the next step even though the input is unexpected, which may cause errors later. The same applies to other data importing scenarios.
Solution It is recommended to set the columns and DataType explicitly in data processing.
Impact Readability

Example:


### Pandas Column Selection
import pandas as pd
df = pd.read_csv('data.csv')
+ df = df[['col1', 'col2', 'col3']]

### Pandas Set DataType
import pandas as pd
- df = pd.read_csv('data.csv')
+ df = pd.read_csv('data.csv', dtype={'col1': 'str', 'col2': 'int', 'col3': 'float'})

You can find the code related to this smell in this link: https://github.com/uber/causalml/blob/7dec7fe271bc61f993ca5824ef2f1464cba608ea/tests/test_sensitivity.py#L144-L164.

I also found instances of this smell in other files, such as:

File: https://github.com/uber/causalml/blob/master/causalml/match.py#L137-L147 Line: 142 File: https://github.com/uber/causalml/blob/master/causalml/match.py#L140-L150 Line: 145 File: https://github.com/uber/causalml/blob/master/causalml/dataset/classification.py#L98-L108 Line: 103 File: https://github.com/uber/causalml/blob/master/causalml/dataset/synthetic.py#L104-L114 Line: 109 File: https://github.com/uber/causalml/blob/master/causalml/dataset/synthetic.py#L436-L446 Line: 441 .

I hope this information is helpful!