Skip to content

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

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