Merge pull request #25 from autonomio/fix_create_synth_data

Fix create synth data
autonomio · Jan 28, 2022 · 613e32c · 613e32c
2 parents f01807a + d918af2
commit 613e32c
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/.github/workflows/ci-push.yml b/.github/workflows/ci-push.yml
@@ -9,8 +9,8 @@ jobs:
     strategy:
       max-parallel: 9
       matrix:
-        python-version: [3.5, 3.6, 3.7]
-        os: [ubuntu-latest, windows-latest, macos-latest]
+        python-version: [3.7, 3.8]
+        os: [ubuntu-latest, macos-latest]
 
     steps:
     - uses: actions/checkout@v1

diff --git a/setup.py b/setup.py
@@ -14,7 +14,7 @@
 URL = 'http://autonom.io'
 LICENSE = 'MIT'
 DOWNLOAD_URL = 'https://github.com/autonomio/wrangle/'
-VERSION = '0.6.9'
+VERSION = '0.6.10'
 
 try:
     from setuptools import setup

diff --git a/wrangle/utils/create_synth_data.py b/wrangle/utils/create_synth_data.py
@@ -22,7 +22,9 @@ def create_synth_data(mode='binary', n=1000, features=20, classes=4):
         x, y = np.random.rand(n, features), np.random.randint(0, classes, n)
 
     elif mode == 'multi_label':
-        x, y = data.make_multilabel_classification(n, features, classes)
+        x, y = data.make_multilabel_classification(n_samples=n,
+                                                   n_features=features, 
+                                                   n_classes=classes)
 
     elif mode == 'continuous':
         x, y = data.make_regression(n, features)