Install does not include augmentations so census example fails

parsifal9 commented 2 years ago

What is the current behavior?

conda install -c conda-forge pytorch-tabnet

then working through the census example https://github.com/dreamquark-ai/tabnet/blob/develop/census_example.ipynb I get

>>> from pytorch_tabnet.augmentations import ClassificationSMOTE
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ModuleNotFoundError: No module named 'pytorch_tabnet.augmentations'

Expected behavior

expect the worked example to work

Screenshots

none

Other relevant information: poetry version: not using poetry python version: 3.9.5 Operating System: Linux (Ubuntu 18.04) Additional tools: conda 4.13.0

Additional context

Some of the other worked examples fail the same way. Some (like the regression example) don't use pytorch_tabnet.augmentations and they work.

Optimox commented 2 years ago

Hello @parsifal9,

This is not a bug, augmentations exist only on the development branch. You can specify to install the development with !pip install "git+https://github.com/dreamquark-ai/tabnet.git@develop#egg=pytorch_tabnet" --upgrade

The next release is coming soon.

parsifal9 commented 2 years ago

thanks. Next time I will check the development branch first.

dreamquark-ai / tabnet

Install does not include augmentations so census example fails #401