From 70908bb65cd68c3a2cb9b4b723a4fdeadc8a2afa Mon Sep 17 00:00:00 2001
From: albert bou <albertbou92@gmail.com>
Date: Thu, 8 Aug 2024 18:45:35 +0200
Subject: [PATCH] improve tests

---
 .github/unittest/install_dependencies.sh      | 29 +++++--------------
 .../unittest/install_dependencies_nightly.sh  | 29 +++++--------------
 2 files changed, 14 insertions(+), 44 deletions(-)

diff --git a/.github/unittest/install_dependencies.sh b/.github/unittest/install_dependencies.sh
index e24299d..81a3e85 100644
--- a/.github/unittest/install_dependencies.sh
+++ b/.github/unittest/install_dependencies.sh
@@ -1,25 +1,10 @@
-# Upgrade pip and install dependencies
-python -m pip install --upgrade pip && \
-python -m pip install \
-    flake8 \
-    pytest \
-    pytest-cov \
-    hydra-core \
-    tqdm \
-    torch \
-    torchvision \
-    transformers \
-    promptsmiles \
-    torchrl \
-    rdkit==2023.09.03 \
-    MolScore \
-    deepsmiles \
-    selfies \
-    smi2sdf \
-    smi2svg \
-    atomInSmiles \
-    safe-mol \
-    molbloom
+# Upgrade pip
+python -m pip install --upgrade pip
+
+# Install dependencies
+python -m pip install flake8 pytest pytest-cov hydra-core tqdm deepsmiles selfies smi2sdf smi2svg atomInSmiles safe-mol molbloom
+python -m pip install torch torchvision
+python -m pip install transformers promptsmiles torchrl rdkit==2023.09.03 MolScore # causal-conv1d>=1.4.0 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 3cffa2e..8e04879 100644
--- a/.github/unittest/install_dependencies_nightly.sh
+++ b/.github/unittest/install_dependencies_nightly.sh
@@ -1,25 +1,10 @@
-# Upgrade pip and install dependencies
-python -m pip install --upgrade pip && \
-python -m pip install \
-    flake8 \
-    pytest \
-    pytest-cov \
-    hydra-core \
-    tqdm \
-    torch \
-    torchvision \
-    transformers \
-    promptsmiles \
-    torchrl \
-    rdkit==2023.09.03 \
-    MolScore \
-    deepsmiles \
-    selfies \
-    smi2sdf \
-    smi2svg \
-    atomInSmiles \
-    safe-mol \
-    molbloom
+# Upgrade pip
+python -m pip install --upgrade pip
+
+# Install dependencies
+python -m pip install flake8 pytest pytest-cov hydra-core tqdm deepsmiles selfies smi2sdf smi2svg atomInSmiles safe-mol molbloom
+python -m pip install torch torchvision
+python -m pip install transformers promptsmiles torchrl rdkit==2023.09.03 MolScore # causal-conv1d>=1.4.0 mamba-ssm==1.2.2
 
 # Verify installations
 python -c "import transformers; print(transformers.__version__)"