fix imports

McTavishLab · Nov 13, 2019 · 25f4cae · 25f4cae
1 parent 57370eb
commit 25f4cae
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/physcraper/__init__.py b/physcraper/__init__.py
@@ -11,6 +11,7 @@
 from physcraper.scrape import PhyscraperScrape
 from physcraper.opentree_helpers import generate_ATT_from_phylesystem, OtuJsonDict, get_mrca_ott
 from physcraper.aligntreetax import generate_ATT_from_files
+from physcraper.treetaxon import TreeTax
 
 if sys.version_info < (3,):
     from urllib2 import HTTPError

diff --git a/tests/test_treetaxon_func.py b/tests/test_treetaxon_func.py
@@ -1,7 +1,7 @@
 import os
 import filecmp
 from physcraper import opentree_helpers
-from physcraper.treetaxon import TreeTax
+from physcraper import TreeTax
 
 
 json_file = "tests/data/treetaxon/main.json"
@@ -28,8 +28,6 @@ def test_get_synth_tree():
     resp = opentree_helpers.get_tree_from_synth(ott_ids=ott_ids)
     print resp
 
-
 def test_get_phyle_tree():
     tr = opentree_helpers.get_tree_from_study(study_id='pg_1144', tree_id='tree2324', label_format="name")
 
-