Merge pull request #14 from csinva/python3.8-support

Python3.8 support

.github/workflows/python-package.yml

@@ -3,19 +3,15 @@
 
 name: tests
 
-on:
-  push:
-    branches: [ master ]
-  pull_request:
-    branches: [ master ]
+on: [push, pull_request]
 
 jobs:
   build:
 
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: [3.7, 3.8]
+        python-version: [3.7, 3.8.5]
 
     steps:
     - uses: actions/checkout@v2
@@ -26,7 +22,7 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        pip install .
+        pip install . --no-cache-dir
     - name: Test with pytest
       run: |
         pytest

tests/brl_test.py

@@ -1,6 +1,8 @@
 import unittest
+from urllib.request import urlretrieve
 
 import numpy as np
+from scipy.io.arff import loadarff
 from sklearn.datasets import fetch_openml
 from sklearn.model_selection import train_test_split
 
@@ -30,9 +32,11 @@ def test_integration_fitting(self):
                           "Triceps skin fold thickness(mm)",
                           "2-Hour serum insulin (mu U/ml)", "Body mass index", "Diabetes pedigree function",
                           "Age (years)"]
-        data = fetch_openml("diabetes")  # get dataset
-        X = data.data
-        y = (data.target == 'tested_positive').astype(np.int)  # labels 0-1
+
+        data = loadarff("tests/test_data/diabetes.arff")
+        data_np = np.array(list(map(lambda x: np.array(list(x)), data[0])))
+        X, y_text = data_np[:, :-1].astype('float32'), data_np[:, -1].astype('str')
+        y = (y_text == 'tested_positive').astype(np.int)  # labels 0-1
 
         X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.75)  # split