From e22b491893f8b250b60bf14a698b353a67b79d6c Mon Sep 17 00:00:00 2001 From: albert bou Date: Thu, 8 Aug 2024 19:11:41 +0200 Subject: [PATCH] improve tests --- .github/unittest/install_dependencies.sh | 4 +++- .github/unittest/install_dependencies_nightly.sh | 2 ++ 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/.github/unittest/install_dependencies.sh b/.github/unittest/install_dependencies.sh index f52ce9d..0bf44c6 100644 --- a/.github/unittest/install_dependencies.sh +++ b/.github/unittest/install_dependencies.sh @@ -4,7 +4,7 @@ python -m pip install --upgrade pip # Install dependencies python -m pip install flake8 pytest pytest-cov hydra-core tqdm python -m pip install torch torchvision -python -m pip install transformers promptsmiles torchrl rdkit==2023.3.3 MolScore # causal-conv1d>=1.4.0 mamba-ssm==1.2.2 +python -m pip install transformers promptsmiles torchrl rdkit==2023.3.3 MolScore python -m pip install deepsmiles python -m pip install selfies python -m pip install smi2sdf @@ -13,6 +13,8 @@ python -m pip install atomInSmiles python -m pip install safe-mol python -m pip install smizip python -m pip install molbloom +python -m pip causal-conv1d>=1.4.0 +python -m pip mamba-ssm==1.2.2 # Verify installations python -c "import transformers; print(transformers.__version__)" diff --git a/.github/unittest/install_dependencies_nightly.sh b/.github/unittest/install_dependencies_nightly.sh index 0d589ea..0366e49 100644 --- a/.github/unittest/install_dependencies_nightly.sh +++ b/.github/unittest/install_dependencies_nightly.sh @@ -13,6 +13,8 @@ python -m pip install atomInSmiles python -m pip install safe-mol python -m pip install smizip python -m pip install molbloom +python -m pip causal-conv1d>=1.4.0 +python -m pip mamba-ssm==1.2.2 # Verify installations python -c "import transformers; print(transformers.__version__)"