lint

rapidsai · Dec 19, 2023 · a1b8a31 · a1b8a31
1 parent d3c36c7
commit a1b8a31
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 18 deletions.
diff --git a/crossfit/backend/torch/op/base.py b/crossfit/backend/torch/op/base.py
@@ -13,11 +13,11 @@
 # limitations under the License.
 
 import gc
+from typing import Optional
 
 import cudf
 import cupy as cp
 import torch
-from typing import Optional
 
 from crossfit.backend.cudf.series import (
     create_list_series_from_2d_ar,

diff --git a/crossfit/backend/torch/op/embed.py b/crossfit/backend/torch/op/embed.py
@@ -12,14 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import gc
-
-import cudf
-import cupy as cp
-import torch
-
-from crossfit.backend.cudf.series import create_list_series_from_2d_ar
-from crossfit.backend.torch.loader import DEFAULT_BATCH_SIZE, InMemoryLoader, SortedSeqLoader
+from crossfit.backend.torch.loader import DEFAULT_BATCH_SIZE
 from crossfit.backend.torch.model import Model
 from crossfit.backend.torch.op.base import Predictor
 

diff --git a/tests/examples/test_scripts.py b/tests/examples/test_scripts.py
@@ -1,14 +1,15 @@
-import os
-import runpy
-import shutil
-import sys
-import tempfile
-from uuid import uuid4
-
-import cudf
 import pytest
 
-from crossfit.dataset.load import load_dataset
+cudf = pytest.importorskip("cudf")
+
+import os  # noqa: E402
+import runpy  # noqa: E402
+import shutil  # noqa: E402
+import sys  # noqa: E402
+import tempfile  # noqa: E402
+from uuid import uuid4  # noqa: E402
+
+from crossfit.dataset.load import load_dataset  # noqa: E402
 
 examples_dir = os.path.join(os.path.dirname(os.path.realpath(__file__)), "..", "..", "examples")