Update mimicit_dataset.py

Luodian · Jan 17, 2024 · a87e8f0 · a87e8f0
1 parent 366a09d
commit a87e8f0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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