From d62d84c62328277c638faf0cec8a33d3fda530e7 Mon Sep 17 00:00:00 2001
From: Rene <rene.winchenbach@gmail.com>
Date: Tue, 23 Jul 2024 21:50:11 +0200
Subject: [PATCH] Minor Cleanup v1

---
 src/BasisConvolution/util/augment.py   | 1 -
 src/BasisConvolution/util/testcases.py | 2 ++
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/BasisConvolution/util/augment.py b/src/BasisConvolution/util/augment.py
index 0e92d8d..e6eb71a 100644
--- a/src/BasisConvolution/util/augment.py
+++ b/src/BasisConvolution/util/augment.py
@@ -82,7 +82,6 @@ def loadAugmentedFrame(index, dataset, hyperParameterDict, unrollLength = 8):
 
     # config['neighborhood']['verletScale'] = 1.0
     # config['neighborhood']['scheme'] = 'compact'
-
     searchNeighbors(augmentedStates[0], config, computeKernels = True)
 
     currentState = augmentedStates[0]
diff --git a/src/BasisConvolution/util/testcases.py b/src/BasisConvolution/util/testcases.py
index 4fffea3..1575e88 100644
--- a/src/BasisConvolution/util/testcases.py
+++ b/src/BasisConvolution/util/testcases.py
@@ -420,6 +420,8 @@ def parseSPHConfig(inFile, device, dtype):
     if hasDiffSPH:
         parseDefaultParameters(config)
         parseModuleParameters(config)
+    else:
+        raise ModuleNotFoundError('diffSPH is not installed, cannot parse SPH config')
     
     return config