Merge pull request #5 from ttngu207/master

Version lock `marshmallow` and `argschema`, minor bugfix

Author: drewyangdev

ecephys_spike_sorting/scripts/create_input_json.py

@@ -3,7 +3,8 @@
 
 if sys.platform == 'linux':
     import pwd
-from helpers import SpikeGLX_utils
+
+from .helpers import SpikeGLX_utils
 
 import numpy as np
 

setup.py

@@ -31,9 +31,12 @@
         'pandas',
         'GitPython',
         'pillow',
-        'argschema',
+        'argschema==1.*',
+        'marshmallow==2.*',
         'xmljson',
         'xarray',
         'scikit-learn',
+        'python-dotenv',
+        'h5py'
     ],
 )