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

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

Databricks Connect の統合により、ノートブックのセルを実行およびデバッグすることもできます。「Visual Studio Code の Databricks 拡張機能を使用して Databricks Connect でノートブックセルを実行およびデバッグする」を参照してください。

要件

Visual Studio Code の Databricks 拡張機能内から Databricks Connect を使用するには、まず Databricks Connect の要件を満たす必要があります。これらの要件には、Unity Catalog 対応ワークスペース、コンピュート要件、 Pythonのローカルインストールのバージョン要件などが含まれます。

Python 仮想環境をアクティブ化する

Python プロジェクトの Python 仮想環境 をアクティブ化します。 Python 仮想環境は、プロジェクトが互換性のあるバージョンの Python と Python パッケージ (この場合は Databricks Connect パッケージ) を使用していることを確認するのに役立ちます。

[Configuration] ペインで、次の操作を行います。

「Python Environment」の下にある赤い「Activate Virtual Environment」項目をクリックします。
コマンドパレットで、[Venv] または [Conda] を選択します。
インストールする依存関係を選択します (存在する場合)。

Databricks Connect のインストール

「Python環境」の下の「構成」ビューで、次のようにします。

赤い [Databricks のインストール] 再生ボタンをクリックします。
クラスターをアタッチします。
- クラスターセクションが拡張機能でまだ構成されていない場合は、「Databricks Connectを使用するためにクラスターをアタッチしてください」というメッセージが表示されます。[クラスターのアタッチ] をクリックし、Databricks Connect要件を満たすクラスターを選択します。
- クラスターセクションが構成されているが、クラスターがDatabricks Connectと互換性がない場合は、赤いDatabricks Connect無効ボタンをクリックし、[クラスターのアタッチ]をクリックして、互換性のあるクラスターを選択します。
Databricks Connect パッケージ (およびその依存関係) がまだインストールされていない場合は、"対話型デバッグとオートコンプリートには Databricks Connect が必要です。 <environment-name>環境に設置してみませんか。「インストール」をクリックします。
Visual Studio Code のステータスバーで、赤い Databricks Connect の無効化 ボタンが表示されている場合は、それをクリックし、画面の指示を完了して有効にします。
[Databricks Connect が有効] ボタンが表示されたら、Databricks Connect を使用する準備が整いました。

注：

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

poetry add databricks-connect==13.3.2

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

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

プロジェクトで、実行またはデバッグする Python ファイルを開きます。
Python ファイル内のデバッグブレークポイントを設定します。
エディタータブの一覧の横にある [Databricks で実行 ] アイコンをクリックし、[ Databricks Connect: 現在のファイルをデバッグ] をクリックします。

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

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