Skip to content

Import of sklearn kit is slow, due to tensorflow import #73

@jfischer

Description

@jfischer

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.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions