formatting

Author: AnFreTh

mambular/arch_utils/layer_utils/embedding_layer.py

@@ -156,7 +156,11 @@ def forward(self, num_features, cat_features, emb_features):
         # Process categorical embeddings
         if self.cat_embeddings and cat_features is not None:
             cat_embeddings = [
-                emb(cat_features[i]) if emb(cat_features[i]).ndim == 3 else emb(cat_features[i]).unsqueeze(1)
+                (
+                    emb(cat_features[i])
+                    if emb(cat_features[i]).ndim == 3
+                    else emb(cat_features[i]).unsqueeze(1)
+                )
                 for i, emb in enumerate(self.cat_embeddings)
             ]
 

mambular/base_models/tabm.py

@@ -22,8 +22,6 @@ def __init__(
         # Pass config to BaseModel
         super().__init__(config=config, **kwargs)
 
-        
-
         # Save hparams including config attributes
         self.save_hyperparameters(ignore=["feature_information"])
         if not self.hparams.average_ensembles: