From bae48f28d13bcb8daf732af4c83b11c158a4df84 Mon Sep 17 00:00:00 2001
From: David Bau <david.bau@gmail.com>
Date: Thu, 17 Mar 2022 19:50:19 -0400
Subject: [PATCH] Fix imports, add Range step.

---
 torchkit/labwidget.py  | 7 +++++--
 torchkit/plotwidget.py | 4 ++--
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/torchkit/labwidget.py b/torchkit/labwidget.py
index 79d2f29..e92ef25 100644
--- a/torchkit/labwidget.py
+++ b/torchkit/labwidget.py
@@ -43,6 +43,8 @@ DEALINGS IN THE SOFTWARE.
 
 """
 
+import base64
+import io
 import json
 import html
 import re
@@ -621,12 +623,13 @@ class Textbox(Widget):
 
 
 class Range(Widget):
-    def __init__(self, value=50, min=0, max=100, **kwargs):
+    def __init__(self, value=50, min=0, max=100, step=1, **kwargs):
         super().__init__(**kwargs)
         # databinding is defined using Property objects.
         self.value = Property(value)
         self.min = Property(min)
         self.max = Property(max)
+        self.step = Property(step)
 
     def widget_js(self):
         # Note that the 'input' event would enable during-drag feedback,
@@ -644,7 +647,7 @@ class Range(Widget):
 
     def widget_html(self):
         return f'''<input {self.std_attrs()} type="range" value="{
-            self.value}" min="{self.min}" max="{self.max}">'''
+            self.value}" min="{self.min}" max="{self.max}" step="{self.step}">'''
 
 
 class ColorPicker(Widget):
diff --git a/torchkit/plotwidget.py b/torchkit/plotwidget.py
index 62a0701..67543b8 100644
--- a/torchkit/plotwidget.py
+++ b/torchkit/plotwidget.py
@@ -1,8 +1,8 @@
-from .labwidget import ImageWidget, Property
+from .labwidget import Image, Property
 import matplotlib, matplotlib.pyplot
 import inspect
 
-class PlotWidget(ImageWidget):
+class PlotWidget(Image):
     """
     A widget to create interactive matplotlib plots by defining a simple function.
     Example of usage: