Fix

kevjumba · kevjumba · commit f283f720805d · 2022-06-16T18:14:19.000-07:00
Signed-off-by: Kevin Zhang &lt;kzhang@tecton.ai&gt;
diff --git a/sdk/python/tests/integration/offline_store/test_offline_push.py b/sdk/python/tests/integration/offline_store/test_offline_push.py
@@ -56,9 +56,7 @@ def test_writing_incorrect_order_fails(environment, universal_data_sources):
         },
     )
     with pytest.raises(ValueError):
-        store.write_to_offline_store(
-            driver_stats.name, df, allow_registry_cache=False
-        )
+        store.write_to_offline_store(driver_stats.name, df, allow_registry_cache=False)
 
 
 @pytest.mark.integration
@@ -107,9 +105,7 @@ def test_writing_incorrect_schema_fails(environment, universal_data_sources):
         },
     )
     with pytest.raises(ValueError):
-        store.write_to_offline_store(
-            driver_stats.name, df, allow_registry_cache=False
-        )
+        store.write_to_offline_store(driver_stats.name, df, allow_registry_cache=False)
 
 
 @pytest.mark.integration
