diff --git a/pytorch_widedeep/preprocessing/wide_preprocessor.py b/pytorch_widedeep/preprocessing/wide_preprocessor.py
index 72d67a2..c5cdf3a 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 8955cdc..17d71f9 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 33d3820..a466aad 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