Spaces:

ericmichael
/

simon_says_v2

Sleeping

ericmichael commited on Jan 26

Commit

4594826

•

1 Parent(s): a40c76f

update

Files changed (1) hide show

utils/json_loader.py CHANGED Viewed

@@ -21,10 +21,10 @@ class JsonDataLoader:
             ]
         if category == "medium":
-            inputs = [
                 example["inputs"] for example in data if example["category"] == "medium"
             ]
-            outputs = [
                 example["outputs"]
                 for example in data
                 if example["category"] == "medium"

             ]
         if category == "medium":
+            inputs += [
                 example["inputs"] for example in data if example["category"] == "medium"
             ]
+            outputs += [
                 example["outputs"]
                 for example in data
                 if example["category"] == "medium"