Drop Python3.6 support & Fix test (#90)

* Fix wheel & test

* Restore wheels.yml

Co-authored-by: tohtsky <you@example.com>

Author: tohtsky

.github/workflows/wheels.yml

@@ -61,8 +61,7 @@ jobs:
           - os: ubuntu-20.04
             name: manylinux1
             cibw:
-# include python 3.6 because I want to run it on sagemaker notebook instance.
-              build: "cp36* cp37*"
+              build: "cp37*"
               skip: "*musllinux*"
               manylinux_image: manylinux2010
               env: CFLAGS='-march=core-avx-i'

tests/recommenders/test_ials.py

@@ -72,12 +72,12 @@ def test_ials_overfit_cg(test_interaction_data: Dict[str, sps.csr_matrix]) -> No
     X = test_interaction_data["X_small"]
     rec = IALSRecommender(
         X,
-        n_components=4,
+        n_components=3,
         alpha0=100,
         loss_type="ORIGINAL",
-        reg=1e-4,
+        reg=1e-1,
         solver_type="CG",
-        max_cg_steps=4,
+        max_cg_steps=3,
         max_epoch=100,
         nu=0,
     )