提交 74896c0d 编写于 作者: J jrzaurin

modified tests dir to it can run in one command

上级 71a9b559
import numpy as np import numpy as np
import pandas as pd import pandas as pd
import os
import pytest import pytest
from pytorch_widedeep.preprocessing import ImagePreprocessor from pytorch_widedeep.preprocessing import ImagePreprocessor
full_path = os.path.realpath(__file__)
path = os.path.split(full_path)[0]
df = pd.DataFrame({'galaxies': ['galaxy1.png', 'galaxy2.png']}) df = pd.DataFrame({'galaxies': ['galaxy1.png', 'galaxy2.png']})
img_col = 'galaxies' img_col = 'galaxies'
imd_dir = 'images' imd_dir = os.path.join(path,'images')
processor = ImagePreprocessor() processor = ImagePreprocessor()
X_imgs = processor.fit_transform(df, img_col, img_path=imd_dir) X_imgs = processor.fit_transform(df, img_col, img_path=imd_dir)
############################################################################### ###############################################################################
# There is not much to test here, since I only resize. # There is not much to test here, since I only resize.
############################################################################### ###############################################################################
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册