From 047ef44310e22d39d2839da54ba3d78107418e6f Mon Sep 17 00:00:00 2001
From: bnebgen-LANL <77704429+bnebgen-LANL@users.noreply.github.com>
Date: Mon, 8 Jan 2024 14:49:35 -0700
Subject: [PATCH] bugfix: Undo Change to VecMag (#52)

Introduced due to a mistake made by git in merging excited state nodes and examples
---
 hippynn/layers/physics.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/hippynn/layers/physics.py b/hippynn/layers/physics.py
index 3debf31d..6c6282e4 100644
--- a/hippynn/layers/physics.py
+++ b/hippynn/layers/physics.py
@@ -247,7 +247,7 @@ def forward(self, features, species):
 
 class VecMag(torch.nn.Module):
     def forward(self, vector_feature):
-        return torch.norm(vector_feature, dim=1).unsqueeze(1)
+        return torch.norm(vector_feature, dim=1)
 
 
 class CombineEnergy(torch.nn.Module):