test update to be py 3.7 compat

GitOrigin-RevId: 579d26031497704ce4791cfa8768a0e95141cd5e
gretelai · Dec 22, 2022 · b0c6c15 · b0c6c15
1 parent cc933ea
commit b0c6c15
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/actgan/test_actgan.py b/tests/actgan/test_actgan.py
@@ -29,7 +29,8 @@ def test_auto_transform_datetimes(test_df):
     model._fit = Mock()
     model.fit(test_df)
 
-    transformed_df = model._fit.mock_calls[0].args[0]
+    _, args, _ = model._fit.mock_calls[0]
+    transformed_df = args[0]
     assert is_number(transformed_df[transformed_df.columns[0]][0])