jit_geometric_problems.txt

Traceback (most recent call last):
  File "mnist_nn_conv.py", line 71, in <module>
    model = Net().to(device)
  File "mnist_nn_conv.py", line 45, in __init__
    self.conv1 = conv1.jittable(x=init_data.x, edge_index=init_data.edge_index, edge_attr=init_data.edge_attr)
  File "/Users/lagray/pytorch_work/pytorch_geometric/torch_geometric/nn/conv/message_passing.py", line 608, in jittable
    out = torch.jit.script(out)
  File "/anaconda3/envs/torch/lib/python3.7/site-packages/torch/jit/__init__.py", line 1261, in script
    return torch.jit._recursive.create_script_module(obj, torch.jit._recursive.infer_methods_to_compile)
  File "/anaconda3/envs/torch/lib/python3.7/site-packages/torch/jit/_recursive.py", line 305, in create_script_module
    return create_script_module_impl(nn_module, concrete_type, stubs_fn)
  File "/anaconda3/envs/torch/lib/python3.7/site-packages/torch/jit/_recursive.py", line 361, in create_script_module_impl
    create_methods_from_stubs(concrete_type, stubs)
  File "/anaconda3/envs/torch/lib/python3.7/site-packages/torch/jit/_recursive.py", line 279, in create_methods_from_stubs
    concrete_type._create_methods(defs, rcbs, defaults)
  File "/anaconda3/envs/torch/lib/python3.7/site-packages/torch/jit/_recursive.py", line 587, in compile_unbound_method
    create_methods_from_stubs(concrete_type, (stub,))
  File "/anaconda3/envs/torch/lib/python3.7/site-packages/torch/jit/_recursive.py", line 279, in create_methods_from_stubs
    concrete_type._create_methods(defs, rcbs, defaults)
  File "/anaconda3/envs/torch/lib/python3.7/site-packages/torch/jit/_recursive.py", line 587, in compile_unbound_method
    create_methods_from_stubs(concrete_type, (stub,))
  File "/anaconda3/envs/torch/lib/python3.7/site-packages/torch/jit/_recursive.py", line 279, in create_methods_from_stubs
    concrete_type._create_methods(defs, rcbs, defaults)
  File "/anaconda3/envs/torch/lib/python3.7/site-packages/torch/jit/_recursive.py", line 583, in compile_unbound_method
    stub = make_stub(fn)
  File "/anaconda3/envs/torch/lib/python3.7/site-packages/torch/jit/_recursive.py", line 33, in make_stub
    rcb = _jit_internal.createResolutionCallbackFromClosure(func)
  File "/anaconda3/envs/torch/lib/python3.7/site-packages/torch/_jit_internal.py", line 151, in createResolutionCallbackFromClosure
    closure = get_closure(fn)
  File "/anaconda3/envs/torch/lib/python3.7/site-packages/torch/_jit_internal.py", line 96, in get_closure
    captures.update(fn.__globals__)
AttributeError: type object 'type' has no attribute '__globals__'