I am trying to create a new custom module and import it in my research notebook but am having some issues.

The module named test_mod.py looks like this:

from AlgorithmImports import *

def test_func():
    return 'success'

And then in the research notebook, I do the following:

import test_mod


But I get this error:

ModuleNotFoundError                       Traceback (most recent call last)
Cell In [3], line 1
----> 1 import test_mod

ModuleNotFoundError: No module named 'test_mod'


I am not sure what I am missing here. Any ideas? Thanks