update

luochang212 · May 24, 2024 · 7a72dfd · 7a72dfd
1 parent 0a2506a
commit 7a72dfd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/nox.yml b/.github/workflows/nox.yml
@@ -36,7 +36,6 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           pip install nox
-          pip install flake8-import-order
 
       - name: Run nox
         run: nox --non-interactive --error-on-missing-interpreter
diff --git a/tests/test_preprocessing.py b/tests/test_preprocessing.py
@@ -1,11 +1,12 @@
 from flameai.preprocessing import DataLoader, gen_scale_pos_weight, label_encoder
+import numpy as np
 import pandas as pd
 
 
 def test_label_encoder():
     df = pd.DataFrame({'a': ['a', 'b', 'c', 'a', 'b', 'c'],
                        'b': [1, 2, 3, 2, 1, 0]})
-    result = label_encoder(df)
+    result = label_encoder(df).reset_index(drop=True)
     assert result.equals(pd.DataFrame({'a': [0, 1, 2, 0, 1, 2],
                                        'b': [1, 2, 3, 2, 1, 0]}))