Update comments in simple ensemble files

Remove / change generic comments which are results of file copies.
elcorto · Mar 18, 2024 · 983db81 · 983db81
1 parent 6c48669
commit 983db81
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/mala/interfaces/ase_calculator_simple_ensemble.py b/mala/interfaces/ase_calculator_simple_ensemble.py
@@ -1,4 +1,4 @@
-"""ASE calculator for MALA predictions."""
+"""ASE SimpleEnsemble calculator for MALA predictions."""
 
 from ase.calculators.calculator import Calculator, all_changes
 import numpy as np
@@ -10,9 +10,7 @@
 
 class MALASimpleEnsemble(MALA):
     """
-    Implements an ASE calculator based on MALA predictions.
-
-    With this, MD simulations can be performed.
+    Implements an ASE calculator based on  SimpleEnsemble MALA predictions.
 
     Parameters
     ----------
@@ -153,4 +151,3 @@ def save_calculator(self, filename):
 
         """
         raise Exception("Saving currently not implemented for this class.")
-
diff --git a/mala/network/simple_ensemble_predictor.py b/mala/network/simple_ensemble_predictor.py
@@ -1,4 +1,3 @@
-"""Tester class for testing a network."""
 import glob
 import os
 from zipfile import ZipFile, ZIP_STORED
@@ -22,7 +21,7 @@
 
 class SimpleEnsemblePredictor(Predictor):
     """
-    A class for running predictions using a neural network.
+    A class for running predictions using a Simple Ensemble of neural networks.
 
     It enables production-level inference.