minor. Fixed duplicate subject-side check for missing entities. Resto…

…red jenkins script.
Accenture · Apr 16, 2019 · 066d8a9 · 066d8a9
1 parent 177ad00
commit 066d8a9
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/ampligraph/evaluation/protocol.py b/ampligraph/evaluation/protocol.py
@@ -417,7 +417,7 @@ def _convert_to_idx(X, ent_to_idx, rel_to_idx, obj_to_idx):
     x_idx_p = np.vectorize(rel_to_idx.get)(X[:, 1])
     x_idx_o = np.vectorize(obj_to_idx.get)(X[:, 2])
 
-    if None in x_idx_s or None in x_idx_s:
+    if None in x_idx_s or None in x_idx_o:
         msg = 'Input triples include one or more entities not present in the training set. ' \
               'Please filter X using evaluation.filter_unseen_entities(), or retrain the model on a training set ' \
               'that includes all the desired distinct entities.'

diff --git a/jenkins.sh b/jenkins.sh
@@ -11,8 +11,6 @@ source activate ampligraph
 # Install library
 conda install tensorflow-gpu==1.12.0
 
-git checkout develop
-
 pip install . -v
 
 # configure dataset location