lint

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

Hide file tree

Showing 2 changed files with 2 additions and 9 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