Skip to content

Commit 7882799

Browse files
authored
Merge branch 'mlcommons:dev' into dev
2 parents d794d3a + 2022d39 commit 7882799

5 files changed

Lines changed: 17 additions & 7 deletions

File tree

script/get-generic-python-lib/README.md

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
# README for get-generic-python-lib
2-
This README is automatically generated. Add custom content in [info.md](info.md). Please follow the [script execution document](https://docs.mlcommons.org/mlcflow/targets/script/execution-flow/) to understand more about the MLC script execution.
2+
This README is automatically generated. Create and add custom content in info.md. Please follow the [script execution document](https://docs.mlcommons.org/mlcflow/targets/script/execution-flow/) to understand more about the MLC script execution.
33

44
`mlcflow` stores all local data under `$HOME/MLC` by default. So, if there is space constraint on the home directory and you have more space on say `/mnt/$USER`, you can do
55
```
@@ -51,10 +51,12 @@ mlcr get generic-python-lib
5151
| `--output` | Output from the script passed using the env key `MLC_OUTPUT` | | `` |
5252
| `--outdirname` | The directory to store the script output | | `cache directory ($HOME/MLC/repos/local/cache/<>) if the script is cacheable or else the current directory` |
5353
| `--outbasename` | The output file/folder name | | `` |
54+
| `--search_folder_path` | The folder path where executables of a given script need to be searched. Search is done recursively upto 4 levels. | | `` |
5455
| `--name` | | | `` |
5556
| `--extra_cache_tags` | Extra cache tags to be added to the cached entry when the script results are saved | | `` |
5657
| `--skip_compile` | Skip compilation | | `False` |
5758
| `--skip_run` | Skip run | | `False` |
59+
| `--skip_sudo` | Skip SUDO detection | | `False` |
5860
| `--accept_license` | Accept the required license requirement to run the script | | `False` |
5961
| `--skip_system_deps` | Skip installing any system dependencies | | `False` |
6062
| `--git_ssh` | Use SSH for git repos | | `False` |
@@ -89,6 +91,7 @@ mlcr get generic-python-lib
8991
- `find_links_url.#` _(# can be substituted dynamically)_
9092
- `google-api-python-client`
9193
- `google-auth-oauthlib`
94+
- `google-genai`
9295
- `google-generativeai`
9396
- `groq`
9497
- `huggingface_hub`

script/get-generic-python-lib/meta.yaml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -171,6 +171,11 @@ variations:
171171
MLC_GENERIC_PYTHON_PACKAGE_NAME: google-generativeai
172172
new_env_keys:
173173
- MLC_GOOGLE_GENERATIVEAI_VERSION
174+
google-genai:
175+
env:
176+
MLC_GENERIC_PYTHON_PACKAGE_NAME: google-genai
177+
new_env_keys:
178+
- MLC_GOOGLE_GENAI_VERSION
174179
groq:
175180
env:
176181
MLC_GENERIC_PYTHON_PACKAGE_NAME: groq

script/get-huggingface-cli/customize.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ def preprocess(i):
1818
hf_cli = _get_hf_cli()
1919
env['MLC_HF_LOGIN_CMD'] = (
2020
f"git config --global credential.helper store && "
21-
f"{hf_cli} login --token {env['MLC_HF_TOKEN']} --add-to-git-credential\n"
21+
f"{hf_cli} auth login --token {env['MLC_HF_TOKEN']} --add-to-git-credential\n"
2222
)
2323
elif is_true(str(env.get('MLC_HF_DO_LOGIN'))):
2424
env['MLC_HF_LOGIN_CMD'] = f"""git config --global credential.helper store && huggingface-cli login

script/get-preprocessed-dataset-nuscenes/README.md

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
# README for get-preprocessed-dataset-nuscenes
2-
This README is automatically generated. Add custom content in [info.md](info.md). Please follow the [script execution document](https://docs.mlcommons.org/mlcflow/targets/script/execution-flow/) to understand more about the MLC script execution.
2+
This README is automatically generated. Create and add custom content in info.md. Please follow the [script execution document](https://docs.mlcommons.org/mlcflow/targets/script/execution-flow/) to understand more about the MLC script execution.
33

44
`mlcflow` stores all local data under `$HOME/MLC` by default. So, if there is space constraint on the home directory and you have more space on say `/mnt/$USER`, you can do
55
```
@@ -45,10 +45,12 @@ No script specific inputs
4545
| `--output` | Output from the script passed using the env key `MLC_OUTPUT` | | `` |
4646
| `--outdirname` | The directory to store the script output | | `cache directory ($HOME/MLC/repos/local/cache/<>) if the script is cacheable or else the current directory` |
4747
| `--outbasename` | The output file/folder name | | `` |
48+
| `--search_folder_path` | The folder path where executables of a given script need to be searched. Search is done recursively upto 4 levels. | | `` |
4849
| `--name` | | | `` |
4950
| `--extra_cache_tags` | Extra cache tags to be added to the cached entry when the script results are saved | | `` |
5051
| `--skip_compile` | Skip compilation | | `False` |
5152
| `--skip_run` | Skip run | | `False` |
53+
| `--skip_sudo` | Skip SUDO detection | | `False` |
5254
| `--accept_license` | Accept the required license requirement to run the script | | `False` |
5355
| `--skip_system_deps` | Skip installing any system dependencies | | `False` |
5456
| `--git_ssh` | Use SSH for git repos | | `False` |
@@ -72,8 +74,8 @@ No script specific inputs
7274

7375
### Download-tool
7476

75-
- `r2-downloader`
76-
- `rclone` (default)
77+
- `r2-downloader` (default)
78+
- `rclone`
7779

7880
### Run-mode
7981

script/get-preprocessed-dataset-nuscenes/meta.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,8 @@ variations:
2828
tags: _rclone
2929
dae_ac:
3030
tags: _rclone
31-
default: true
3231
r2-downloader:
32+
default: true
3333
env:
3434
MLC_DOWNLOAD_TOOL: r2-downloader
3535
add_deps_recursive:
@@ -155,6 +155,7 @@ variations:
155155
validation,mlc,r2-downloader:
156156
env:
157157
MLC_DOWNLOAD_URL: https://nuscenes.mlcommons-storage.org/metadata/nuscenes-dataset.uri
158+
MLC_DOWNLOAD_FILENAME: nuscenes_dataset
158159
calibration,mlc,r2-downloader:
159160
MLC_DOWNLOAD_URL: https://nuscenes.mlcommons-storage.org/metadata/nuscenes-dataset.uri
160161
tests:
@@ -164,6 +165,5 @@ tests:
164165
- validation,prebuilt,rclone,mlc,dry-run
165166
- calibration,prebuilt,rclone,mlc,dry-run
166167
- validation,prebuilt,r2-downloader,mlc,dry-run
167-
168168
- calibration,prebuilt,r2-downloader,mlc,dry-run
169169

0 commit comments

Comments
 (0)