Merge pull request #442 from malmaud/active_project

Start Python task in current project environment.

Author: malmaud

src/TensorFlow.jl

@@ -142,7 +142,9 @@ end
 function load_python_process(;force_reload=false)
     if myid() == 1
         (pyproc[] > 0 && !force_reload) && return pyproc[] # Python process already loaded
-        addprocs(1)
+        withenv("JULIA_PROJECT"=>Base.active_project()) do
+            addprocs(1)
+        end
         pyproc[] = nprocs()
         py_file = joinpath(dirname(@__FILE__), "py.jl")
         Base.eval(Main, quote

src/core.jl

@@ -1819,7 +1819,8 @@ function get_tensor_shape(tensor::AbstractTensor)
 end
 
 function get_shape(tensor::AbstractTensor)
-    if ismissing(get_tensor_num_dims(tensor))
+    # TODO understand/fix why gradient tensors have 'nothing' for their graph
+    if get_graph(tensor) === nothing || ismissing(get_tensor_num_dims(tensor))
         return TensorShape([], true)
     else
         return TensorShape(get_tensor_shape(tensor), false)