fix: migrate command add new flags option to config

agarsev · Sep 20, 2021 · d7e2f31 · d7e2f31
1 parent 3b4a1e7
commit d7e2f31
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/quevedo/migrate.py b/quevedo/migrate.py
@@ -37,6 +37,9 @@ def list_to_dict(a: Grapheme):
     dataset.config['train_folds'] = [0, 1, 2, 3, 4, 5, 6, 7]
     dataset.config['test_folds'] = [8, 9]
 
+    # Add "flags"
+    dataset.config['flags'] = {"done": "✔️", "problem": "⚠️", "notes": "📝"}
+
 
 @click.command('migrate')
 @click.pass_obj