From a87e8f05742412c4ab7ee5866840bb1f248a9c4e Mon Sep 17 00:00:00 2001 From: Li Bo Date: Thu, 18 Jan 2024 00:34:09 +0800 Subject: [PATCH] Update mimicit_dataset.py --- pipeline/mimicit_utils/mimicit_dataset.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pipeline/mimicit_utils/mimicit_dataset.py b/pipeline/mimicit_utils/mimicit_dataset.py index 3213b500..8c148767 100755 --- a/pipeline/mimicit_utils/mimicit_dataset.py +++ b/pipeline/mimicit_utils/mimicit_dataset.py @@ -12,6 +12,7 @@ from io import BytesIO import pandas as pd import numpy as np +import pyarrow.parquet as pq import orjson import torch from PIL import Image, ImageFile