[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

src/pytorch_tabular/config/config.py

@@ -192,9 +192,9 @@ class DataConfig:
     )
 
     def __post_init__(self):
-        assert len(self.categorical_cols) + len(self.continuous_cols) + len(self.date_columns) > 0, (
-            "There should be at-least one feature defined in categorical, continuous, or date columns"
-        )
+        assert (
+            len(self.categorical_cols) + len(self.continuous_cols) + len(self.date_columns) > 0
+        ), "There should be at-least one feature defined in categorical, continuous, or date columns"
         _validate_choices(self)
         if os.name == "nt" and self.num_workers != 0:
             print("Windows does not support num_workers > 0. Setting num_workers to 0")
@@ -255,9 +255,9 @@ class InferredConfig:
 
     def __post_init__(self):
         if self.embedding_dims is not None:
-            assert all((isinstance(t, Iterable) and len(t) == 2) for t in self.embedding_dims), (
-                "embedding_dims must be a list of tuples (cardinality, embedding_dim)"
-            )
+            assert all(
+                (isinstance(t, Iterable) and len(t) == 2) for t in self.embedding_dims
+            ), "embedding_dims must be a list of tuples (cardinality, embedding_dim)"
             self.embedded_cat_dim = sum([t[1] for t in self.embedding_dims])
         else:
             self.embedded_cat_dim = 0

src/pytorch_tabular/ssl_models/base_model.py

@@ -85,9 +85,9 @@ def __init__(
         self._setup_metrics()
 
     def _setup_encoder_decoder(self, encoder, encoder_config, decoder, decoder_config, inferred_config):
-        assert (encoder is not None) or (encoder_config is not None), (
-            "Either encoder or encoder_config must be provided"
-        )
+        assert (encoder is not None) or (
+            encoder_config is not None
+        ), "Either encoder or encoder_config must be provided"
         # assert (decoder is not None) or (decoder_config is not None),
         # "Either decoder or decoder_config must be provided"
         if encoder is not None: