Databricks IDE 拡張機能の Databricks Connect を使用してコードをデバッグする

この記事では、Databricks IDE 拡張機能の Databricks Connect 統合を使用して、個々の Python (.py) ファイルを実行およびデバッグする方法について説明します。拡張機能の詳細については、Databricks IDE 拡張機能を参照してください。

Databricks Connect 統合では、ノートブック セルのランとデバッグも可能です。Databricks IDE 拡張機能を使用して Databricks Connect でノートブック セルを実行およびデバッグするを参照してください。

必要条件

Databricks IDE 拡張機能内から Databricks Connect を使用するには、まず Databricks Connect の要件を満たす必要があります。これらの要件には、Unity Catalog 対応のワークスペースやコンピュート要件などが含まれます。拡張機能を使用して Python 環境をセットアップすると、正しいバージョンの Python がインストールされます。「Python 環境と Databricks Connect をセットアップする」を参照してください。

Python 環境をセットアップして Databricks Connect をインストールする

拡張機能を使用すると、ターゲットのコンピュートと一致するローカル Python 環境を簡単にセットアップできます。これは、Python バージョンとパッケージセットがランタイムまたはServerlessバージョンと一致するローカル .venv (uv で管理) を構築し、セットアップの一環として互換性のあるバージョンの Databricks Connect をインストールします。

拡張機能の Configuration ビューで、 Set up Python environment をクリックし、コンピュートを選択します。Databricks CLI の代替手段を含む詳細については、「Python 環境と Databricks Connect をセットアップする」を参照してください。

Databricks Connect のインストール後、Visual Studio Code のステータスバーに赤い Databricks Connect disabled ボタンが表示された場合は、それをクリックし、画面の指示に従って有効化してください。 Databricks Connect enabled ボタンが表示されたら、Databricks Connect を使用する準備は完了です。

注記

Poetry を使用している場合は、次のコマンドを実行することで、pyproject.toml ファイルと poetry.lock ファイルをインストール済みの Databricks Connect パッケージ（およびその依存関係）と同期できます。16.4.1 を、プロジェクトの Databricks IDE 拡張機能によってインストールされたものと一致する Databricks Connect パッケージのバージョンに必ず置き換えてください。

Bash

poetry add databricks-connect==16.4.1

Python コードを実行またはデバッグする

Databricks Connect を有効にした後、Python ファイル (.pyを実行またはデバッグします。

1. プロジェクトで、実行またはデバッグする Python ファイルを開きます。

2. Python ファイル内のデバッグ ブレークポイントを設定します。

3. エディター タブの一覧の横にある [Databricks で実行 ] アイコンをクリックし、[ Databricks Connect を使用して現在のファイルをデバッグ ] をクリックします。

   

   出力は「 デバッグコンソール 」ペインに表示されます。

   .py ファイルを右クリックし、[ Databricks で実行 ] > [Databricks Connect を使用して現在のファイルをデバッグ ] をクリックすることもできます。