fix scipy import

mlfoundations · Aug 4, 2023 · c309447 · c309447
1 parent 43dd9f6
commit c309447
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/open_flamingo/train/data.py b/open_flamingo/train/data.py
@@ -14,7 +14,7 @@
 import webdataset as wds
 from PIL import Image
 import base64
-from scipy import linear_sum_assignment
+from scipy.optimize import linear_sum_assignment
 
 from data_utils import *