Skip to content

Commit f8390f0

Browse files
committed
fix saint config import
1 parent eb84fa5 commit f8390f0

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

mambular/configs/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
from .ndtf_config import DefaultNDTFConfig
77
from .node_config import DefaultNODEConfig
88
from .resnet_config import DefaultResNetConfig
9+
from .saint_config import DefaultSAINTConfig
910
from .tabm_config import DefaultTabMConfig
1011
from .tabtransformer_config import DefaultTabTransformerConfig
1112
from .tabularnn_config import DefaultTabulaRNNConfig
@@ -19,6 +20,7 @@
1920
"DefaultNDTFConfig",
2021
"DefaultNODEConfig",
2122
"DefaultResNetConfig",
23+
"DefaultSAINTConfig",
2224
"DefaultTabMConfig",
2325
"DefaultTabTransformerConfig",
2426
"DefaultTabulaRNNConfig",

0 commit comments

Comments
 (0)