minor style changes

jrzaurin · Feb 16, 2024 · 11ad157 · 11ad157
1 parent 3901e07
commit 11ad157
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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
@@ -1,8 +1,8 @@
 import string
 from random import choices
 
-import torch
 import numpy as np
+import torch
 import pandas as pd
 import pytest