feat: sync selection back to python

iamDecode · iamDecode · commit a557c74a41d6 · 2021-01-31T17:36:04.000+01:00
diff --git a/cvplot/cvplot.py b/cvplot/cvplot.py
@@ -1,5 +1,5 @@
 import ipywidgets as widgets
-from traitlets import Unicode, Dict
+from traitlets import Unicode, Dict, List
 from sklearn.exceptions import NotFittedError
 from joblib import delayed, Parallel, cpu_count
 import numpy as np
@@ -32,6 +32,7 @@ class CVPlot(widgets.DOMWidget):
   _model_module_version = Unicode('^0.1.0').tag(sync=True)
 
   value = Dict().tag(sync=True)
+  selection = List().tag(sync=True)
 
   def __init__(self, X, y, features, model, explainer=None, target_index=0, max=500, **kwargs):
     super().__init__(**kwargs)
diff --git a/js/lib/components/Contributions.vue b/js/lib/components/Contributions.vue
@@ -89,6 +89,8 @@
         this.$children.forEach(component => component.updatePossible(values))
       },
       updateSelected(values) {
+        this.widget.model.set('selection', values)
+        this.widget.model.save_changes()
         this.$children.forEach(component => component.updateSelected(values))
       }
     }
diff --git a/js/lib/cvplot.js b/js/lib/cvplot.js
@@ -29,7 +29,8 @@ var Model = widgets.DOMWidgetModel.extend({
     _view_module : 'cvplot',
     _model_module_version : '0.1.0',
     _view_module_version : '0.1.0',
-    value : {}
+    value : {},
+    selection : []
   })
 });
 
@@ -52,8 +53,10 @@ var View = widgets.DOMWidgetView.extend({
   },
 
   value_changed: function() {
+    const widget = this
     const value = this.model.get('value')
     vm.$children[0].$nextTick(function() {
+      this.$refs.contributions.widget = widget
     	this.$refs.contributions.update(value)
     })
   }

Original file line number	Diff line number	Diff line change
`@@ -89,6 +89,8 @@`
`89`	`89`	`this.$children.forEach(component => component.updatePossible(values))`
`90`	`90`	`},`
`91`	`91`	`updateSelected(values) {`
	`92`	`+ this.widget.model.set('selection', values)`
	`93`	`+ this.widget.model.save_changes()`
`92`	`94`	`this.$children.forEach(component => component.updateSelected(values))`
`93`	`95`	`}`
`94`	`96`	`}`