Merge pull request #7 from NoahStapp/polish

More small fixes
jupytercalpoly · Aug 6, 2018 · 84158fc · 84158fc
2 parents 2fceb10 + c251baa
commit 84158fc
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 4 deletions.
diff --git a/plyto/keras_accuracy_loss_callback.py b/plyto/keras_accuracy_loss_callback.py
@@ -99,5 +99,7 @@ def on_batch_end(self, batch, logs={}):
                 "accuracy": self.accuracy,
             }
         )
-        if self.total_progress % 256 == 0 or self.total_progress == (self.epochs * self.sample_amount):
+        if self.total_progress % 256 == 0 or self.total_progress == (
+            self.epochs * self.sample_amount
+        ):
             self.plyto.send_data()
diff --git a/plyto/plyto.py b/plyto/plyto.py
@@ -17,7 +17,7 @@ class PlytoAPI:
     :param progress: the current progress of iterating through the dataset
     """
 
-    def __init__(self, spec, size={}, steps={}):
+    def __init__(self, spec, size=0, steps=0):
         self.spec = spec
         self.size = size
         self.total_steps = steps

diff --git a/src/ModelViewWidget.tsx b/src/ModelViewWidget.tsx
@@ -79,9 +79,17 @@ class ModelViewPanel extends React.Component<
   }
 
   render() {
-    if (this.state.updateGraph && this.state.dataSet.length !== 0) {
+    let options = {
+      actions: {
+        export: true,
+        source: false,
+        compiled: false,
+        editor: false
+      }
+    };
+    if (this.state.updateGraph) {
       this.state.spec.forEach(spec => {
-        VegaEmbed('#' + spec['name'], spec).then(res => {
+        VegaEmbed('#' + spec['name'], spec, options).then(res => {
           res.view.insert('dataSet', this.state.dataSet).run();
         });
       });