pytorch_tabnet 3.1.1导入augmentations报ModuleNotFoundError

pytorch-tabnet 3.1.1版本的PyPI官方发行包存在打包疏漏：该版本的源码中确实包含augmentations模块与ClassificationSMOTE类，但维护者发布时未将augmentations相关文件纳入打包清单，导致通过pip安装的3.1.1版本包内缺失对应模块，导入时直接抛出ModuleNotFoundError，该问题和Colab运行环境、安装操作流程无关。

任选一种方案操作即可修复：

• 清理原有安装残留后，直接从官方代码仓库拉取3.1.1版本源码安装，在Colab代码单元执行以下命令：

!pip uninstall -y pytorch-tabnet
!pip install git+https://github.com/dreamquark-ai/tabnet.git@v3.1.1#egg=pytorch-tabnet

命令执行完成后重启Colab运行时内核，即可正常导入ClassificationSMOTE类。

• 升级到后续修复了该打包问题的正式版本，安装完成后重启内核即可正常使用对应功能。
• 手动补全缺失文件：定位到Colab环境中pytorch_tabnet的安装目录（默认路径为/usr/local/lib/python<对应Python版本>/dist-packages/pytorch_tabnet/），将官方源码中的augmentations.py文件上传至该目录，无需重装即可正常导入。

注意：如果之前安装过其他版本的pytorch-tabnet，必须先完全卸载旧版本再操作，避免缓存文件冲突导致导入失败。

内容的提问来源于stack exchange，提问作者Abdelkabir MENANI