File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -15,12 +15,12 @@ def download_file(url: str) -> dict[str, str | int | float]:
1515 return response .content
1616
1717
18- def parse_response (raw : bytes ) -> pd .DataFrame :
18+ def parse_response (raw : bytes ) -> pl .DataFrame :
1919 data = json .loads (raw )["body" ]
2020 return pl .DataFrame (json .loads (data )["benchmarks" ])
2121
2222
23- def get_data_df (url : str ) -> pd .DataFrame :
23+ def get_data_df (url : str ) -> pl .DataFrame :
2424 url = build_proxy_request (url )
2525 data = download_file (url )
2626 return parse_response (data )
@@ -77,7 +77,7 @@ def create_df_categories(df: pl.DataFrame, pattern: str) -> pl.DataFrame:
7777 return df .with_columns (cast_int_or_categorical (df [c ]) for c in parts )
7878
7979
80- def get_preprocessed_df (url : str ) -> pd .DataFrame :
80+ def get_preprocessed_df (url : str ) -> pl .DataFrame :
8181 df = get_data_df (url )
8282 df = preprocess_df (df )
8383 return df
You can’t perform that action at this time.
0 commit comments