diff --git a/datalab/datalab_session/data_operations/utils.py b/datalab/datalab_session/data_operations/utils.py index a3bf4c1..13d103d 100644 --- a/datalab/datalab_session/data_operations/utils.py +++ b/datalab/datalab_session/data_operations/utils.py @@ -1,5 +1,6 @@ from pkgutil import walk_packages import inspect +from importlib import import_module from django.utils.module_loading import import_string from datalab.datalab_session import data_operations @@ -8,7 +9,7 @@ def available_operations(): operations = {} base_operation = import_string('datalab.datalab_session.data_operations.data_operation.BaseDataOperation') for (loader, module_name, _) in walk_packages(data_operations.__path__): - module = loader.find_module(module_name).load_module() + module = import_module(f'{data_operations.__name__}.{module_name}') members = inspect.getmembers(module, inspect.isclass) for member in members: if member[0] != 'BaseDataOperation' and issubclass(member[1], base_operation):