Skip to content

Commit 07bd4a2

Browse files
author
florian
committed
renamed new column
1 parent 325a570 commit 07bd4a2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

category_encoders/tests/helpers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,9 +50,9 @@ def create_dataset(n_rows=1000, extras=False, has_none=True):
5050
random.choice(['A', 'B', 'C', np.nan]) # Categorical with missing values
5151
] for row in range(n_rows)]
5252

53-
df = pd.DataFrame(ds, columns=['float', 'float_edge', 'unique_int', 'unique_str', 'invariant', 'underscore', 'none', 'extra', 321, 'categorical', 'categorical_na'])
53+
df = pd.DataFrame(ds, columns=['float', 'float_edge', 'unique_int', 'unique_str', 'invariant', 'underscore', 'none', 'extra', 321, 'categorical', 'na_categorical'])
5454
df['categorical'] = pd.Categorical(df['categorical'], categories=['A', 'B', 'C'])
55-
df['categorical_na'] = pd.Categorical(df['categorical_na'], categories=['A', 'B', 'C'])
55+
df['na_categorical'] = pd.Categorical(df['na_categorical'], categories=['A', 'B', 'C'])
5656
return df
5757

5858

0 commit comments

Comments
 (0)