Change package reference

Author: freeman-lab

lightning/main.py

@@ -138,7 +138,7 @@ def plot(self, data=None, type=None):
             Name of plot (e.g. 'line' or 'scatter')
         """
 
-        from types.plots import Generic
+        from .types.plots import Generic
 
         if not hasattr(self, 'session'):
             self.create_session()