Fix ImportError (#46) (#47)

natekupp · Jan 15, 2020 · a4ab861 · a4ab861
1 parent 0e1552a
commit a4ab861
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 4 deletions.
diff --git a/CHANGES.md b/CHANGES.md
@@ -1,5 +1,9 @@
 # Changes
 
+## 2.0.1
+
+- Fix ImportError introduced in 2.0.0
+
 ## 2.0.0
 
 - Added a Makefile

diff --git a/ffx/api.py b/ffx/api.py
@@ -6,9 +6,9 @@
 
 
 def run(train_X, train_y, test_X, test_y, varnames=None, verbose=False):
-    from ffx import core
+    from .core import MultiFFXModelFactory
 
-    return core.MultiFFXModelFactory().build(train_X, train_y, test_X, test_y, varnames, verbose)
+    return MultiFFXModelFactory().build(train_X, train_y, test_X, test_y, varnames, verbose)
 
 
 class FFXRegressor(BaseEstimator, RegressorMixin):

diff --git a/ffx/cli.py b/ffx/cli.py
@@ -3,8 +3,8 @@
 import click
 import numpy as np
 import pandas as pd
-from ffx.api import run
 
+from .api import run
 from .time_utils import time_execution_scope
 
 

diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name='ffx',
-    version='2.0.0',
+    version='2.0.1',
     author='Trent McConaghy',
     author_email='[email protected]',
     maintainer='Nate Kupp',