diff --git a/Dockerfile b/Dockerfile index 21c8e975..6bd15db6 100644 --- a/Dockerfile +++ b/Dockerfile @@ -427,6 +427,7 @@ RUN pip install flashtext && \ pip install jax==0.2.12 jaxlib==0.1.64 && \ # ipympl adds interactive widget support for matplotlib pip install ipympl==0.7.0 && \ + pip install pandarallel && \ /tmp/clean-layer.sh # Download base easyocr models. diff --git a/tests/test_pandarralel.py b/tests/test_pandarralel.py new file mode 100644 index 00000000..fe74b0e1 --- /dev/null +++ b/tests/test_pandarralel.py @@ -0,0 +1,11 @@ +import unittest + +import pandas as pd +from pandarallel import pandarallel + +pandarallel.initialize() + +class TestPandarallel(unittest.TestCase): + def test_pandarallel(self): + data = pd.read_csv("/input/tests/data/train.csv") + data['label_converted'] = data['label'].parallel_apply(lambda x: x+1)