Fix for older python version

blue-yonder · Jun 25, 2020 · 122adaa · 122adaa
1 parent 7a8fdfe
commit 122adaa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/units/feature_extraction/test_data.py b/tests/units/feature_extraction/test_data.py
@@ -57,7 +57,7 @@ def test_wide_tsframe(self):
 
     def test_wide_tsframe_without_sort(self):
         df = self.create_test_data_sample_wide()
-        df = df.drop(columns=["sort"])
+        del df["sort"]
         data = WideTsFrameAdapter(df, "id")
 
         self.assert_tsdata(data, WIDE_TEST_DATA_EXPECTED_TUPLES)