From 4a54fb141e186f26b2693d0016076066ae60358f Mon Sep 17 00:00:00 2001 From: ShuaishuaiYuan Date: Tue, 22 Oct 2024 22:31:41 -0400 Subject: [PATCH] Update Type Hint for energy_correction Fix the type hint, it can either be a list of float or a single float. so I change it to a union. --- flare/learners/lmpotf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flare/learners/lmpotf.py b/flare/learners/lmpotf.py index a6f7817a9..f390e6213 100644 --- a/flare/learners/lmpotf.py +++ b/flare/learners/lmpotf.py @@ -76,7 +76,7 @@ def __init__( rcut: float, type2number: Union[(int, List[int])], dftcalc: object, - energy_correction: List[float] = 0.0, + energy_correction: Union[(float, List[float])] = 0.0, force_training=True, energy_training=True, stress_training=True,