Fixed Python3.6 errors; added missing dependency.

4AI · Sep 4, 2017 · 39a6b3d · 39a6b3d
1 parent 9344e1c
commit 39a6b3d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -13,6 +13,7 @@ This repo supports Linux and Python installation via Anaconda.
 1. Install [PyTorch](https://github.com/pytorch/pytorch) using [Anaconda](https://www.continuum.io/downloads)
 2. Install the requirements `pip install -r requirements`
 3. Run the preprocessing script for WN18RR, FB15k-237, and YAGO3-10: `sh preprocess.sh`
+4. (Optional) Install spacy English models with `python -m spacy download en`.
 4. You can now run the model
 
 ## Running a model

diff --git a/requirements.txt b/requirements.txt
@@ -1 +1,2 @@
 -e git://github.com/TimDettmers/spodernet.git#egg=spodernet
+-e git://github.com/TimDettmers/bashmagic.git#egg=bashmagic
diff --git a/wrangle_KG.py b/wrangle_KG.py
@@ -7,7 +7,7 @@
 import requests
 import json
 import urllib
-import cPickle as pickle
+import pickle
 import os
 import numpy as np
 import operator
Original file line number	Diff line number	Diff line change
		@@ -1 +1,2 @@
		-e git://github.com/TimDettmers/spodernet.git#egg=spodernet
		-e git://github.com/TimDettmers/bashmagic.git#egg=bashmagic