From 3d818333dcaf4e70872b72852bd1ba9055f85ce6 Mon Sep 17 00:00:00 2001
From: Amanda Paulson <amanda.paulson@ucsf.edu>
Date: Tue, 21 Jan 2025 15:42:43 -0800
Subject: [PATCH] update test to work with graphconv nn

---
 atomsci/ddm/test/integrative/ad_index/test_ad_index.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/atomsci/ddm/test/integrative/ad_index/test_ad_index.py b/atomsci/ddm/test/integrative/ad_index/test_ad_index.py
index 443700b3..7774f31b 100644
--- a/atomsci/ddm/test/integrative/ad_index/test_ad_index.py
+++ b/atomsci/ddm/test/integrative/ad_index/test_ad_index.py
@@ -35,7 +35,10 @@ def test():
     hp_params["python_path"] = python_path
     
     for feat in ['ecfp','mordred_filtered','rdkit_raw','graphconv']:
-        if feat in ['ecfp','graphconv']:
+        if feat == 'ecfp':
+            hp_params['featurizer']=feat
+        elif feat =='graphconv':
+            hp_params['model_type']='NN'
             hp_params['featurizer']=feat
         else:
             hp_params['featurizer']='computed_descriptors'