Another attempt at fixing edgeml
import pytest
import json
+import os
from fastai.vision import *
from functools import partial
import wandb
@@ -17,6 +18,9 @@ import glob
def mnist_data(request_mocker, scope='module'):
# TODO: not sure why request_mocker was getting pulled in by default
request_mocker.stop()
+ path = os.path.join(os.path.expanduser("~"), ".fastai", "data", "mnist_tiny")
+ # A bug in fastai was causing this to blow up after the data was first downloaded
+ if not os.path.exists(path):
path = untar_data(URLs.MNIST_TINY)
data = ImageDataBunch.from_folder(path)
return data
