Conda environment for top2vec

top2vec.yml

name: top2vec
channels:
  - defaults
  - conda-forge
dependencies:
  - python ==3.8
#  - tensorflow ==2.2.0  # tensorflow_text 2.2.0 doesn't work!
  # tf dependecies
  - numpy <1.19.0,>=1.16.0
  - termcolor
  - google-pasta
  - h5py
  - wrapt
  - opt-einsum
  - gast
  - astunparse
  #
  - scipy ==1.4.1
  - tensorboard >=2.2,<2.3
  - tensorflow-hub ==0.8.0
  - pandas
  - pynndescent>=0.4
  - future
  - cython
  - pillow
  - matplotlib
  - typing_extensions
  - tqdm
  - nltk
  - conda-forge::tokenizers
  - conda-forge::umap-learn
  - conda-forge::wordcloud
  - conda-forge::gensim
  - conda-forge::hdbscan
#  - conda-forge::transformers <3.4.0,>=3.1.0  # conda version requires cuda, which is not useful for non-nvidia gpus and pytorch, which is not used by top2vec
#  - conda-forge::sentence-transformers  # same as above
  - pip
  - pip:
    - sentence-transformers
    - transformers
    - torch
    - tensorflow ==2.3.0
    - tensorflow_text ==2.3.0  # should be the same version as tensorflow
    - top2vec