Updated travis and test helper.

Author: chcost

codeflare/pipelines/tests/test_helper.py

@@ -6,6 +6,7 @@
 from sklearn.preprocessing import StandardScaler, OneHotEncoder
 from sklearn.ensemble import RandomForestClassifier, GradientBoostingClassifier
 
+import pathlib
 
 def get_pipeline(train) -> dm.Pipeline:
     imputer = SimpleImputer(strategy='median')
@@ -45,12 +46,12 @@ def get_pipeline(train) -> dm.Pipeline:
 
 
 def get_data():
-    train = pd.read_csv('../../../resources/data/train_ctrUa4K.csv')
+    train = pd.read_csv(str(pathlib.Path(__file__).parent.absolute())+'/../../../resources/data/train_ctrUa4K.csv')
     train = train.drop('Loan_ID', axis=1)
 
     X = train.drop('Loan_Status', axis=1)
     y = train['Loan_Status']
     from sklearn.model_selection import train_test_split
     X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)
 
-    return X_train, X_test, y_train, y_test
+    return X_train, X_test, y_train, y_test

setup.py

@@ -28,7 +28,6 @@
     name='codeflare-pipelines',
     version='1.0.0',
     packages=['codeflare', 'codeflare.pipelines', 'codeflare.pipelines.tests'],
-    package_data={'codeflare.pipelines.tests': ['*.csv']},
     install_requires=[
         'ray[default,serve,k8s]>=1.3.0'
     ],