If I run:
from dataworkspaces.kits.scikit_learn import LineagePredictor, load_dataset_from_resource
it takes over 10 seconds. This is because code in wrapper_utils.py is trying to import TensorFlow, which is huge. The solution is to import tensorflow in specific functions, only if specific tests fail.