Merge pull request #69 from LuoXueling/dev-xlluo

Fix some bugs

tabensemb/model/_autogluon/__init__.py

@@ -0,0 +1 @@
+from .multilabel import *

tabensemb/model/_pytorch_tabular/__init__.py

@@ -0,0 +1 @@
+from .mute_track import *

tabensemb/model/_sample/__init__.py

@@ -0,0 +1,2 @@
+from .cat_embed import *
+from .require_model import *

tabensemb/model/_widedeep/__init__.py

@@ -0,0 +1 @@
+from .widedeep_callback import *

tabensemb/model/pytorch_tabular.py

@@ -10,6 +10,7 @@
 from .base import PytorchLightningLossCallback
 from .base import AbstractWrapper
 from typing import Dict, Any
+from packaging import version
 from torch import nn
 import re
 import inspect
@@ -26,6 +27,7 @@ def _new_model(self, model_name, verbose, **kwargs):
         mute_track()
 
         from functools import partialmethod
+        import pytorch_tabular
         from pytorch_tabular.config import ExperimentRunManager
 
         erm_original_init = ExperimentRunManager.__init__
@@ -105,7 +107,11 @@ def _new_model(self, model_name, verbose, **kwargs):
             "GATE": GatedAdditiveTreeEnsembleConfig,
         }
         special_configs = {
-            "NODE": {"embed_categorical": True},
+            "NODE": (
+                {"embed_categorical": True}
+                if version.parse(pytorch_tabular.__version__) < version.parse("1.1.0")
+                else {}
+            ),
         }
         legal_kwargs = {
             key: value

tabensemb/utils/imputers/__init__.py

@@ -0,0 +1 @@
+from .gain import *

tabensemb/utils/processors/__init__.py

@@ -0,0 +1 @@
+from .rfecv import *