Merge pull request EpistasisLab#68 from EpistasisLab/travisfix

travis fix branch
aadu · Feb 12, 2020 · 1830056 · 1830056
2 parents adae7e0 + 97fc98a
commit 1830056
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 6 deletions.
diff --git a/.travis.yml b/.travis.yml
@@ -1,12 +1,14 @@
 language: python
+python:
+    - "3.5"
 virtualenv:
   system_site_packages: true
 env:
   matrix:
     # let's start simple:
-    - PYTHON_VERSION="2.7" LATEST="true"
-    - PYTHON_VERSION="3.6" COVERAGE="true" LATEST="true"
-    - PYTHON_VERSION="3.6" LATEST="true"
+    - PYTHON_VERSION="3.5" LATEST="true"
+    - PYTHON_VERSION="3.5" COVERAGE="true" LATEST="true"
+    - PYTHON_VERSION="3.5" LATEST="true"
 install: source ./ci/.travis_install.sh
 script: bash ./ci/.travis_test.sh
 after_success:

diff --git a/skrebate/vlsrelief.py b/skrebate/vlsrelief.py
@@ -26,7 +26,7 @@ class VLSRelief(BaseEstimator, TransformerMixin):
 
     """
 
-    def __init__(self, core_algorithm, n_features_to_select=2, n_neighbors=100, num_feature_subset=40, size_feature_subset=5, discrete_threshold=10, verbose=False, n_jobs=1):
+    def __init__(self, core_algorithm, n_features_to_select=2, n_neighbors=100, num_feature_subset=20, size_feature_subset=10, discrete_threshold=10, verbose=False, n_jobs=1):
         """Sets up VLSRelief to perform feature selection.
 
         Parameters

diff --git a/tests.py b/tests.py
@@ -142,8 +142,8 @@ def test_vlsrelief_init():
     assert clf.discrete_threshold == 20
     assert clf.verbose == True
     assert clf.n_jobs == 3
-    assert clf.num_feature_subset == 40
-    assert clf.size_feature_subset == 5
+    assert clf.num_feature_subset == 20
+    assert clf.size_feature_subset == 10
 
 
 # Basic Parallelization Tests and Core binary data and discrete feature data testing (Focus on ReliefF only for efficiency)------------------------------------------------------------