adding corpus_file option

ddangelov · Jun 26, 2020 · 2484e4a · 2484e4a
1 parent 0bbf44f
commit 2484e4a
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/top2vec/tests/test_top2vec.py b/top2vec/tests/test_top2vec.py
@@ -1,11 +1,5 @@
 import pytest
-
-# from top2vec import Top2Vec
-import sys
-
-sys.path.insert(1, '/Users/dimo_angelov/PycharmProjects/Top2Vec')
-from Top2Vec import Top2Vec
-
+from top2vec import Top2Vec
 from sklearn.datasets import fetch_20newsgroups
 from sklearn.metrics.pairwise import cosine_similarity
 import numpy as np