update embedding_dims if new features are added (#358)

sorenmacbeth · Soren Macbeth · web-flow · commit d86bfa61e840 · 2023-12-28T08:10:51.000+05:30
Co-authored-by: Soren Macbeth &lt;soren@abracadaniel.local&gt;
diff --git a/src/pytorch_tabular/tabular_datamodule.py b/src/pytorch_tabular/tabular_datamodule.py
@@ -359,6 +359,7 @@ def preprocess_data(self, data: DataFrame, stage: str = "inference") -> Tuple[Da
             self.config.categorical_dim = (
                 len(self.config.categorical_cols) if self.config.categorical_cols is not None else 0
             )
+            self._inferred_config = self._update_config(self.config)
         # Encoding Categorical Columns
         if len(self.config.categorical_cols) > 0:
             data = self._encode_categorical_columns(data, stage)

Original file line number	Diff line number	Diff line change
`@@ -359,6 +359,7 @@ def preprocess_data(self, data: DataFrame, stage: str = "inference") -> Tuple[Da`
`359`	`359`	`self.config.categorical_dim = (`
`360`	`360`	`len(self.config.categorical_cols) if self.config.categorical_cols is not None else 0`
`361`	`361`	`)`
	`362`	`+ self._inferred_config = self._update_config(self.config)`
`362`	`363`	`# Encoding Categorical Columns`
`363`	`364`	`if len(self.config.categorical_cols) > 0:`
`364`	`365`	`data = self._encode_categorical_columns(data, stage)`