fixed imports

Author: eyrei123

polars_v_pandas/dataframe_and_lazyframe_time_tests.py

@@ -21,9 +21,7 @@ def create_polars_lazyframe(test_data):
 
 
 def analyze_pandas_dataframe(pandas_df):
-    pandas_df.groupby(["region", "product", "sales_person"])[
-        "sales_income"
-    ].sum()
+    pandas_df.groupby(["region", "product", "sales_person"])["sales_income"].sum()
 
 
 def analyze_polars_dataframe(polars_df):
@@ -56,18 +54,12 @@ def analyze_polars_lazyframe(polars_lf):
 polars_lf = create_polars_lazyframe(test_data)
 
 print(f"Pandas dataframe analysis time for {int(sys.argv[1])} rows:")
-print(
-    Timer(functools.partial(analyze_pandas_dataframe, pandas_df)).timeit(100)
-)
+print(Timer(functools.partial(analyze_pandas_dataframe, pandas_df)).timeit(100))
 
 print()
 print(f"Polars dataframe analysis time for {int(sys.argv[1])} rows:")
-print(
-    Timer(functools.partial(analyze_polars_dataframe, polars_df)).timeit(100)
-)
+print(Timer(functools.partial(analyze_polars_dataframe, polars_df)).timeit(100))
 
 print()
 print(f"Polars lazyframe analysis time for {int(sys.argv[1])} rows:")
-print(
-    Timer(functools.partial(analyze_polars_lazyframe, polars_lf)).timeit(100)
-)
+print(Timer(functools.partial(analyze_polars_lazyframe, polars_lf)).timeit(100))

polars_v_pandas/dataframe_conversions.py

@@ -28,4 +28,3 @@ def agnositic_groupby(df):
 agnositic_groupby(pandas_df)
 
 agnositic_groupby(polars_df)
-

polars_v_pandas/streaming_test.py

@@ -2,7 +2,6 @@
 import sys
 from timeit import Timer
 
-import pandas as pd
 import polars as pl
 
 from data_generation import data_generation
@@ -30,11 +29,7 @@ def analyze_polars_streaming(polars_lf):
 
 print()
 print(f"Polars lazyframe analysis time for {int(sys.argv[1])} rows:")
-print(
-    Timer(functools.partial(analyze_polars_lazyframe, polars_lf)).timeit(100)
-)
+print(Timer(functools.partial(analyze_polars_lazyframe, polars_lf)).timeit(100))
 
 print(f"Polars streaming analysis time for {int(sys.argv[1])} rows:")
-print(
-    Timer(functools.partial(analyze_polars_streaming, polars_lf)).timeit(100)
-)
+print(Timer(functools.partial(analyze_polars_streaming, polars_lf)).timeit(100))