dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix huggingface downloader Windows support (#564) (#575)

This commit is contained in:
Chris Cowan 2023-08-03 06:23:12 -07:00 committed by GitHub
parent b7ad23bd87
commit 706a6ea99e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
burn-dataset/src/source/huggingface/downloader.rs
View File

@ -10,6 +10,10 @@ use thiserror::Error;
const PYTHON: &str = "python3";
const PYTHON_SOURCE: &str = include_str!("importer.py");
#[cfg(not(target_os = "windows"))]
const VENV_BIN_PYTHON: &str = "bin/python3";
#[cfg(target_os = "windows")]
const VENV_BIN_PYTHON: &str = "Scripts\\python";
/// Error type for [HuggingfaceDatasetLoader](HuggingfaceDatasetLoader).
#[derive(Error, Debug)]
@ -218,7 +222,7 @@ fn install_python_deps(base_dir: &Path) -> Result<PathBuf, ImporterError> {
ImporterError::FailToDownloadPythonDependencies(format!(" error: {}", err))
})?;
let venv_python_path = venv_dir.join("bin").join(PYTHON);
let venv_python_path = venv_dir.join(VENV_BIN_PYTHON);
let mut command = Command::new(&venv_python_path);
command.args([