mirror of
https://github.com/vale981/ray
synced 2025-03-06 02:21:39 -05:00
16 lines
567 B
Bash
16 lines
567 B
Bash
|
#!/bin/bash
|
||
|
|
|
||
|
|
pip install pytest
|
||
|
|
# Uninstall any existing lightgbm_ray repositories
|
||
|
|
pip uninstall -y lightgbm_ray || true
|
||
|
|
|
||
|
|
# Install lightgbm package
|
||
|
|
pip install -U "${LIGHTGBM_RAY_PACKAGE:-lightgbm_ray}"
|
||
|
|
|
||
|
|
# Create test dataset
|
||
|
|
sudo mkdir -p /data || true
|
||
|
|
sudo chown ray:1000 /data || true
|
||
|
|
rm -rf /data/classification.parquet || true
|
||
|
|
cp -R /tmp/ray_tmp_mount/lightgbm_tests ~/lightgbm_tests || echo "Copy failed"
|
||
|
|
python ~/lightgbm_tests/create_test_data.py /data/classification.parquet --seed 1234 --num-rows 1000000 --num-cols 40 --num-partitions 100 --num-classes 2
|