Some requirements are added to the requirement.txt and some unnecessa…

…ry lines are removed from main
priba · May 29, 2017 · dd54611 · dd54611
1 parent 954a18f
commit dd54611
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/main.py b/main.py
@@ -19,13 +19,9 @@
 import time
 import argparse
 import os
-import sys
 import numpy as np
 
 # Our Modules
-reader_folder = os.path.realpath(os.path.abspath('..'))
-if reader_folder not in sys.path:
-    sys.path.append(reader_folder)
 import datasets
 from datasets import utils
 from models.MPNN import MPNN

diff --git a/requirements.txt b/requirements.txt
@@ -3,3 +3,5 @@ torchvision
 numpy
 wget
 networkx
+joblib
+tensorboard
-Original file line number
+Diff line change
@@ Expand Up / @@ -3,3 +3,5 @@ torchvision @@
     numpy
     wget
     networkx
+    joblib
+    tensorboard