Merge branch 'main' into gdpr

NVIDIA-Merlin · Apr 25, 2024 · f608860 · f608860
2 parents a03eabe + 426e034
commit f608860
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 9 deletions.
diff --git a/.github/actionlint.yaml b/.github/actionlint.yaml
@@ -3,3 +3,4 @@ self-hosted-runner:
   labels:
     - 1GPU
     - 2GPU
+    - linux-amd64-gpu-p100-latest-1
diff --git a/merlin/io/writer.py b/merlin/io/writer.py
@@ -195,7 +195,7 @@ def _add_data_slice(self, df):
         # the df and write out slices.
         if self.shuffle:
             df = shuffle_df(df)
-        int_slice_size = df.shape[0] // self.num_out_files
+        int_slice_size = max(df.shape[0] // self.num_out_files, 1)
         slice_size = int_slice_size if df.shape[0] % int_slice_size == 0 else int_slice_size + 1
         for x in range(self.num_out_files):
             start = x * slice_size

diff --git a/requirements-docs.txt b/requirements-docs.txt
@@ -8,12 +8,12 @@ sphinxcontrib-qthelp==1.0.3
 sphinxcontrib-serializinghtml==1.1.5
 sphinx-multiversion@git+https://github.com/mikemckiernan/sphinx-multiversion.git
 sphinxcontrib-copydirs@git+https://github.com/mikemckiernan/sphinxcontrib-copydirs.git
-recommonmark==0.7.1
+recommonmark~=0.7.1
 Jinja2<3.1
-natsort==8.0.1
-myst-nb==0.17.2
-linkify-it-py==1.0.3
-sphinx-external-toc==0.2.4
-attrs==21.4.0
-sphinx-book-theme==1.0.1
-sphinx_design
+natsort~=8.4.0
+myst-nb~=1.1.0
+linkify-it-py~=2.0.3
+sphinx-external-toc~=1.0.1
+attrs~=23.2.0
+sphinx-book-theme~=1.1.2
+sphinx_design~=0.5.0
-Original file line number
+Diff line change
@@ Expand Up / @@ -3,3 +3,4 @@ self-hosted-runner: @@
       labels:
         - 1GPU
         - 2GPU
+        - linux-amd64-gpu-p100-latest-1