diff --git a/.github/actionlint.yaml b/.github/actionlint.yaml index e0fa30b4b..f5f4e4c59 100644 --- 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 index 66c24170d..b4e472a67 100644 --- 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 index 5d19ed444..c615bb670 100644 --- 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