From 9efa96f55ee7a52d77e84bbcc63e59f7b0077dbb Mon Sep 17 00:00:00 2001 From: Christian Geng Date: Wed, 29 May 2024 14:09:33 +0200 Subject: [PATCH] add utils. --- benchmarks/utils.py | 20 ++++---------------- 1 file changed, 4 insertions(+), 16 deletions(-) diff --git a/benchmarks/utils.py b/benchmarks/utils.py index 9797836b..f103cae5 100644 --- a/benchmarks/utils.py +++ b/benchmarks/utils.py @@ -1,23 +1,11 @@ -import errno -import os -import re -import tempfile -import typing - -import pandas as pd -import pyarrow as pa -import pyarrow.csv as csv -import pyarrow.parquet as parquet -import polars as pl -from pprint import pprint as pr - -import audbackend -import audeer import collections +import polars as pl +import pyarrow as pa from audb.core import define + polars_mappings = pl.datatypes.convert._DataTypeMappings() pl2arrow_mappings = polars_mappings.PY_TYPE_TO_ARROW_TYPE arrow2pl_mapping = dict(zip(pl2arrow_mappings.values(), pl2arrow_mappings.keys())) @@ -51,7 +39,7 @@ def pascheme_to_plscheme(schema: pa.lib.Schema) -> collections.OrderedDict: r"""Convert pyarrow schema to polars schema. Args: - schema a pyarrow scheme + schema: a pyarrow scheme Returns: polars scheme