#358 Fix Fastai module import when loading artifact (#359)

parano · web-flow · commit 00555cdf4759 · 2019-10-23T14:19:47.000-07:00
diff --git a/bentoml/artifact/fastai_model_artifact.py b/bentoml/artifact/fastai_model_artifact.py
@@ -25,7 +25,7 @@
 
 def _import_fastai_module():
     try:
-        import fastai
+        import fastai.basic_train
     except ImportError:
         raise ImportError(
             "fastai package is required to use " "bentoml.artifacts.FastaiModelArtifact"
