Cannot import lassonet

[barveaditya]

Doing:

from lassonet import LassoNetRegressor

Traceback (most recent call last):

  File "/home/barvead1/.conda/envs/misc_conda27/lib/python3.7/site-packages/IPython/core/interactiveshell.py", line 3552, in run_code
    exec(code_obj, self.user_global_ns, self.user_ns)

  File "/scratch/tmp/34611171.1.default.q/ipykernel_11034/2278597197.py", line 4, in <module>
    from lassonet import LassoNetRegressor

  File "/home/barvead1/.conda/envs/misc_conda27/lib/python3.7/site-packages/lassonet/__init__.py", line 4, in <module>
    from .interfaces import (

  File "<fstring>", line 1
    (current_lambda=)

Infact, I get the same error for importing anything as expected. Any ideas? Is there a specific pytorch version, python version etc?