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

bentoml · Oct 23, 2019 · 00555cd · 00555cd
1 parent dd23a76
commit 00555cd
Showing 1 changed file with 1 addition and 1 deletion.
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"