rename new objects for greater consistency

lanl · Feb 20, 2025 · 2f9ace9 · 2f9ace9
1 parent 6ff3d12
commit 2f9ace9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/hippynn/graphs/nodes/indexers.py b/hippynn/graphs/nodes/indexers.py
@@ -211,9 +211,9 @@ def __init__(self, name, parents, length, vmin, vmax, module="auto", **kwargs):
 
         super().__init__(name, parents, module=module, **kwargs)
 
-class SpeciesIndexed(AutoNoKw, SingleNode, ExpandParents):
+class SpeciesIndexer(AutoNoKw, SingleNode, ExpandParents):
     _input_names = "values", "onehot_encoding"
-    _auto_module_class = index_modules.SpeciesIndex
+    _auto_module_class = index_modules.SpeciesIndexer
     _index_state = IdxType.Atoms
 
     @_parent_expander.match(_BaseNode)

diff --git a/hippynn/layers/indexers.py b/hippynn/layers/indexers.py
@@ -272,7 +272,7 @@ def forward(self, values):
         histo = torch.exp(-((x / self.sigma) ** 2) / 4)
         return torch.flatten(histo, end_dim=1)
 
-class SpeciesIndex(torch.nn.Module):
+class SpeciesIndexer(torch.nn.Module):
     def forward(self, values, onehot_encoding):
         n_species = onehot_encoding.shape[1]
         values_by_species = []