From 11ad157158bb8e7143579f50d7b4b7ffb3346d45 Mon Sep 17 00:00:00 2001 From: Javier Date: Fri, 16 Feb 2024 14:42:01 +0000 Subject: [PATCH] minor style changes --- pytorch_widedeep/preprocessing/wide_preprocessor.py | 2 +- tests/test_bayesian_models/test_bayes_tab2vec/test_b_t2v.py | 2 +- tests/test_tab2vec/test_t2v.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/pytorch_widedeep/preprocessing/wide_preprocessor.py b/pytorch_widedeep/preprocessing/wide_preprocessor.py index 72d67a21..c5cdf3af 100644 --- a/pytorch_widedeep/preprocessing/wide_preprocessor.py +++ b/pytorch_widedeep/preprocessing/wide_preprocessor.py @@ -145,7 +145,7 @@ def inverse_transform(self, encoded: np.ndarray) -> pd.DataFrame: """ decoded = pd.DataFrame(encoded, columns=self.wide_crossed_cols) - if pd.__version__ >= '2.1.0': + if pd.__version__ >= "2.1.0": decoded = decoded.map(lambda x: self.inverse_encoding_dict[x]) else: decoded = decoded.applymap(lambda x: self.inverse_encoding_dict[x]) diff --git a/tests/test_bayesian_models/test_bayes_tab2vec/test_b_t2v.py b/tests/test_bayesian_models/test_bayes_tab2vec/test_b_t2v.py index 8955cdc0..17d71f9f 100644 --- a/tests/test_bayesian_models/test_bayes_tab2vec/test_b_t2v.py +++ b/tests/test_bayesian_models/test_bayes_tab2vec/test_b_t2v.py @@ -1,8 +1,8 @@ import string from random import choices -import torch import numpy as np +import torch import pandas as pd import pytest diff --git a/tests/test_tab2vec/test_t2v.py b/tests/test_tab2vec/test_t2v.py index 33d38202..a466aadf 100644 --- a/tests/test_tab2vec/test_t2v.py +++ b/tests/test_tab2vec/test_t2v.py @@ -1,8 +1,8 @@ import string from random import choices -import torch import numpy as np +import torch import pandas as pd import pytest