Skip to content

Commit 0762f18

Browse files
shiffmancvalenzuela
authored andcommitted
Additional arguments option for SketchRNN (#237)
* Fixing up SketchRNN * arguments fix for SketchRNN interactive * adding build * update version
1 parent cccccbe commit 0762f18

File tree

4 files changed

+12
-8
lines changed

4 files changed

+12
-8
lines changed

dist/ml5.min.js

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

dist/ml5.min.js.map

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

package.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
{
22
"name": "ml5",
3-
"version": "0.1.1",
3+
"version": "0.1.3",
44
"description": "A friendly machine learning library for the web.",
55
"main": "dist/ml5.min.js",
66
"directories": {

src/SketchRNN/index.js

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -62,21 +62,25 @@ class SketchRNN {
6262
return result;
6363
}
6464

65-
async generate(optionsOrCallback, seedOrCallback, cb) {
65+
async generate(optionsOrSeedOrCallback, seedOrCallback, cb) {
6666
let callback;
6767
let options;
6868
let seedStrokes;
6969

70-
if (typeof optionsOrCallback === 'function') {
70+
if (typeof optionsOrSeedOrCallback === 'function') {
7171
options = {};
7272
seedStrokes = [];
73-
callback = optionsOrCallback;
73+
callback = optionsOrSeedOrCallback;
74+
} else if (Array.isArray(optionsOrSeedOrCallback)) {
75+
options = {};
76+
seedStrokes = optionsOrSeedOrCallback;
77+
callback = seedOrCallback;
7478
} else if (typeof seedOrCallback === 'function') {
75-
options = optionsOrCallback || {};
79+
options = optionsOrSeedOrCallback || {};
7680
seedStrokes = [];
7781
callback = seedOrCallback;
7882
} else {
79-
options = optionsOrCallback || {};
83+
options = optionsOrSeedOrCallback || {};
8084
seedStrokes = seedOrCallback || [];
8185
callback = cb;
8286
}

0 commit comments

Comments
 (0)