Skip to content

Remove package's tensorflow dependency or loosen version requirements #17

@cxgoogle

Description

@cxgoogle

Removing the dep slims the package down, making it easier to install, and easier for us package maintainers (fewer bugs/conflicts to deal with).

We already decided that TF model code can go purely in the notebooks, and not be part of the package. The current remaining purposes of TF in the package that prevent us from moving the install to the notebooks are:

  • Reading tfrecord files
  • Generating serialized tfrecords required for the embeddings API

Benefits:

  • Pytorch users will be able to install the package without needing to install both TF and Pytorch in the same environment.
  • TF users may want to run analysis with a different version than the ones required by this package. Similarly, we could consider loosening the version requirements.
  • Colab - once again, the TF version may clash with the existing version, requiring uninstall, etc.

Side note: we also depend on the tensorflow-models package (only in the notebooks) and its version must match the tensorflow version. Currently, we only use tf-models for the optimizer. If the demo notebooks could use an optimizer built into TF, that would remove the need for tf-models. But this isn't an important issue since the package itself doesn't require tf-models.

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