Use custom functions written in Python within a Databricks notebook

Question

I have created a python file (.py) that includes two different functions.
For example,

def function1():
    print("Hello World")

def funtion2(a, b):
    y=a+b
    if y>=5:
        print("Correct")
    else:
        print("Wrong")

Now, I want to use those two functions in a Databricks Notebook. What I did, was to follow the steps written in this databricks post. Even though I succeeded on creating an egg file that was later imported as a library in databricks I didn't manage to import my custom functions from the egg file. Please check the screenshots below:

setup.py

egg file on Databricks

When I try the python command:

import function2

I get an error that this module was not found. I appreciate any help and comments.
Note: init.py file is not used

Try "from python_functions.py import function2" or try " import python_function.py" then use function2. — Zubad Ibrahim
– Zubad Ibrahim, Commented May 5, 2020 at 8:19
@ZubadIbrahim Both ways you have written already gave me "No module named 'python_functions'". That's why I tried the .egg file approach. — NikSp
– NikSp, Commented May 5, 2020 at 8:40

LopDev · Accepted Answer · 2020-10-01 09:07:35Z

4

I found the following solution after some searching in the Web:

1. Create library notebook.
For example - Lib with any functions/classes there (no runnable code).
1. To import into Main all the classes and functions from Lib to Main use this command:
```
%run "./Lib" 
```
  (this will work like: from Lib import *)
1. After that you can call any functions / use classes that used in the Lib from Main notebook.

This is the online post with the information.

edited Oct 1, 2020 at 9:07

LopDev

83912 silver badges27 bronze badges

answered May 6, 2020 at 11:08

NikSp

1,5894 gold badges26 silver badges52 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Use custom functions written in Python within a Databricks notebook

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related