Merge pull request #42 from autonomio/fix_pd_drop

Fix pd drop
autonomio · Apr 20, 2024 · 166e4a4 · 166e4a4
2 parents c6ac9da + e58b203
commit 166e4a4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/wrangle/__init__.py b/wrangle/__init__.py
@@ -9,4 +9,4 @@
 
 del array, col, df, dic
 
-__version__ = '0.7.5'
+__version__ = '0.7.6'
diff --git a/wrangle/df/df_corr_randomforest.py b/wrangle/df/df_corr_randomforest.py
@@ -21,7 +21,7 @@ def df_corr_randomforest(data, y, destructive=True):
     if destructive is False:
         data = data.copy(deep=True)
 
-    x = data.drop(y, 1).values
+    x = data.drop(y, axis=1).values
     labels = data.drop(y, axis=1).columns
     y = data[y].values
Original file line number	Diff line number	Diff line change
Expand Up		@@ -9,4 +9,4 @@

		del array, col, df, dic

		__version__ = '0.7.5'
		__version__ = '0.7.6'