Update benchmarks to work properly in the absence of networkx

Author: pavlin-policar

benchmarks/benchmark_tsne.py

@@ -8,19 +8,20 @@
 import fire
 import matplotlib.pyplot as plt
 import numpy as np
-from MulticoreTSNE import MulticoreTSNE
+# from MulticoreTSNE import MulticoreTSNE
 from sklearn.decomposition import PCA
 from sklearn.manifold import TSNE as SKLTSNE
 from sklearn.model_selection import train_test_split
 
-from tsne.affinity import NxGraphAffinities
-from tsne.callbacks import ErrorLogger, ErrorApproximations
-from tsne.tsne import TSNE, TSNEEmbedding
+from fastTSNE.callbacks import ErrorLogger, ErrorApproximations
+from fastTSNE.tsne import TSNE, TSNEEmbedding
 
 try:
     import networkx as nx
+    from fastTSNE.affinity import NxGraphAffinities
 except ImportError:
     nx = None
+    NxGraphAffinities = None
 
 
 FILE_DIR = dirname(abspath(__file__))