如何使用GitHub Actions构建wheel并导出至DataBricks
实现GitHub Actions构建Wheel包并上传至Databricks指定文件夹
你已经完成了Wheel包的构建流程,接下来可以通过Databricks CLI实现将构建好的Wheel包上传到Databricks的指定文件夹(DBFS路径)——你找到的microsoft/databricks-import-notebook是专门用于Notebook导入的Action,并不适用于Wheel文件上传。
修改后的完整GitHub Actions配置
on: push: branches: - main - master jobs: build_wheel: runs-on: ubuntu-latest steps: - uses: actions/checkout@v2 - name: Set up Python uses: actions/setup-python@v2 with: python-version: 3.9 - name: Build wheel run: | python -m pip install --user --upgrade build python -m build - name: Install Databricks CLI run: | pip install databricks-cli - name: Configure Databricks credentials run: | databricks configure --token <<EOF ${{ secrets.DATABRICKS_HOST }} ${{ secrets.DATABRICKS_TOKEN }} EOF - name: Upload Wheel to Databricks DBFS run: | # 替换为你的Databricks目标DBFS路径,比如 /FileStore/wheels/${{ github.sha }} databricks fs cp ./dist/*.whl dbfs:/your/target/path/ --overwrite
关键步骤说明
- 安装Databricks CLI:通过pip直接安装官方CLI工具,用于和Databricks API交互
- 配置凭据:使用你已存入GitHub Secrets的
DATABRICKS_HOST和DATABRICKS_TOKEN完成CLI授权,避免硬编码敏感信息 - 上传Wheel包:用
databricks fs cp命令将dist目录下的Wheel文件上传到指定的DBFS路径,--overwrite参数用于覆盖同名文件(可选)
注意事项
- 确保GitHub Secrets中已正确配置
DATABRICKS_HOST(格式如https://adb-xxxx.azuredatabricks.net/)和DATABRICKS_TOKEN(从Databricks控制台生成的个人访问令牌) - 目标DBFS路径需要提前存在,或者可以在上传前添加
databricks fs mkdirs dbfs:/your/target/path/命令创建目录
内容的提问来源于stack exchange,提问作者shalva_t
相关产品推荐
相关产品推荐

