Replace sklearn with scikit-learn (#741)

GoogleCloudPlatform · Nov 10, 2022 · ef1e92f · ef1e92f
1 parent ba604e6
commit ef1e92f
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 6 deletions.
diff --git a/tests/pytorch/nightly/dlrm.libsonnet b/tests/pytorch/nightly/dlrm.libsonnet
@@ -176,7 +176,7 @@ local utils = import 'templates/utils.libsonnet';
   local tpuVm = common.PyTorchTpuVmMixin {
     tpuSettings+: {
       tpuVmExtraSetup: |||
-        pip3 install tqdm sklearn tensorboardX google-cloud-storage
+        pip3 install tqdm scikit-learn tensorboardX google-cloud-storage
         git clone -b tpu-xrt --single-branch https://github.com/darisoy/dlrm.git dlrm-xrt/
         echo 'export PATH=~/.local/bin:$PATH' >> ~/.bash_profile
       |||,

diff --git a/tests/pytorch/nightly/hf-glue.libsonnet b/tests/pytorch/nightly/hf-glue.libsonnet
@@ -22,7 +22,7 @@ local utils = import 'templates/utils.libsonnet';
     git clone https://github.com/huggingface/transformers.git
     cd transformers && pip install .
     git log -1
-    pip install datasets evaluate sklearn
+    pip install datasets evaluate scikit-learn
     python examples/pytorch/xla_spawn.py \
       --num_cores 8 \
       examples/pytorch/text-classification/run_glue.py \

diff --git a/tests/pytorch/nightly/hf-lm.libsonnet b/tests/pytorch/nightly/hf-lm.libsonnet
@@ -22,7 +22,7 @@ local utils = import 'templates/utils.libsonnet';
     git clone https://github.com/huggingface/transformers.git
     cd transformers && pip install .
     git log -1
-    pip install datasets evaluate sklearn
+    pip install datasets evaluate scikit-learn
     python examples/pytorch/xla_spawn.py \
       --num_cores 8 \
       examples/pytorch/language-modeling/run_mlm.py \

diff --git a/tests/pytorch/r1.13/dlrm.libsonnet b/tests/pytorch/r1.13/dlrm.libsonnet
@@ -176,7 +176,7 @@ local utils = import 'templates/utils.libsonnet';
   local tpuVm = common.PyTorchTpuVmMixin {
     tpuSettings+: {
       tpuVmExtraSetup: |||
-        pip3 install tqdm sklearn tensorboardX google-cloud-storage
+        pip3 install tqdm scikit-learn tensorboardX google-cloud-storage
         git clone -b tpu-xrt --single-branch https://github.com/darisoy/dlrm.git dlrm-xrt/
         echo 'export PATH=~/.local/bin:$PATH' >> ~/.bash_profile
       |||,

diff --git a/tests/pytorch/r1.13/hf-glue.libsonnet b/tests/pytorch/r1.13/hf-glue.libsonnet
@@ -22,7 +22,7 @@ local utils = import 'templates/utils.libsonnet';
     git clone https://github.com/huggingface/transformers.git
     cd transformers && pip install .
     git log -1
-    pip install datasets evaluate sklearn
+    pip install datasets evaluate scikit-learn
     python examples/pytorch/xla_spawn.py \
       --num_cores 8 \
       examples/pytorch/text-classification/run_glue.py \

diff --git a/tests/pytorch/r1.13/hf-lm.libsonnet b/tests/pytorch/r1.13/hf-lm.libsonnet
@@ -22,7 +22,7 @@ local utils = import 'templates/utils.libsonnet';
     git clone https://github.com/huggingface/transformers.git
     cd transformers && pip install .
     git log -1
-    pip install datasets evaluate sklearn
+    pip install datasets evaluate scikit-learn
     python examples/pytorch/xla_spawn.py \
       --num_cores 8 \
       examples/pytorch/language-modeling/run_mlm.py \