From f721b849d163eb4588d3618576d9730f3c749ca5 Mon Sep 17 00:00:00 2001
From: knc6 <writetokamal.1989@gmail.com>
Date: Sat, 22 Jun 2024 17:53:48 -0400
Subject: [PATCH] Fix test.

---
 .github/workflows/main.yml | 2 +-
 setup.py                   | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index 30ff3c3..3842902 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -35,7 +35,7 @@ jobs:
             find . -type f > before_test_files.txt
             conda env create -f environment.yml
             conda activate atomgpt
-            conda install pytest coverage codecov 
+            conda install pytest coverage codecov  alignn -y
             #conda install pytest coverage codecov pandas numpy matplotlib phonopy scikit-learn jarvis-tools --quiet
             #export DGLBACKEND=pytorch
             #export CUDA_VISIBLE_DEVICES="-1"
diff --git a/setup.py b/setup.py
index aa12453..0a407e4 100644
--- a/setup.py
+++ b/setup.py
@@ -17,7 +17,7 @@
         "pydantic_settings",
         "peft",
         "trl",
-        "alignn",
+        #"alignn",
         "triton",
         "torch",