fixed test

amaiya · Apr 21, 2023 · 4ca37d1 · 4ca37d1
1 parent 9dc21d1
commit 4ca37d1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_tabular.py b/tests/test_tabular.py
@@ -17,8 +17,8 @@
 class TestTabular(TestCase):
     def test_classification(self):
         train_df = pd.read_csv("resources/tabular_data/train.csv", index_col=0)
-        train_df = train_df.drop("Name", 1)
-        train_df = train_df.drop("Ticket", 1)
+        train_df = train_df.drop("Name", axis=1)
+        train_df = train_df.drop("Ticket", axis=1)
         trn, val, preproc = tabular.tabular_from_df(
             train_df, label_columns="Survived", random_state=42
         )