resolve Python precompile issue

xukai92 · Apr 27, 2019 · a66dc31 · a66dc31
1 parent 2f101d6
commit a66dc31
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/MLToolkit.jl b/src/MLToolkit.jl
@@ -1,9 +1,14 @@
 __precompile__()
-
 module MLToolkit
 
 greet() = print("Welcome to Kai's machine learning toolkit!")
 
+# Module init
+function __init__()
+    # Bind Python libraries
+    copy!(axes_grid1, PyCall.pyimport("mpl_toolkits.axes_grid1"))
+end
+
 # Package level imports all go here
 import AutoGrad, Knet, PyCall, PyPlot, Distributions, Reexport, Distributed
 

diff --git a/src/plotting.jl b/src/plotting.jl
@@ -18,7 +18,6 @@ end
 
 function plot_grayimg(img, args...; ax=plt.gca())
     @assert length(args) == 0 || length(args) == 2 "You can either plot a single image or declare the `n_rows` and `n_cols`"
-    axes_grid1 == PyCall.PyNULL() && copy!(axes_grid1, PyCall.pyimport("mpl_toolkits.axes_grid1"))
     im = ax."imshow"(make_imggrid(img, args...), cmap="gray")
     plt.axis("off")
     divider = axes_grid1.make_axes_locatable(ax)