Merge pull request #3638 from JoseAntonioSiguenza/accn_test_fix

Fix PR for acnn test
deepchem · Oct 27, 2023 · cef411a · cef411a
2 parents 3e0f9c5 + 59a9a31
commit cef411a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deepchem/models/torch_models/tests/test_acnn.py b/deepchem/models/torch_models/tests/test_acnn.py
@@ -10,12 +10,12 @@
 
 @pytest.mark.torch
 def test_atomic_convolution_module():
-    from deepchem.models.torch_models.layers import AtomicConvolutionModule
+    from deepchem.models.torch_models.layers import AtomicConv
     f1_num_atoms = 100  # maximum number of atoms to consider in the ligand
     f2_num_atoms = 1000  # maximum number of atoms to consider in the protein
     max_num_neighbors = 12  # maximum number of spatial neighbors for an atom
 
-    acm = AtomicConvolutionModule(
+    acm = AtomicConv(
         n_tasks=1,
         frag1_num_atoms=f1_num_atoms,
         frag2_num_atoms=f2_num_atoms,