allow more positional arguments to make the tests happy

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Mar 10, 2024 · 92ac826 · 92ac826
1 parent 3e7de0b
commit 92ac826
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/pt/model/model/dp_model.py b/deepmd/pt/model/model/dp_model.py
@@ -37,7 +37,7 @@ def __new__(
         cls,
         descriptor=None,
         fitting=None,
-        *,
+        *args,
         # disallow positional atomic_model_
         atomic_model_: Optional[DPAtomicModel] = None,
         **kwargs,