correct error in tests

Author: statmlben

rehline/_path_sol.py

@@ -76,9 +76,6 @@ def plqERM_Ridge_path_sol(
     return_time : bool, default=True
         If True, return timing information for each value of `C`.
 
-    plot_path : bool, default=False
-        If True, generate a plot of the coefficient paths as a function of `C`.
-
     Returns
     -------
     Cs : ndarray of shape (n_Cs,)

setup.py

@@ -6,7 +6,7 @@
 from pybind11.setup_helpers import Pybind11Extension, build_ext
 from setuptools import setup
 
-__version__ = "0.1.0"
+__version__ = "0.1.1"
 
 # The main interface is through Pybind11Extension.
 # * You can add cxx_std=11/14/17, and then build_ext can be removed.

tests/_test_fairsvm.py

@@ -20,7 +20,7 @@
 
 # specific the param of FairSVM
 X_sen = X[:,sen_idx]
-A = np.repeat([X_sen @ X], repeats=[2], axis=0) / n
+A = np.repeat([X_sen.flatten() @ X], repeats=[2], axis=0) / n
 A[1] = -A[1]
 # suppose the fair tolerance is 0.01
 b = np.array([.01, .01])

tests/_test_path_sol.py

@@ -24,7 +24,6 @@
 #     warm_start=False,
 #     constraint=[],
 #     return_time=True,
-#     plot_path=True
 # )
 
 # calculate with warm_start=True
@@ -38,7 +37,6 @@
     warm_start=True,
     constraint=[],
     return_time=True,
-    plot_path=True
 )
 
 

tests/_test_svr.py

@@ -1,7 +1,7 @@
 ## Test SVR on simulated dataset
 import numpy as np
 
-from rehline import ReHLine
+from rehline import ReHLine, plqERM_Ridge
 
 np.random.seed(1024)
 # simulate regression dataset