update test to work with graphconv nn

ATOMScience-org · Jan 21, 2025 · 3d81833 · 3d81833
1 parent edb9a74
commit 3d81833
Showing 1 changed file with 4 additions and 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
@@ -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'