Отладочный код с использованием Databricks Connect для расширения Databricks IDE

В этой статье описывается, как использовать интеграцию Databricks Connect в расширении Databricks IDE для запуска и отладки отдельных файлов Python (.py). Для получения информации о расширении см. расширение Databricks IDE.

Интеграция Databricks Connect также позволяет запускать и отлаживать ячейки блокнота. См. Запуск и отладка ячеек ноутбуков с помощью Databricks Connect с помощью расширения Databricks IDE.

Требования

Прежде чем использовать Databricks Connect внутри расширения Databricks IDE, сначала необходимо выполнить требования Databricks Connect. К этим требованиям относятся такие вещи, как рабочее пространство с поддержкой Unity Catalog и вычислительные требования. Когда вы настраиваете свою среду Python с помощью расширения, для вас устанавливается правильная версия Python. См. «Настройте свою Python-среду» и «Databricks Connect».

Настройте свою среду Python и установите Databricks Connect

Вы можете использовать это расширение, чтобы легко настроить локальную Python-среду, соответствующую вашей целевой вычислительной системе. Он создаёт локальный .venv (под управлением uv), версия Python и набор пакетов которого соответствуют версии runtime или serverless, и в процессе настройки устанавливает совместимую версию Databricks Connect.

В представлении расширения Конфигурация нажмите Настроить среду Python и выберите вычислительный ресурс. Для подробностей, включая альтернативу CLI Azure Databricks, см. разделы «Настройте вашу среду Python» и «Databricks Connect».

После установки Databricks Connect, если в строке статуса Visual Studio Code появится красная кнопка отключения Databricks Connect, нажмите на неё и выполните инструкции на экране для её включения. Когда появится кнопка с включением Databricks Connect , вы готовы использовать Databricks Connect.

Примечание.

При использовании Poetry вы можете синхронизировать файлы pyproject.toml и poetry.lock с установленным пакетом Databricks Connect и его зависимостями, выполнив следующую команду. Обязательно замените 16.4.1 на версию пакета Databricks Connect, которая соответствует версии, установленной расширением Databricks IDE для вашего проекта.

poetry add databricks-connect==16.4.1

Запустите или отладьте свой код на Python.

После включения Databricks Connect запустите или отладите файл Python (.py):

1. В проекте откройте файл Python, который требуется запустить или отлаживать.

2. Задайте все точки останова отладки в файле Python.

3. Щелкните значок Запустить в Databricks рядом со списком вкладок редактора, а затем щелкните Отладка текущего файла с помощью Databricks Connect.

   

   Выходные данные отображаются в Консоли отладки.

   Вы также можете щелкнуть правой кнопкой мыши файл .py, а затем нажать кнопку Запустить в Databricks>, Отладка текущего файла с помощью Databricks Connect.