How to run a notebook in a .py file in databricks

[Littlesheep_]

The situation is that my colleague was using pycharm and now needs to adapt to databricks. They are now doing their job by connecting VScode to databricks and run the .py file using databricks clusters.

The problem is they want to call a notebook in databricks in the .py file in VScode, the %run command is not working because this command only works in notebook. I am wondering if there is a way to do it?

Thanks so much!

[Hkesharwani]

Hi, 
One way I can think of to achieve this is by calling the rest API of Databricks job (Create a job with only one task)
Check out this documentation for more details.
https://docs.databricks.com/api/workspace/jobs/runnow

[jacovangelder]

You can do so by adding your own dbutils function in your py file:

def get_dbutils():
    """
    This is to make your local env (and flake  happy."""
    from pyspark.sql import SparkSession

    spark = SparkSession.getActiveSession()
    if spark.conf.get("spark.databricks.service.client.enabled") == "true":
        from pyspark.dbutils import DBUtils

        return DBUtils(spark)
    else:
        import IPython

        return IPython.get_ipython().user_ns["dbutils"]

And then run the notebook from the py file using the following way:

get_dbutils().notebook.run('path/to/notebook', <timeout_in_seconds>) 

Good luck!

[Rishabh_Tiwari]

Hi @Littlesheep_ ,

Thank you for reaching out to our community! We're here to help you. 

To ensure we provide you with the best support, could you please take a moment to review the response and choose the one that best answers your question? Your feedback not only helps us assist you better but also benefits other community members who may have similar questions in the future.

If you found the answer helpful, consider giving it a kudo. If the response fully addresses your question, please mark it as the accepted solution. This will help us close the thread and ensure your question is resolved.

We appreciate your participation and are here to assist you further if you need it!

Thanks,

Rishabh