fix linting errors

NVIDIA-Merlin · Jan 17, 2024 · 903f17f · 903f17f
1 parent 622a43d
commit 903f17f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/merlin/core/compat/__init__.py b/merlin/core/compat/__init__.py
@@ -101,7 +101,6 @@ def device_mem_size(kind="total", cpu=False):
     return pynvml_mem_size(kind=kind)
 
 
-
 try:
     import numpy
 except ImportError:

diff --git a/merlin/io/writer.py b/merlin/io/writer.py
@@ -196,7 +196,11 @@ def _add_data_slice(self, df):
         if self.shuffle:
             df = shuffle_df(df)
         int_slice_size = df.shape[0] // self.num_out_files
-        slice_size = int_slice_size if int_slice_size > 0 and df.shape[0] % int_slice_size == 0 else int_slice_size + 1
+        slice_size = (
+            int_slice_size
+            if int_slice_size > 0 and df.shape[0] % int_slice_size == 0
+            else int_slice_size + 1
+        )
         for x in range(self.num_out_files):
             start = x * slice_size
             end = start + slice_size
-Original file line number
+Diff line change
@@ Expand Up / @@ -101,7 +101,6 @@ def device_mem_size(kind="total", cpu=False): @@
         return pynvml_mem_size(kind=kind)
     try:
         import numpy
     except ImportError:
@@ Expand Down @@