Adding in small tweak to the kano-wand-test script

collect_spell_data.py

@@ -1,11 +1,12 @@
-from KanoWandAsync import Shop, Wand
-from KanoWandAsync.constants import *
 import asyncio
 import pandas as pd
-from uuid import uuid4
+from kanowandasync import Shop, Wand
+from kanowandasync.constants import *
 from pathlib import Path
-import os
+from uuid import uuid4
+
 global spell
+
 def save_dataframe(data):
     df = pd.DataFrame(data)
     df.columns = ["x", "y", "pitch", "roll"]
@@ -38,7 +39,7 @@ class DataCollectWand(Wand):
             await self.unsubscribe_position()
 
     async def on_position(self, x, y, pitch, roll):
-        # print(x, y, pitch, roll)
+
         self.data.append([x, y, pitch, roll])
 
 def get_prompt_text(spells):